Skip to content

Commit

Permalink
Merge https://github.com/cisagov/skeleton-docker into lineage/skeleton
Browse files Browse the repository at this point in the history
# Conflicts:
#	tests/conftest.py
#	tests/container_test.py
  • Loading branch information
mcdonnnj committed Jul 28, 2023
2 parents b6c8951 + a9d6c92 commit ab98639
Show file tree
Hide file tree
Showing 3 changed files with 79 additions and 3 deletions.
2 changes: 1 addition & 1 deletion requirements-test.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
--requirement requirements.txt
pre-commit
pytest
pytest-dockerc
python-on-whales
18 changes: 17 additions & 1 deletion tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,26 +4,42 @@
"""
# Third-Party Libraries
import pytest
from python_on_whales import docker

MAIN_SERVICE_NAME = "saver"


@pytest.fixture(scope="session")
def dockerc():
"""Start up the Docker composition."""
docker.compose.up(detach=True)
yield docker
docker.compose.down()


@pytest.fixture(scope="session")
def main_container(dockerc):
"""Return the main container from the Docker composition."""
# find the container by name even if it is stopped already
return dockerc.containers(service_names=[MAIN_SERVICE_NAME], stopped=True)[0]
return dockerc.compose.ps(services=[MAIN_SERVICE_NAME], all=True)[0]


# See #60
# @pytest.fixture(scope="session")
# def version_container(dockerc):
# """Return the version container from the Docker composition.

<<<<<<< HEAD
# The version container should just output the version of its underlying contents.
# """
# # find the container by name even if it is stopped already
# return dockerc.containers(service_names=[VERSION_SERVICE_NAME], stopped=True)[0]
=======
The version container should just output the version of its underlying contents.
"""
# find the container by name even if it is stopped already
return dockerc.compose.ps(services=[VERSION_SERVICE_NAME], all=True)[0]
>>>>>>> a9d6c92ea3ca2760e4a18276d06c668058dd3670


def pytest_addoption(parser):
Expand Down
62 changes: 61 additions & 1 deletion tests/container_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,11 @@ def test_container_count(dockerc):
"""Verify the test composition and container."""
# stopped parameter allows non-running containers in results
assert (
len(dockerc.containers(stopped=True)) == 2
len(dockerc.compose.ps(all=True)) == 2
), "Wrong number of containers were started."


<<<<<<< HEAD
# See #60
# def test_wait_for_ready(main_container):
# """Wait for container to be ready."""
Expand Down Expand Up @@ -42,6 +43,38 @@ def test_container_count(dockerc):
# main_container.wait() # make sure container exited if running test isolated
# log_output = main_container.logs().decode("utf-8")
# assert SECRET_QUOTE in log_output, "Secret not found in log output."
=======
def test_wait_for_ready(main_container):
"""Wait for container to be ready."""
TIMEOUT = 10
for i in range(TIMEOUT):
if READY_MESSAGE in main_container.logs():
break
time.sleep(1)
else:
raise Exception(
f"Container does not seem ready. "
f'Expected "{READY_MESSAGE}" in the log within {TIMEOUT} seconds.'
)


def test_wait_for_exits(dockerc, main_container, version_container):
"""Wait for containers to exit."""
assert (
dockerc.wait(main_container.id) == 0
), "Container service (main) did not exit cleanly"
assert (
dockerc.wait(version_container.id) == 0
), "Container service (version) did not exit cleanly"


def test_output(dockerc, main_container):
"""Verify the container had the correct output."""
# make sure container exited if running test isolated
dockerc.wait(main_container.id)
log_output = main_container.logs()
assert SECRET_QUOTE in log_output, "Secret not found in log output."
>>>>>>> a9d6c92ea3ca2760e4a18276d06c668058dd3670


# See #60
Expand All @@ -59,6 +92,7 @@ def test_container_count(dockerc):
# ), "RELEASE_TAG does not match the project version"


<<<<<<< HEAD
# See #60
# def test_log_version(version_container):
# """Verify the container outputs the correct version to the logs."""
Expand All @@ -83,3 +117,29 @@ def test_container_count(dockerc):
# assert (
# version_container.labels["org.opencontainers.image.version"] == project_version
# ), "Dockerfile version label does not match project version"
=======
def test_log_version(dockerc, version_container):
"""Verify the container outputs the correct version to the logs."""
# make sure container exited if running test isolated
dockerc.wait(version_container.id)
log_output = version_container.logs().strip()
pkg_vars = {}
with open(VERSION_FILE) as f:
exec(f.read(), pkg_vars) # nosec
project_version = pkg_vars["__version__"]
assert (
log_output == project_version
), f"Container version output to log does not match project version file {VERSION_FILE}"


def test_container_version_label_matches(version_container):
"""Verify the container version label is the correct version."""
pkg_vars = {}
with open(VERSION_FILE) as f:
exec(f.read(), pkg_vars) # nosec
project_version = pkg_vars["__version__"]
assert (
version_container.config.labels["org.opencontainers.image.version"]
== project_version
), "Dockerfile version label does not match project version"
>>>>>>> a9d6c92ea3ca2760e4a18276d06c668058dd3670

0 comments on commit ab98639

Please sign in to comment.