def test_happy_path_add_default_jupyter(self):
runner = mock.MagicMock()
runner.return_value.returncode = 0
environment = self.temporary_dir
env_name = os.path.basename(self.temporary_dir)
with mock.patch("sys.stdout", new=io.StringIO()) as new_stdout:
commands.add(environment, None, None, runner, {}, None)
output = new_stdout.getvalue().split()
assert_that(output, has_items_in_order(environment, env_name, "jupyter"))
assert_that(runner.call_count, equal_to(3))
def test_happy_path_existing_files(self):
runner = mock.MagicMock()
runner.return_value.returncode = 0
with temp_dir() as dirname:
environment = os.path.join(dirname, "best-env")
os.makedirs(os.path.join(environment, "bin"))
with open(os.path.join(environment, "bin", "python"), "w"):
pass
with mock.patch("sys.stdout", new=io.StringIO()) as new_stdout:
commands.add(environment, None, None, runner, {}, None)
output = new_stdout.getvalue().split()
assert_that(output, has_items_in_order(environment, "best-env", "jupyter"))
assert_that(runner.call_count, equal_to(3))
def test_not_finding_env(self):
runner = mock.MagicMock(name="runner")
runner.side_effect = OSError("Cannot run this")
environment = os.path.join(self.temporary_dir, "no-such-env")
with mock.patch("sys.stdout", new=io.StringIO()) as new_stdout:
commands.add(environment, None, None, runner, {}, None)
lines = new_stdout.getvalue().splitlines()
assert_that(
lines,
contains_exactly(
contains_string("no-such-env"),
),
)
def test_happy_path_workon_home_default_env(self):
runner = mock.MagicMock()
runner.return_value.returncode = 0
with temp_dir() as dirname:
environment = os.path.join(dirname, "best-env")
cwd = os.path.join(dirname, "this-doesnt-exist", "best-env")
os_environ = dict(WORKON_HOME=dirname)
os.makedirs(os.path.join(environment, "bin"))
with open(os.path.join(environment, "bin", "python"), "w"):
pass
with mock.patch("sys.stdout", new=io.StringIO()) as new_stdout:
commands.add(None, None, None, runner, os_environ, cwd)
output = new_stdout.getvalue().split()
assert_that(output, has_items_in_order(environment, "best-env", "jupyter"))
assert_that(runner.call_count, equal_to(3))
def test_not_running_env_add(self):
runner = mock.MagicMock(name="runner")
runner.side_effect = OSError("Cannot run this")
environment = self.temporary_dir
name = "an-awesome-env"
jupyter = "/path/to/jupyter"
with mock.patch("sys.stdout", new=io.StringIO()) as new_stdout:
commands.add(environment, name, jupyter, runner, {}, None)
lines = new_stdout.getvalue().splitlines()
assert_that(
lines,
contains_exactly(
contains_string("install ipykernel"),
contains_string("Cannot run this"),
),
)
def test_no_workon_no_env(self):
runner = mock.MagicMock()
runner.return_value.returncode = 0
with temp_dir() as dirname:
environment = os.path.join(dirname, "best-env")
cwd = os.path.join(dirname, "this-doesnt-exist", "best-env")
os_environ = {}
os.makedirs(os.path.join(environment, "bin"))
with open(os.path.join(environment, "bin", "python"), "w"):
pass
with mock.patch("sys.stdout", new=io.StringIO()) as new_stdout:
commands.add(None, None, None, runner, os_environ, cwd)
lines = new_stdout.getvalue().splitlines()
assert_that(
lines,
contains_exactly(
contains_string("WORKON_HOME"),
),
)
def test_bad_env_add(self):
runner = mock.MagicMock(name="runner")
runner.return_value.returncode = 1
runner.return_value.stderr = "that environment, it does not exist\n"
runner.return_value.stdout = "I'm sorry dave, I can't do that\n"
environment = self.temporary_dir
name = "an-awesome-env"
jupyter = "/path/to/jupyter"
with mock.patch("sys.stdout", new=io.StringIO()) as new_stdout:
commands.add(environment, name, jupyter, runner, {}, None)
lines = new_stdout.getvalue().splitlines()
assert_that(
lines,
contains_exactly(
contains_string("install ipykernel"),
"Output:",
runner.return_value.stdout.strip(),
"Error:",
runner.return_value.stderr.strip(),
),
)
def test_happy_path_add(self):
runner = mock.MagicMock()
runner.return_value.returncode = 0
environment = self.temporary_dir
name = "an-awesome-env"
jupyter = "/path/to/jupyter"
with mock.patch("sys.stdout", new=io.StringIO()) as new_stdout:
commands.add(environment, name, jupyter, runner, {}, None)
output = new_stdout.getvalue().split()
assert_that(output, has_items_in_order(environment, name, jupyter))
assert_that(runner.call_count, equal_to(3))
install, ipykernel, jupyter = runner.call_args_list
[args], kwargs = install
assert_that(
args,
contains_exactly(
*f"{self.temporary_dir}/bin/python -m pip install ipykernel".split()
),
)
[args], kwargs = ipykernel
assert_that(
args,
contains_exactly(
*f"{self.temporary_dir}/bin/python -m ipykernel install "
"--name an-awesome-env-venv "
"--display-name an-awesome-env "
f"--prefix {self.temporary_dir}".split()
),
)
[args], kwargs = jupyter
assert_that(
args,
contains_exactly(
*"/path/to/jupyter kernelspec install "
f"{self.temporary_dir}/share/jupyter/kernels/"
"an-awesome-env-venv --sys-prefix".split()
),
)
