feat: ds
This commit is contained in:
@@ -81,6 +81,19 @@ class BuildTest(unittest.TestCase):
|
||||
self.assertEqual(ep.origin, "https://lab.example:8443")
|
||||
self.assertEqual(ep.url, "https://lab.example:8443/next-chain")
|
||||
|
||||
def test_page_apis_use_location_origin(self) -> None:
|
||||
root = TOOLS.parent / "source"
|
||||
boot = (root / "boot.js").read_text(encoding="utf-8")
|
||||
loader = (root / "rce_loader.js").read_text(encoding="utf-8")
|
||||
worker = (root / "rce_worker_18.5.js").read_text(encoding="utf-8")
|
||||
self.assertIn("function pageOrigin()", boot)
|
||||
self.assertIn("function apiBase() {\n return pageOrigin();", boot)
|
||||
self.assertNotIn("window.__LAB_EXFIL__", boot.split("function apiBase()")[1].split("function applyExfil")[0])
|
||||
self.assertIn("location.origin", loader.split("function labApiBase()")[1].split("function resolveLabDeviceUUID")[0])
|
||||
self.assertNotIn("__LAB_EXFIL__", loader.split("function labApiBase()")[1].split("function resolveLabDeviceUUID")[0])
|
||||
self.assertIn("location.origin", worker.split("function labC2LogUrl")[1].split("function print")[0])
|
||||
self.assertNotIn("__labExfilUrl", worker.split("function labC2LogUrl")[1].split("function print")[0])
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user