mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-08 11:46:35 +00:00
kevinveenbirkenbachandClaude Opus 5.5
08fba6b3ef
network_url moves a URL into the network of the vhost being rendered, network_urls lists it once per network sibling with that sibling's scheme, and network_redirects fans a redirect source out to its siblings while the target stays canonical. Consumers that hard-code the canonical origin can now emit the sibling a request actually arrives on. Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
import unittest
|
|
import unittest.mock as mock
|
|
|
|
from ansible.errors import AnsibleError
|
|
|
|
import plugins.lookup.network_url as network_url_mod
|
|
from utils.domains.default_primary import default_domain_primary
|
|
|
|
PRIMARY = default_domain_primary()
|
|
ONION = "abcdefghijklmnopqrstuvwxyz234567abcdefghijklmnopqrstuvw.onion"
|
|
SIBLINGS = {f"file.{PRIMARY}": [f"file.{PRIMARY}", f"file.{ONION}"]}
|
|
BASES = {
|
|
f"file.{PRIMARY}": f"https://file.{PRIMARY}/",
|
|
f"file.{ONION}": f"http://file.{ONION}/",
|
|
}
|
|
|
|
|
|
def _loader():
|
|
def _get(name, **_k):
|
|
if name == "network_siblings":
|
|
return mock.MagicMock(
|
|
run=lambda terms, **_kw: [SIBLINGS.get(terms[0], [terms[0]])]
|
|
)
|
|
return mock.MagicMock(run=lambda terms, **_kw: [BASES[terms[0]]])
|
|
|
|
loader = mock.MagicMock()
|
|
loader.get.side_effect = _get
|
|
return mock.patch.object(network_url_mod, "lookup_loader", loader)
|
|
|
|
|
|
class TestNetworkUrl(unittest.TestCase):
|
|
def _run(self, url, variables):
|
|
lookup = network_url_mod.LookupModule()
|
|
lookup._loader = mock.MagicMock()
|
|
with _loader():
|
|
return lookup.run([url], variables=variables)[0]
|
|
|
|
def test_an_onion_vhost_gets_the_onion_sibling_with_its_scheme(self):
|
|
self.assertEqual(
|
|
self._run(
|
|
f"https://file.{PRIMARY}/assets/img/logo.png?v=1",
|
|
{"domain": f"dashboard.{ONION}"},
|
|
),
|
|
f"http://file.{ONION}/assets/img/logo.png?v=1",
|
|
)
|
|
|
|
def test_a_clearnet_vhost_keeps_the_url(self):
|
|
url = f"https://file.{PRIMARY}/assets/img/logo.png"
|
|
self.assertEqual(self._run(url, {"domain": f"dashboard.{PRIMARY}"}), url)
|
|
|
|
def test_a_host_the_node_does_not_serve_stays_unchanged(self):
|
|
url = "https://example.org/logo.png"
|
|
self.assertEqual(self._run(url, {"domain": f"dashboard.{ONION}"}), url)
|
|
|
|
def test_a_missing_vhost_fails_loudly(self):
|
|
with self.assertRaises(AnsibleError):
|
|
self._run(f"https://file.{PRIMARY}/logo.png", {})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|