1 files changed, 73 insertions, 0 deletions
diff --git a/tests/test_rdap.py b/tests/test_rdap.py
index 681e2bb..3fc2c65 100644
--- a/tests/test_rdap.py
+++ b/tests/test_rdap.py
@@ -52,5 +52,78 @@ class RedirectPolicy(unittest.TestCase):
         self.assertIsNotNone(result)
 
 
+import json
+import tempfile
+import time
+from pathlib import Path
+
+
+class Bootstrap(unittest.TestCase):
+    def setUp(self):
+        self.tmp = tempfile.TemporaryDirectory()
+        self.cache = Path(self.tmp.name)
+        self.addCleanup(self.tmp.cleanup)
+
+    def test_a_missing_file_is_fetched_and_cached(self):
+        calls = []
+
+        def fetch(url):
+            calls.append(url)
+            return {"services": []}
+
+        data = rdap.bootstrap("ipv4", cache_root=self.cache, fetch=fetch)
+
+        self.assertEqual(data, {"services": []})
+        self.assertEqual(calls, ["https://data.iana.org/rdap/ipv4.json"])
+        self.assertTrue((self.cache / "ipv4.json").exists())
+
+    def test_a_fresh_cache_is_not_refetched(self):
+        (self.cache / "ipv4.json").write_text(json.dumps({"services": ["cached"]}))
+
+        def fetch(url):
+            raise AssertionError(f"should not have fetched {url}")
+
+        data = rdap.bootstrap("ipv4", cache_root=self.cache, fetch=fetch)
+        self.assertEqual(data, {"services": ["cached"]})
+
+    def test_a_stale_cache_is_refetched(self):
+        path = self.cache / "ipv4.json"
+        path.write_text(json.dumps({"services": ["old"]}))
+        old = time.time() - (rdap._BOOTSTRAP_TTL + 60)
+        import os
+        os.utime(path, (old, old))
+
+        data = rdap.bootstrap(
+            "ipv4", cache_root=self.cache, fetch=lambda url: {"services": ["new"]}
+        )
+        self.assertEqual(data, {"services": ["new"]})
+
+    def test_a_failed_refetch_falls_back_to_the_stale_copy(self):
+        """Losing IANA must not stop the user filing a report.
+
+        Last week's map is almost certainly still correct, and a stale
+        bootstrap fails safe: the worst case is querying a server that has
+        moved, which misses and reads as no contact.
+        """
+        path = self.cache / "ipv4.json"
+        path.write_text(json.dumps({"services": ["old"]}))
+        old = time.time() - (rdap._BOOTSTRAP_TTL + 60)
+        import os
+        os.utime(path, (old, old))
+
+        def fetch(url):
+            raise OSError("network is unreachable")
+
+        data = rdap.bootstrap("ipv4", cache_root=self.cache, fetch=fetch)
+        self.assertEqual(data, {"services": ["old"]})
+
+    def test_a_failed_fetch_with_no_cache_raises(self):
+        def fetch(url):
+            raise OSError("network is unreachable")
+
+        with self.assertRaises(rdap.BootstrapUnavailable):
+            rdap.bootstrap("ipv4", cache_root=self.cache, fetch=fetch)
+
+
 if __name__ == "__main__":
     unittest.main()