Initiales Halloween-Projekt

tests/test_web.py

@@ -0,0 +1,54 @@
+import asyncio
+import tempfile
+import unittest
+from pathlib import Path
+
+import httpx
+
+from halloween.web import create_app
+
+
+class WebTest(unittest.TestCase):
+    def test_demo_live_image_and_controls(self):
+        async def scenario():
+            with tempfile.TemporaryDirectory() as folder:
+                app = create_app(demo=True, state_dir=Path(folder))
+                app.state.camera.start()
+                try:
+                    transport = httpx.ASGITransport(app=app)
+                    async with httpx.AsyncClient(transport=transport, base_url="http://local") as client:
+                        page = await client.get("/")
+                        self.assertEqual(page.status_code, 200)
+                        self.assertIn("/stream.mjpg", page.text)
+                        self.assertIn("Aktoren nur simuliert", page.text)
+                        for _ in range(30):
+                            if (await client.get("/api/status")).json()["camera"]["ready"]:
+                                break
+                            await asyncio.sleep(0.05)
+                        image = await client.get("/snapshot.jpg")
+                        self.assertEqual(image.status_code, 200)
+                        self.assertTrue(image.content.startswith(b"\xff\xd8"))
+                        self.assertEqual((await client.post("/api/stop")).status_code, 403)
+                        headers = {"X-Halloween-UI": "1"}
+                        self.assertEqual((await client.post("/api/stop", headers=headers)).status_code, 200)
+                        self.assertTrue((await client.get("/api/status")).json()["controller"]["stopped"])
+                        self.assertEqual((await client.post("/api/reset", headers=headers)).status_code, 200)
+                        result = await client.post(
+                            "/api/mode", headers=headers, json={"mode": "sanft"}
+                        )
+                        self.assertEqual(result.status_code, 200)
+                        self.assertEqual((await client.get("/api/status")).json()["controller"]["mode"], "sanft")
+
+                    route = next(route for route in app.routes if route.path == "/stream.mjpg")
+                    response = await route.endpoint()
+                    first = await response.body_iterator.__anext__()
+                    self.assertTrue(first.startswith(b"--frame\r\nContent-Type: image/jpeg"))
+                    await response.body_iterator.aclose()
+                finally:
+                    app.state.camera.stop()
+
+        asyncio.run(scenario())
+
+
+if __name__ == "__main__":
+    unittest.main()