/usr/lib/python3/dist-packages/jupyter_core/tests/test_command.py is in python3-jupyter-core 4.4.0-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 | """Test the Jupyter command-line"""
import json
import os
import sys
from subprocess import check_output, CalledProcessError
import pytest
try:
from unittest.mock import patch
except ImportError:
# py2
from mock import patch
from jupyter_core import __version__
from jupyter_core.command import list_subcommands
from jupyter_core.paths import (
jupyter_config_dir, jupyter_data_dir, jupyter_runtime_dir,
jupyter_path, jupyter_config_path,
)
def get_jupyter_output(cmd):
"""Get output of a jupyter command"""
if not isinstance(cmd, list):
cmd = [cmd]
return check_output([sys.executable, '-m', 'jupyter_core'] + cmd).decode('utf8').strip()
def write_executable(path, source):
if sys.platform == 'win32':
script = path.dirpath() / path.purebasename + '-script.py'
exe = path.dirpath() / path.purebasename + '.exe'
else:
script = path
script.write(source)
script.chmod(0o700)
if sys.platform == 'win32':
try:
import pkg_resources
w = pkg_resources.resource_string('setuptools', 'cli-32.exe')
except (ImportError, FileNotFoundError):
pytest.skip('Need pkg_resources/setuptools to make scripts executable on Windows')
exe.write(w, 'wb')
exe.chmod(0o700)
def assert_output(cmd, expected):
assert get_jupyter_output(cmd) == expected
def test_config_dir():
assert_output('--config-dir', jupyter_config_dir())
def test_data_dir():
assert_output('--data-dir', jupyter_data_dir())
def test_runtime_dir():
assert_output('--runtime-dir', jupyter_runtime_dir())
def test_paths():
output = get_jupyter_output('--paths')
for d in (jupyter_config_dir(), jupyter_data_dir(), jupyter_runtime_dir()):
assert d in output
for key in ('config', 'data', 'runtime'):
assert ('%s:' % key) in output
for path in (jupyter_config_path(), jupyter_path()):
for d in path:
assert d in output
def test_paths_json():
output = get_jupyter_output(['--paths', '--json'])
data = json.loads(output)
assert sorted(data) == ['config', 'data', 'runtime']
for key, path in data.items():
assert isinstance(path, list)
def test_subcommand_not_given():
with pytest.raises(CalledProcessError):
get_jupyter_output([])
def test_help():
output = get_jupyter_output('-h')
def test_subcommand_not_found():
with pytest.raises(CalledProcessError):
output = get_jupyter_output('nonexistant-subcommand')
@patch.object(sys, 'argv', [__file__] + sys.argv[1:])
def test_subcommand_list(tmpdir):
a = tmpdir.mkdir("a")
for cmd in ('jupyter-foo-bar',
'jupyter-xyz',
'jupyter-babel-fish'):
a.join(cmd).write('')
b = tmpdir.mkdir("b")
for cmd in ('jupyter-foo',
'jupyterstuff',
'jupyter-yo-eyropa-ganymyde-callysto'):
b.join(cmd).write('')
path = os.pathsep.join(map(str, [a, b]))
with patch.dict('os.environ', {'PATH': path}):
subcommands = list_subcommands()
assert subcommands == [
'babel-fish',
'foo',
'xyz',
'yo-eyropa-ganymyde-callysto',
]
def test_not_on_path(tmpdir):
a = tmpdir.mkdir("a")
jupyter = a.join('jupyter')
jupyter.write(
'from jupyter_core import command; command.main()'
)
jupyter.chmod(0o700)
witness = a.join('jupyter-witness')
witness_src = '#!%s\n%s\n' % (sys.executable, 'print("WITNESS ME")')
write_executable(witness, witness_src)
env = {'PATH': ''}
if 'SYSTEMROOT' in os.environ: # Windows http://bugs.python.org/issue20614
env[str('SYSTEMROOT')] = os.environ['SYSTEMROOT']
if sys.platform == 'win32':
env[str('PATHEXT')] = '.EXE'
# This won't work on windows unless
out = check_output([sys.executable, str(jupyter), 'witness'], env=env)
assert b'WITNESS' in out
def test_path_priority(tmpdir):
a = tmpdir.mkdir("a")
jupyter = a.join('jupyter')
jupyter.write(
'from jupyter_core import command; command.main()'
)
jupyter.chmod(0o700)
witness_a = a.join('jupyter-witness')
witness_a_src = '#!%s\n%s\n' % (sys.executable, 'print("WITNESS A")')
write_executable(witness_a, witness_a_src)
b = tmpdir.mkdir("b")
witness_b = b.join('jupyter-witness')
witness_b_src = '#!%s\n%s\n' % (sys.executable, 'print("WITNESS B")')
write_executable(witness_b, witness_b_src)
env = {'PATH': str(b)}
if 'SYSTEMROOT' in os.environ: # Windows http://bugs.python.org/issue20614
env[str('SYSTEMROOT')] = os.environ['SYSTEMROOT']
if sys.platform == 'win32':
env[str('PATHEXT')] = '.EXE'
out = check_output([sys.executable, str(jupyter), 'witness'], env=env)
assert b'WITNESS A' in out
|