Refactor tests

docker-images/python3.6.dockerfile

@@ -2,7 +2,7 @@ FROM python:3.6
 
 LABEL maintainer="Sebastian Ramirez <[email protected]>"
 
-RUN pip --no-cache-dir install uvicorn gunicorn
+RUN pip install --no-cache-dir uvicorn gunicorn
 
 COPY ./start.sh /start.sh
 RUN chmod +x /start.sh

docker-images/python3.7.dockerfile

@@ -2,7 +2,7 @@ FROM python:3.7
 
 LABEL maintainer="Sebastian Ramirez <[email protected]>"
 
-RUN pip --no-cache-dir install uvicorn gunicorn
+RUN pip install --no-cache-dir uvicorn gunicorn
 
 COPY ./start.sh /start.sh
 RUN chmod +x /start.sh

mypy.ini

@@ -0,0 +1,3 @@
+[mypy]
+disallow_untyped_defs = True
+ignore_missing_imports = True

pyproject.toml

@@ -11,6 +11,10 @@ docker = "^4.2.0"
 pytest = "^5.4.1"
 
 [tool.poetry.dev-dependencies]
+black = "^19.10b0"
+isort = "^4.3.21"
+autoflake = "^1.3.1"
+mypy = "^0.770"
 
 [build-system]
 requires = ["poetry>=0.12"]

scripts/format-imports.sh

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+set -x
+
+# Sort imports one per line, so autoflake can remove unused imports
+isort --recursive  --force-single-line-imports --apply ./
+sh ./scripts/format.sh

scripts/lint.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+
+set -e
+set -x
+
+mypy ./
+black ./ --check
+isort --multi-line=3 --trailing-comma --force-grid-wrap=0 --combine-as --line-width 88 --recursive --check-only

tests/.gitignore

@@ -0,0 +1 @@
+Dockerfile

tests/test_01_main/test_defaults.py

@@ -3,6 +3,7 @@
 
 import docker
 import requests
