klemek/stapler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 10 Wiki Activity

feat: X-Host-Only

Browse Source
This commit is contained in:
klemek
2026-04-21 22:51:35 +02:00
parent 18c718d24a
commit c0c281e25c
11 changed files with 229 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_registry.py
+26
View File
@@ -28,20 +28,24 @@ class TestRegistry(BaseTestCase):
self.data_dir.get_file,
[
["test_1", Registry.HOST_FILE],
["test_1", Registry.HOST_ONLY_FILE],
["test_1", Registry.TOKEN_FILE],
["test_1", Registry.REDIRECT_FILE],
["test_1", Registry.PROXY_FILE],
["test_2", Registry.HOST_FILE],
["test_2", Registry.HOST_ONLY_FILE],
["test_2", Registry.TOKEN_FILE],
["test_2", Registry.REDIRECT_FILE],
["test_2", Registry.PROXY_FILE],
],
[
None,
"test_1_host",
"test_1_token",
None,
None,
None,
None,
"test_2_token",
"test_2_redirect",
None,
@@ -58,6 +62,7 @@ class TestRegistry(BaseTestCase):
"test_1",
True, # noqa: FBT003
"test_1_host",
True, # noqa: FBT003
"test_1_token",
None,
),
@@ -68,6 +73,7 @@ class TestRegistry(BaseTestCase):
"test_2",
False, # noqa: FBT003
None,
False, # noqa: FBT003
"test_2_token",
"test_2_redirect",
),
@@ -98,10 +104,30 @@ class TestRegistry(BaseTestCase):
self.mock_call(
self.data_dir.set_file, ["test_1", Registry.HOST_FILE, "new_value"]
),
self.mock_call(
self.data_dir.remove_file, ["test_1", Registry.HOST_ONLY_FILE]
),
self.seal_mocks(),
):
self.registry.set_host("test_1", "new_value")
self.assertEqual(self.registry.pages["test_1"].host, "new_value")
assert not self.registry.pages["test_1"].host_only
def test_set_host_only(self) -> None:
self.registry.pages["test_1"] = Page(
"test_1",
host="test_1_host",
)
with (
self.mock_call(
self.data_dir.set_file, ["test_1", Registry.HOST_ONLY_FILE, "new_value"]
),
self.mock_call(self.data_dir.remove_file, ["test_1", Registry.HOST_FILE]),
self.seal_mocks(),
):
self.registry.set_host_only("test_1", "new_value")
self.assertEqual(self.registry.pages["test_1"].host, "new_value")
assert self.registry.pages["test_1"].host_only
def test_set_token_hash(self) -> None:
self.registry.pages["test_1"] = Page(