mirror of
https://github.com/Such-Software/adaptive-multi-protocol-browser.git
synced 2026-10-17 14:31:46 +02:00
95 lines
3.3 KiB
Python
95 lines
3.3 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from ampbrowser.config import AppConfig
|
|
from ampbrowser.plan import plan_url
|
|
from ampbrowser.transports import TransportStatus
|
|
|
|
|
|
class PlanUrlTest(unittest.TestCase):
|
|
def test_gemini_uses_builtin_transport(self) -> None:
|
|
plan = plan_url("gemini://wownero.org/")
|
|
|
|
self.assertEqual("gemini", plan.route.transport)
|
|
self.assertEqual("adopt existing transport", plan.action)
|
|
self.assertIsNotNone(plan.status)
|
|
self.assertTrue(plan.status.adoptable)
|
|
|
|
def test_clearnet_opens_with_clearnet_profile(self) -> None:
|
|
plan = plan_url("wownero.org")
|
|
|
|
self.assertEqual("https://wownero.org", plan.route.normalized)
|
|
self.assertEqual("clearnet", plan.route.transport)
|
|
self.assertEqual("open with clearnet profile", plan.action)
|
|
self.assertFalse(plan.requires_consent)
|
|
|
|
def test_missing_transport_prompts_before_install_or_start(self) -> None:
|
|
status = TransportStatus(
|
|
transport="i2p",
|
|
installed=False,
|
|
running=False,
|
|
endpoint="http://127.0.0.1:4444",
|
|
adoptable=False,
|
|
manage_supported=True,
|
|
note="I2P HTTP proxy",
|
|
)
|
|
|
|
with patch("ampbrowser.plan.inspect_transport", return_value=status):
|
|
plan = plan_url("http://example.b32.i2p/")
|
|
|
|
self.assertEqual("prompt to install and start managed transport", plan.action)
|
|
self.assertTrue(plan.requires_consent)
|
|
self.assertIn("install and start", plan.prompt)
|
|
|
|
def test_stopped_transport_prompts_before_start(self) -> None:
|
|
status = TransportStatus(
|
|
transport="tor",
|
|
installed=True,
|
|
running=False,
|
|
endpoint="socks5://127.0.0.1:9050",
|
|
adoptable=False,
|
|
manage_supported=True,
|
|
note="Tor SOCKS proxy",
|
|
)
|
|
|
|
with patch("ampbrowser.plan.inspect_transport", return_value=status):
|
|
plan = plan_url("http://example.onion/")
|
|
|
|
self.assertEqual("prompt to start managed transport", plan.action)
|
|
self.assertTrue(plan.requires_consent)
|
|
self.assertIn("can start", plan.prompt)
|
|
|
|
def test_disabled_transport_is_blocked_by_policy(self) -> None:
|
|
config = AppConfig(transport_modes={"i2p": "disabled"})
|
|
|
|
plan = plan_url("http://example.b32.i2p/", config=config)
|
|
|
|
self.assertEqual("disabled", plan.policy_mode)
|
|
self.assertEqual("blocked by policy: i2p disabled", plan.action)
|
|
self.assertFalse(plan.requires_consent)
|
|
|
|
def test_adopt_only_transport_does_not_prompt_to_start(self) -> None:
|
|
status = TransportStatus(
|
|
transport="tor",
|
|
installed=True,
|
|
running=False,
|
|
endpoint="socks5://127.0.0.1:9050",
|
|
adoptable=False,
|
|
manage_supported=True,
|
|
note="Tor SOCKS proxy",
|
|
)
|
|
config = AppConfig(transport_modes={"tor": "adopt"})
|
|
|
|
with patch("ampbrowser.plan.inspect_transport", return_value=status):
|
|
plan = plan_url("http://example.onion/", config=config)
|
|
|
|
self.assertEqual("adopt", plan.policy_mode)
|
|
self.assertEqual("blocked by policy: tor mode is adopt", plan.action)
|
|
self.assertFalse(plan.requires_consent)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|