mirror of
https://github.com/Such-Software/adaptive-multi-protocol-browser.git
synced 2026-10-17 14:31:46 +02:00
104 lines
3.9 KiB
Python
104 lines
3.9 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import replace
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from ampbrowser.config import AppConfig
|
|
from ampbrowser.desktop_shell import launch_desktop_shell
|
|
from ampbrowser.transports import TransportStatus
|
|
|
|
|
|
class DesktopShellTest(unittest.TestCase):
|
|
def test_shell_cancel_does_not_launch(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 tempfile.TemporaryDirectory() as tmp:
|
|
with patch("ampbrowser.plan.inspect_transport", return_value=status):
|
|
with patch("ampbrowser.desktop_shell.launch_open_plan") as launch:
|
|
result = launch_desktop_shell(
|
|
"http://example.onion/",
|
|
config=AppConfig(transport_modes={}),
|
|
root=Path(tmp),
|
|
prompt_driver=lambda open_plan: False,
|
|
)
|
|
|
|
self.assertEqual("cancelled", result.open_plan.status)
|
|
self.assertTrue(result.prompt_shown)
|
|
self.assertFalse(result.prompt_approved)
|
|
launch.assert_not_called()
|
|
|
|
def test_shell_approval_launches_approved_plan(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",
|
|
)
|
|
|
|
def fake_launch(open_plan, *, config: AppConfig, root: Path):
|
|
self.assertEqual("setup-approved", open_plan.status)
|
|
return replace(open_plan, status="launched", dry_run=False, browser_pid=1234)
|
|
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
with patch("ampbrowser.plan.inspect_transport", return_value=status):
|
|
with patch("ampbrowser.desktop_shell.launch_open_plan", side_effect=fake_launch) as launch:
|
|
result = launch_desktop_shell(
|
|
"http://example.onion/",
|
|
config=AppConfig(transport_modes={}),
|
|
root=Path(tmp),
|
|
prompt_driver=lambda open_plan: True,
|
|
)
|
|
|
|
self.assertEqual("launched", result.open_plan.status)
|
|
self.assertEqual(1234, result.open_plan.browser_pid)
|
|
self.assertTrue(result.prompt_shown)
|
|
self.assertTrue(result.prompt_approved)
|
|
launch.assert_called_once()
|
|
|
|
def test_shell_auto_approve_skips_prompt_driver(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",
|
|
)
|
|
|
|
def fake_launch(open_plan, *, config: AppConfig, root: Path):
|
|
return replace(open_plan, status="launched", dry_run=False, browser_pid=1234)
|
|
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
with patch("ampbrowser.plan.inspect_transport", return_value=status):
|
|
with patch("ampbrowser.desktop_shell.launch_open_plan", side_effect=fake_launch):
|
|
result = launch_desktop_shell(
|
|
"http://example.onion/",
|
|
config=AppConfig(transport_modes={}),
|
|
root=Path(tmp),
|
|
auto_approve=True,
|
|
prompt_driver=lambda open_plan: False,
|
|
)
|
|
|
|
self.assertEqual("launched", result.open_plan.status)
|
|
self.assertFalse(result.prompt_shown)
|
|
self.assertTrue(result.prompt_approved)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|