mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-10 11:32:57 +00:00
130 lines
4.1 KiB
Python
130 lines
4.1 KiB
Python
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
from tests.conftest import TrackedContainer
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
THIS_DIR = Path(__file__).parent.resolve()
|
|
|
|
|
|
def test_run_hooks_zero_args(container: TrackedContainer) -> None:
|
|
logs = container.run_and_wait(
|
|
timeout=5,
|
|
tty=True,
|
|
no_failure=False,
|
|
command=["bash", "-c", "source /usr/local/bin/run-hooks.sh"],
|
|
)
|
|
assert "Should pass exactly one directory" in logs
|
|
|
|
|
|
def test_run_hooks_two_args(container: TrackedContainer) -> None:
|
|
logs = container.run_and_wait(
|
|
timeout=5,
|
|
tty=True,
|
|
no_failure=False,
|
|
command=[
|
|
"bash",
|
|
"-c",
|
|
"source /usr/local/bin/run-hooks.sh first-arg second-arg",
|
|
],
|
|
)
|
|
assert "Should pass exactly one directory" in logs
|
|
|
|
|
|
def test_run_hooks_missing_dir(container: TrackedContainer) -> None:
|
|
logs = container.run_and_wait(
|
|
timeout=5,
|
|
tty=True,
|
|
no_failure=False,
|
|
command=[
|
|
"bash",
|
|
"-c",
|
|
"source /usr/local/bin/run-hooks.sh /tmp/missing-dir/",
|
|
],
|
|
)
|
|
assert "Directory /tmp/missing-dir/ doesn't exist or is not a directory" in logs
|
|
|
|
|
|
def test_run_hooks_dir_is_file(container: TrackedContainer) -> None:
|
|
logs = container.run_and_wait(
|
|
timeout=5,
|
|
tty=True,
|
|
no_failure=False,
|
|
command=[
|
|
"bash",
|
|
"-c",
|
|
"touch /tmp/some-file && source /usr/local/bin/run-hooks.sh /tmp/some-file",
|
|
],
|
|
)
|
|
assert "Directory /tmp/some-file doesn't exist or is not a directory" in logs
|
|
|
|
|
|
def test_run_hooks_empty_dir(container: TrackedContainer) -> None:
|
|
container.run_and_wait(
|
|
timeout=5,
|
|
tty=True,
|
|
command=[
|
|
"bash",
|
|
"-c",
|
|
"mkdir /tmp/empty-dir && source /usr/local/bin/run-hooks.sh /tmp/empty-dir/",
|
|
],
|
|
)
|
|
|
|
|
|
def test_run_hooks_with_files(container: TrackedContainer) -> None:
|
|
host_data_dir = THIS_DIR / "run-hooks-data"
|
|
cont_data_dir = "/home/jovyan/data"
|
|
# https://forums.docker.com/t/all-files-appear-as-executable-in-file-paths-using-bind-mount/99921
|
|
# Unfortunately, Docker treats all files in mounter dir as executable files
|
|
# So we make a copy of mounted dir inside a container
|
|
command = (
|
|
"cp -r /home/jovyan/data/ /home/jovyan/data-copy/ &&"
|
|
"source /usr/local/bin/run-hooks.sh /home/jovyan/data-copy/ &&"
|
|
"echo SOME_VAR is ${SOME_VAR}"
|
|
)
|
|
logs = container.run_and_wait(
|
|
timeout=5,
|
|
volumes={str(host_data_dir): {"bind": cont_data_dir, "mode": "ro"}},
|
|
tty=True,
|
|
command=["bash", "-c", command],
|
|
)
|
|
assert "Executable python file was successfully" in logs
|
|
assert "Ignoring non-executable: /home/jovyan/data-copy//non_executable.py" in logs
|
|
assert "SOME_VAR is 123" in logs
|
|
|
|
|
|
def test_run_hooks_with_failures(container: TrackedContainer) -> None:
|
|
host_data_dir = THIS_DIR / "run-hooks-failures"
|
|
cont_data_dir = "/home/jovyan/data"
|
|
# https://forums.docker.com/t/all-files-appear-as-executable-in-file-paths-using-bind-mount/99921
|
|
# Unfortunately, Docker treats all files in mounter dir as executable files
|
|
# So we make a copy of mounted dir inside a container
|
|
command = (
|
|
"cp -r /home/jovyan/data/ /home/jovyan/data-copy/ &&"
|
|
"source /usr/local/bin/run-hooks.sh /home/jovyan/data-copy/"
|
|
)
|
|
logs = container.run_and_wait(
|
|
timeout=5,
|
|
volumes={str(host_data_dir): {"bind": cont_data_dir, "mode": "ro"}},
|
|
tty=True,
|
|
no_failure=False,
|
|
command=["bash", "-c", command],
|
|
)
|
|
|
|
for file in ["a.sh", "b.py", "c.sh", "d.sh"]:
|
|
assert f"Started: {file}" in logs
|
|
|
|
for file in ["a.sh"]:
|
|
assert f"Finished: {file}" in logs
|
|
for file in ["b.py", "c.sh", "d.sh"]:
|
|
assert f"Finished: {file}" not in logs
|
|
|
|
for file in ["b.py", "c.sh"]:
|
|
assert (
|
|
f"/home/jovyan/data-copy//{file} has failed, continuing execution" in logs
|
|
)
|
|
|
|
assert "OTHER_VAR=456" in logs
|