mirror of
https://github.com/Such-Software/adaptive-multi-protocol-gateway.git
synced 2026-10-17 07:31:32 +02:00
116 lines
4.0 KiB
Python
116 lines
4.0 KiB
Python
import contextlib
|
|
import io
|
|
import json
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from ampg.build import build_gateway
|
|
from ampg.cli import main
|
|
from ampg.config import load_config
|
|
from ampg.preview import preview_endpoints, write_preview_fixture_manifests
|
|
|
|
|
|
class PreviewTest(unittest.TestCase):
|
|
def test_preview_manifest_rewrites_urls_to_loopback_and_preserves_published_metadata(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
config_path = _write_config(Path(tmp))
|
|
config = load_config(config_path)
|
|
build_gateway(config)
|
|
|
|
endpoints = preview_endpoints(config, base_port=19100)
|
|
self.assertEqual(["clearnet", "tor"], [endpoint.protocol for endpoint in endpoints])
|
|
self.assertEqual([19100, 19101], [endpoint.port for endpoint in endpoints])
|
|
self.assertEqual(["ready", "ready"], [endpoint.status for endpoint in endpoints])
|
|
|
|
results = write_preview_fixture_manifests(config, base_port=19100)
|
|
self.assertEqual(1, len(results))
|
|
self.assertEqual(4, results[0].fixture_count)
|
|
|
|
data = json.loads(results[0].path.read_text(encoding="utf-8"))
|
|
self.assertEqual("preview", data["mode"])
|
|
by_protocol_path = {
|
|
(fixture["protocol"], fixture.get("route", {}).get("fixture_path", "/")): fixture
|
|
for fixture in data["fixtures"]
|
|
}
|
|
|
|
tor_root = by_protocol_path[("tor", "/")]
|
|
self.assertEqual("http://127.0.0.1:19101/", tor_root["url"])
|
|
self.assertEqual({"transport": "clearnet", "profile": "clearnet"}, tor_root["checks"])
|
|
self.assertEqual("http://preview.onion/", tor_root["published"]["url"])
|
|
self.assertEqual(
|
|
{"transport": "tor", "profile": "tor", "isolation": "transport-profile"},
|
|
tor_root["published"]["checks"],
|
|
)
|
|
self.assertEqual("preview", tor_root["address_status"])
|
|
self.assertEqual("ready", tor_root["preview"]["status"])
|
|
|
|
tor_play = by_protocol_path[("tor", "/play/")]
|
|
self.assertEqual("http://127.0.0.1:19101/play/", tor_play["url"])
|
|
|
|
def test_preview_manifest_cli(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
config_path = _write_config(Path(tmp))
|
|
config = load_config(config_path)
|
|
build_gateway(config)
|
|
stdout = io.StringIO()
|
|
|
|
with contextlib.redirect_stdout(stdout):
|
|
status = main(["--config", str(config_path), "preview", "manifest", "--base-port", "19200"])
|
|
self.assertTrue((config.sites[0].outputs.root / "ampg-preview-fixture-manifest.json").exists())
|
|
|
|
output = stdout.getvalue()
|
|
self.assertEqual(0, status)
|
|
self.assertIn("AMPG_PREVIEW_ENDPOINT site=preview protocol=clearnet", output)
|
|
self.assertIn("AMPG_PREVIEW_ENDPOINT site=preview protocol=tor", output)
|
|
self.assertIn("AMPG_PREVIEW_MANIFEST site=preview", output)
|
|
|
|
def test_preview_endpoints_report_missing_outputs(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
config = load_config(_write_config(Path(tmp)))
|
|
|
|
endpoints = preview_endpoints(config)
|
|
|
|
self.assertEqual(["missing-output", "missing-output"], [endpoint.status for endpoint in endpoints])
|
|
|
|
|
|
def _write_config(root: Path) -> Path:
|
|
source = root / "site"
|
|
source.mkdir()
|
|
(source / "index.html").write_text("<h1>Preview</h1>", encoding="utf-8")
|
|
config_path = root / "gateway.toml"
|
|
config_path.write_text(
|
|
"""
|
|
[[site]]
|
|
id = "preview"
|
|
domain = "preview.test"
|
|
|
|
[site.source]
|
|
kind = "static-html"
|
|
path = "./site"
|
|
canonical_url = "https://preview.test"
|
|
|
|
[site.outputs]
|
|
root = "./out"
|
|
plan_root = "./plan"
|
|
|
|
[[site.interactions.route]]
|
|
match = "/play/*"
|
|
tier = "interactive-lite"
|
|
|
|
[site.protocols.clearnet]
|
|
enabled = true
|
|
renderer = "clearnet"
|
|
|
|
[site.protocols.tor]
|
|
enabled = true
|
|
renderer = "privacy-html"
|
|
""",
|
|
encoding="utf-8",
|
|
)
|
|
return config_path
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|