Add route-aware setup helper

tests/test_launch.py

@@ -6,7 +6,7 @@ import tempfile
 from pathlib import Path
 
 from ampbrowser.config import AppConfig
-from ampbrowser.launch import execute_open, prepare_open
+from ampbrowser.launch import RouteHelperLaunch, execute_open, prepare_open
 from ampbrowser.transports import TransportStatus
 
 
@@ -138,10 +138,15 @@ class PrepareOpenTest(unittest.TestCase):
             self.assertIn('user_pref("network.proxy.type", 2);', plan.launch_spec.prefs)
             self.assertIn("__AMPB_ROUTE_AWARE_PAC_URL__", "\n".join(plan.launch_spec.prefs))
 
-            with patch("ampbrowser.launch.subprocess.Popen") as popen:
-                launched = execute_open(plan, root=root)
+            helper = RouteHelperLaunch("started", endpoint="http://127.0.0.1:44001/", token="test-token", pid=4321)
+            with patch("ampbrowser.launch._start_route_helper", return_value=helper):
+                with patch("ampbrowser.launch.subprocess.Popen") as popen:
+                    launched = execute_open(plan, root=root)
 
             self.assertEqual("launched", launched.status)
+            self.assertEqual("started", launched.route_helper_status)
+            self.assertEqual("http://127.0.0.1:44001/", launched.route_helper_endpoint)
+            self.assertEqual(4321, launched.route_helper_pid)
             popen.assert_called_once_with(
                 (str(runtime), "-no-remote", "-profile", ".ampb/profiles/route-aware", "https://ampgateway.site/"),
                 cwd=str(root),
@@ -156,6 +161,14 @@ class PrepareOpenTest(unittest.TestCase):
             self.assertIn('hasSuffix(h, ".i2p")', pac)
             self.assertIn('return "PROXY 127.0.0.1:4444";', pac)
             self.assertIn('return "DIRECT";', pac)
+            extension = profile / "extensions/ampb-route-helper@such.software"
+            manifest = (extension / "manifest.json").read_text(encoding="utf-8")
+            background = (extension / "background.js").read_text(encoding="utf-8")
+            setup = (extension / "setup.html").read_text(encoding="utf-8")
+            self.assertIn("webNavigation", manifest)
+            self.assertIn("http://127.0.0.1:44001/", background)
+            self.assertIn("test-token", background)
+            self.assertIn("Set Up Transport", setup)
 
     def test_execute_open_blocks_when_runtime_is_missing(self) -> None:
         config = AppConfig(runtime_path="/missing/ampb/firefox", transport_modes={})

tests/test_route_helper.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+from pathlib import Path
+import tempfile
+import unittest
+from unittest.mock import patch
+
+from ampbrowser.config import AppConfig
+from ampbrowser.route_helper import handle_helper_message
+from ampbrowser.transport_manager import ManagedTransportResult
+
+
+class RouteHelperTest(unittest.TestCase):
+    def test_status_routes_url_to_transport(self) -> None:
+        result = ManagedTransportResult(
+            transport="tor",
+            provider="arti",
+            status="ready",
+            endpoint="socks5://127.0.0.1:9050",
+            owned=True,
+            pid=1234,
+            state_dir=".ampb/transports/tor",
+            command=("/tmp/arti", "proxy"),
+            message="using running AMPB-owned tor",
+            provider_source="bundled-sidecar",
+        )
+
+        with tempfile.TemporaryDirectory() as tmp:
+            with patch("ampbrowser.route_helper.transport_status", return_value=result) as status:
+                response = handle_helper_message(
+                    {"action": "status", "url": "http://example.onion/"},
+                    root=Path(tmp),
+                    config=AppConfig(transport_modes={}),
+                )
+
+        self.assertTrue(response["ok"])
+        self.assertTrue(response["ready"])
+        self.assertEqual("tor", response["transport"])
+        self.assertEqual("bundled-sidecar", response["provider_source"])
+        status.assert_called_once()
+
+    def test_ensure_starts_i2p_transport(self) -> None:
+        result = ManagedTransportResult(
+            transport="i2p",
+            provider="i2pd",
+            status="started",
+            endpoint="http://127.0.0.1:4444",
+            owned=True,
+            pid=1234,
+            state_dir=".ampb/transports/i2p",
+            command=("/opt/i2pd",),
+            message="started managed i2pd transport",
+            provider_source="system-package",
+        )
+
+        with tempfile.TemporaryDirectory() as tmp:
+            with patch("ampbrowser.route_helper.ensure_transport_ready", return_value=result) as ensure:
+                response = handle_helper_message(
+                    {"action": "ensure", "transport": "i2p", "url": "http://example.b32.i2p/"},
+                    root=Path(tmp),
+                    config=AppConfig(transport_modes={}),
+                )
+
+        self.assertTrue(response["ok"])
+        self.assertTrue(response["ready"])
+        self.assertEqual("started", response["status"])
+        ensure.assert_called_once()
+
+    def test_blocks_unsupported_routes(self) -> None:
+        with tempfile.TemporaryDirectory() as tmp:
+            response = handle_helper_message(
+                {"action": "ensure", "url": "https://example.com/"},
+                root=Path(tmp),
+                config=AppConfig(transport_modes={}),
+            )
+
+        self.assertFalse(response["ok"])
+        self.assertEqual("unsupported-transport", response["status"])
+
+
+if __name__ == "__main__":
+    unittest.main()