Generate daemon plan artifacts

2026-10-17 15:41:41 +02:00 · 2026-07-05 19:37:10 -04:00
parent e01d35132f
commit 1179646cc3
14 changed files with 395 additions and 8 deletions
@@ -0,0 +1,71 @@
+import tempfile
+import unittest
+from pathlib import Path
+
+from ampg.config import load_config
+from ampg.plan import plan_gateway, write_plan_artifacts
+
+
+class PlanArtifactsTest(unittest.TestCase):
+    def test_writes_tor_and_i2p_artifacts(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp)
+            source = root / "site"
+            source.mkdir()
+            (source / "index.html").write_text("<h1>Hi</h1>", encoding="utf-8")
+            config_path = root / "gateway.toml"
+            config_path.write_text(
+                """
+[[site]]
+id = "example"
+domain = "example.test"
+
+[site.source]
+kind = "static-html"
+path = "./site"
+
+[site.outputs]
+root = "./out"
+plan_root = "./plan"
+
+[site.protocols.tor]
+enabled = true
+renderer = "privacy-html"
+daemon = "tor"
+daemon_policy = "auto"
+local_port = 19080
+hidden_service_dir = "/var/lib/ampg/tor/example"
+
+[site.protocols.i2p]
+enabled = true
+renderer = "privacy-html"
+daemon = "i2pd"
+daemon_policy = "auto"
+local_port = 19081
+keys_file = "example-web.dat"
+tunnel_name = "example-web"
+""",
+                encoding="utf-8",
+            )
+
+            config = load_config(config_path)
+            lines = plan_gateway(config)
+            artifacts = write_plan_artifacts(config)
+
+            self.assertEqual(len(lines), 2)
+            self.assertEqual(len(artifacts), 4)
+
+            torrc = root / "plan/example/tor/torrc-snippet.conf"
+            tor_nginx = root / "plan/example/tor/nginx-loopback.conf"
+            i2p_tunnel = root / "plan/example/i2p/i2pd-tunnel.conf"
+            i2p_nginx = root / "plan/example/i2p/nginx-loopback.conf"
+
+            self.assertIn("HiddenServicePort 80 127.0.0.1:19080", torrc.read_text())
+            self.assertIn("listen 127.0.0.1:19080;", tor_nginx.read_text())
+            self.assertIn("[example-web]", i2p_tunnel.read_text())
+            self.assertIn("keys = example-web.dat", i2p_tunnel.read_text())
+            self.assertIn("listen 127.0.0.1:19081;", i2p_nginx.read_text())
+
+
+if __name__ == "__main__":
+    unittest.main()