+from docker.client import DockerClient
 
 from ..utils import (
     CONTAINER_NAME,
@@ -15,7 +16,7 @@
 client = docker.from_env()
 
 
-def verify_container(container, response_text):
+def verify_container(container: DockerClient, response_text: str) -> None:
     config_data = get_config(container)
     assert config_data["workers_per_core"] == 1
     assert config_data["host"] == "0.0.0.0"
@@ -33,7 +34,7 @@ def verify_container(container, response_text):
     assert response.text == response_text
 
 
-def test_defaults():
+def test_defaults() -> None:
     name = os.getenv("NAME")
     image = f"tiangolo/uvicorn-gunicorn:{name}"
     response_text = get_response_text1()

tests/test_01_main/test_env_vars_1.py

@@ -3,6 +3,7 @@
 
 import docker
 import requests
+from docker.client import DockerClient
 
 from ..utils import (
     CONTAINER_NAME,
@@ -15,7 +16,7 @@
 client = docker.from_env()
 
 
-def verify_container(container, response_text):
+def verify_container(container: DockerClient, response_text: str) -> None:
     config_data = get_config(container)
     assert config_data["workers_per_core"] == 2
     assert config_data["host"] == "0.0.0.0"
@@ -32,7 +33,7 @@ def verify_container(container, response_text):
     assert response.text == response_text
 
 
-def test_env_vars_1():
+def test_env_vars_1() -> None:
     name = os.getenv("NAME")
     image = f"tiangolo/uvicorn-gunicorn:{name}"
     response_text = get_response_text1()

tests/test_01_main/test_env_vars_2.py

@@ -2,6 +2,7 @@
 import time
 
 import docker
+from docker.models.containers import Container
 
 from ..utils import (
     CONTAINER_NAME,
@@ -14,7 +15,7 @@
 client = docker.from_env()
 
 
-def verify_container(container):
+def verify_container(container: Container) -> None:
     process_names = get_process_names(container)
     config_data = get_config(container)
     assert config_data["workers"] == 1
@@ -31,7 +32,7 @@ def verify_container(container):
     )
 
 
-def test_env_vars_2():
+def test_env_vars_2() -> None:
     name = os.getenv("NAME")
     image = f"tiangolo/uvicorn-gunicorn:{name}"
     sleep_time = int(os.getenv("SLEEP_TIME", 1))

tests/test_01_main/test_env_vars_3.py

@@ -3,6 +3,7 @@
 
 import docker
 import requests
+from docker.client import DockerClient
 
 from ..utils import (
     CONTAINER_NAME,
@@ -15,7 +16,7 @@
 client = docker.from_env()
 
 
-def verify_container(container, response_text):
+def verify_container(container: DockerClient, response_text: str) -> None:
     config_data = get_config(container)
     assert config_data["host"] == "127.0.0.1"
     assert config_data["port"] == "9000"
@@ -30,7 +31,7 @@ def verify_container(container, response_text):
     assert response.text == response_text
 
 
-def test_env_bind():
+def test_env_bind() -> None:
     name = os.getenv("NAME")
     image = f"tiangolo/uvicorn-gunicorn:{name}"
     response_text = get_response_text1()

tests/test_02_app/test_custom_app.py

@@ -1,14 +1,17 @@
 import os
 import time
-from pathlib import Path, PurePath
+from pathlib import Path
+from typing import Dict
 
 import docker
 import pytest
 import requests
+from docker.client import DockerClient
 
 from ..utils import (
     CONTAINER_NAME,
     IMAGE_NAME,
+    generate_dockerfile_content,
     get_config,
     get_logs,
     get_response_text2,
@@ -18,7 +21,9 @@
 client = docker.from_env()
 
 
-def verify_container(container, response_text, prestart_str):
+def verify_container(
+    container: DockerClient, response_text: str, prestart_str: str
+) -> None:
     config_data = get_config(container)
     assert config_data["workers_per_core"] == 1
     assert config_data["host"] == "0.0.0.0"
@@ -60,14 +65,17 @@ def verify_container(container, response_text, prestart_str):
         ),
     ],
 )
-def test_custom_app(environment, prestart_str):
-    name = os.getenv("NAME")
-    dockerfile = f"{name}.dockerfile"
+def test_custom_app(environment: Dict[str, str], prestart_str: str) -> None:
+    name = os.getenv("NAME", "")
+    dockerfile_content = generate_dockerfile_content(name)
+    dockerfile = "Dockerfile"
     response_text = get_response_text2()
     sleep_time = int(os.getenv("SLEEP_TIME", 1))
     remove_previous_container(client)
-    test_path: PurePath = Path(__file__)
+    test_path = Path(__file__)
     path = test_path.parent / "custom_app"
+    dockerfile_path = path / dockerfile
+    dockerfile_path.write_text(dockerfile_content)
     client.images.build(path=str(path), dockerfile=dockerfile, tag=IMAGE_NAME)
     container = client.containers.run(
         IMAGE_NAME,

tests/test_02_app/test_package_app.py

@@ -1,13 +1,15 @@
 import os
 import time
-from pathlib import Path, PurePath
+from pathlib import Path
 
 import docker
 import requests
+from docker.client import DockerClient
 
 from ..utils import (
     CONTAINER_NAME,
     IMAGE_NAME,
+    generate_dockerfile_content,
     get_config,
     get_logs,
     get_response_text2,
@@ -17,7 +19,7 @@
 client = docker.from_env()
 
 
-def verify_container(container, response_text):
+def verify_container(container: DockerClient, response_text: str) -> None:
     config_data = get_config(container)
     assert config_data["workers_per_core"] == 1
     assert config_data["host"] == "0.0.0.0"
@@ -35,14 +37,17 @@ def verify_container(container, response_text):
     assert response.text == response_text
 
 
-def test_package_app():
-    name = os.getenv("NAME")
-    dockerfile = f"{name}.dockerfile"
+def test_package_app() -> None:
+    name = os.getenv("NAME", "")
+    dockerfile_content = generate_dockerfile_content(name)
+    dockerfile = "Dockerfile"
     response_text = get_response_text2()
     sleep_time = int(os.getenv("SLEEP_TIME", 1))
     remove_previous_container(client)
-    test_path: PurePath = Path(__file__)
+    test_path = Path(__file__)
     path = test_path.parent / "package_app"
+    dockerfile_path = path / dockerfile
+    dockerfile_path.write_text(dockerfile_content)
     client.images.build(path=str(path), dockerfile=dockerfile, tag=IMAGE_NAME)
     container = client.containers.run(
         IMAGE_NAME, name=CONTAINER_NAME, ports={"80": "8000"}, detach=True