62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from containerizer import containerize
|
|
|
|
|
|
class ContainerizerTests(unittest.TestCase):
|
|
def service(self, health=True):
|
|
temp = tempfile.TemporaryDirectory()
|
|
root = Path(temp.name)
|
|
(root / "app").mkdir()
|
|
route = '@app.get("/health")\n' if health else ""
|
|
(root / "app/main.py").write_text(
|
|
"from fastapi import FastAPI\napp = FastAPI()\n" + route)
|
|
(root / "requirements.lock").write_text("fastapi==0.115.6\nuvicorn==0.34.0\n")
|
|
return temp, root
|
|
|
|
def payload(self, health="/health"):
|
|
return {
|
|
"service_specification": {"container": {
|
|
"startup_module": "app.main:app", "port": 8000,
|
|
"health_endpoint": health}},
|
|
"target_management_implementation": {"preserved": 1},
|
|
"endpoint_check_implementation": {"preserved": 2},
|
|
"current_status_implementation": {"preserved": 3},
|
|
}
|
|
|
|
def test_writes_artifacts_and_preserves_contract(self):
|
|
temp, root = self.service()
|
|
self.addCleanup(temp.cleanup)
|
|
output = containerize(self.payload(), root)
|
|
self.assertEqual(output["target_management_implementation"], {"preserved": 1})
|
|
self.assertEqual(set(output["container_artifacts"]["files"]),
|
|
{"Dockerfile", "compose.yaml", ".dockerignore"})
|
|
self.assertIn("USER 10001:10001", (root / "Dockerfile").read_text())
|
|
self.assertIn('uvicorn", "app.main:app', (root / "Dockerfile").read_text())
|
|
self.assertIn("healthcheck:", (root / "compose.yaml").read_text())
|
|
|
|
def test_omits_healthcheck_without_matching_configuration(self):
|
|
temp, root = self.service(health=False)
|
|
self.addCleanup(temp.cleanup)
|
|
containerize(self.payload(health=None), root)
|
|
self.assertNotIn("healthcheck:", (root / "compose.yaml").read_text())
|
|
|
|
def test_rejects_healthcheck_not_present_in_app(self):
|
|
temp, root = self.service(health=False)
|
|
self.addCleanup(temp.cleanup)
|
|
with self.assertRaisesRegex(ValueError, "not found"):
|
|
containerize(self.payload(), root)
|
|
|
|
def test_rejects_unpinned_dependencies(self):
|
|
temp, root = self.service()
|
|
self.addCleanup(temp.cleanup)
|
|
(root / "requirements.lock").write_text("fastapi>=0.100\n")
|
|
with self.assertRaisesRegex(ValueError, "exact"):
|
|
containerize(self.payload(), root)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|