Add shared browser session bootstrap

2026-03-16 13:54:00 -04:00
parent 0f797d0a96
commit 7789c2e6ae
7 changed files with 741 additions and 32 deletions
--- a/tests/test_browser_session.py
+++ b/tests/test_browser_session.py
@@ -0,0 +1,62 @@
+import sqlite3
+import tempfile
+import unittest
+from pathlib import Path
+
+import browser_session
+import retailer_sessions
+
+
+class BrowserSessionTests(unittest.TestCase):
+    def test_read_firefox_ls_entries_reads_storage_from_copied_sqlite(self):
+        with tempfile.TemporaryDirectory() as tmpdir:
+            profile_dir = Path(tmpdir) / "abcd.default-release"
+            ls_dir = profile_dir / "storage" / "default" / "https+++www.costco.com" / "ls"
+            ls_dir.mkdir(parents=True)
+            db_path = ls_dir / "data.sqlite"
+
+            with sqlite3.connect(db_path) as connection:
+                connection.execute("CREATE TABLE data (key TEXT, value TEXT)")
+                connection.execute(
+                    "INSERT INTO data (key, value) VALUES (?, ?)",
+                    ("session", '{"costco":{"clientIdentifier":"481b1aec-aa3b-454b-b81b-48187e28f205"}}'),
+                )
+
+            entries = browser_session.read_firefox_storage_entries(
+                profile_dir,
+                origin_filters=["costco.com"],
+            )
+
+            self.assertEqual(1, len(entries))
+            self.assertEqual("https://www.costco.com", entries[0].origin)
+            self.assertEqual("session", entries[0].key)
+
+    def test_extract_costco_headers_from_storage_json(self):
+        entries = [
+            browser_session.StorageEntry(
+                origin="https://www.costco.com",
+                key="authState",
+                value=(
+                    '{"authorization":"Bearer header.payload.signature",'
+                    '"wcsClientId":"4900eb1f-0c10-4bd9-99c3-c59e6c1ecebf",'
+                    '"clientIdentifier":"481b1aec-aa3b-454b-b81b-48187e28f205"}'
+                ),
+                source="memory",
+            )
+        ]
+
+        headers = retailer_sessions.extract_costco_headers(entries)
+
+        self.assertEqual("Bearer header.payload.signature", headers["costco-x-authorization"])
+        self.assertEqual(
+            "4900eb1f-0c10-4bd9-99c3-c59e6c1ecebf",
+            headers["costco-x-wcs-clientId"],
+        )
+        self.assertEqual(
+            "481b1aec-aa3b-454b-b81b-48187e28f205",
+            headers["client-identifier"],
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()