Initial commit

tests/test_host_rate_limit.py

@@ -0,0 +1,40 @@
+from __future__ import annotations
+
+from django.core.cache import cache
+from django.test import override_settings
+
+from yksa_kit import host_rate_limit
+
+
+def test_host_from_url():
+    assert host_rate_limit.host_from_url("https://Db.SatNOGS.org/api/") == "db.satnogs.org"
+    assert host_rate_limit.host_from_url("not a url") == ""
+
+
+def test_development_never_blocks():
+    cache.clear()
+    host_rate_limit.acquire("example.com", per_minute=1)
+    host_rate_limit.acquire("example.com", per_minute=1)
+    assert cache.get("kit:host_rl:example.com") is None
+
+
+@override_settings(PRODUCTION=True)
+def test_production_counts_against_the_minute_window():
+    cache.clear()
+    host_rate_limit.acquire("example.com", per_minute=5)
+    assert cache.get("kit:host_rl:example.com") == 1
+
+
+@override_settings(PRODUCTION=True)
+def test_a_zero_budget_means_unlimited():
+    """per_minute=0 is how a source says 'no published limit', not 'no requests'."""
+    cache.clear()
+    host_rate_limit.acquire("example.com", per_minute=0)
+    assert cache.get("kit:host_rl:example.com") is None
+
+
+@override_settings(PRODUCTION=True)
+def test_the_wait_gives_up_rather_than_stalling_the_worker():
+    cache.clear()
+    cache.set("kit:host_rl:example.com", 99, timeout=60)
+    host_rate_limit.acquire("example.com", per_minute=1, wait_seconds=0.0)