hazza/docker-stacks
1
0
Fork 0
mirror of https://github.com/jupyter/docker-stacks.git synced 2025-10-08 10:34:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add some typing to tests

Browse Source
This commit is contained in:
Ayaz Salikhov
2022-01-18 19:13:17 +03:00
parent ee77b7831e
commit 2a1316c9ac
24 changed files with 211 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tagging/docker_runner.py
View File

@@ -3,8 +3,10 @@
import docker
import logging
from conftest import TrackedContainer
logger = logging.getLogger(__name__)
LOGGER = logging.getLogger(__name__)
class DockerRunner:
@@ -20,27 +22,29 @@ class DockerRunner:
self.docker_client = docker_client
def __enter__(self):
logger.info(f"Creating container for image {self.image_name} ...")
LOGGER.info(f"Creating container for image {self.image_name} ...")
self.container = self.docker_client.containers.run(
image=self.image_name,
command=self.command,
detach=True,
)
logger.info(f"Container {self.container.name} created")
LOGGER.info(f"Container {self.container.name} created")
return self.container
def __exit__(self, exc_type, exc_value, traceback):
logger.info(f"Removing container {self.container.name} ...")
LOGGER.info(f"Removing container {self.container.name} ...")
if self.container:
self.container.remove(force=True)
logger.info(f"Container {self.container.name} removed")
LOGGER.info(f"Container {self.container.name} removed")
@staticmethod
def run_simple_command(container, cmd: str, print_result: bool = True):
logger.info(f"Running cmd: '{cmd}' on container: {container}")
def run_simple_command(
container: TrackedContainer, cmd: str, print_result: bool = True
):
LOGGER.info(f"Running cmd: '{cmd}' on container: {container}")
out = container.exec_run(cmd)
result = out.output.decode("utf-8").rstrip()
if print_result:
logger.info(f"Command result: {result}")
LOGGER.info(f"Command result: {result}")
assert out.exit_code == 0, f"Command: {cmd} failed"
return result