Upload 5 files

tests/conftest.py

@@ -0,0 +1,7 @@
+import sys
+from pathlib import Path
+
+# Ensure "core" can be imported when tests run from repo root.
+ROOT = Path(__file__).resolve().parents[1]
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))

tests/test_cdc.py

@@ -0,0 +1,17 @@
+from core.sources import cdc
+
+
+def test_cdc_cas_exact_match():
+    matches = cdc.search("67-64-1")
+    assert any(m.get("name") == "Acetone" for m in matches)
+
+
+def test_cdc_name_substring_match():
+    matches = cdc.search("xylene")
+    assert len(matches) >= 3
+
+
+def test_cdc_toxprofile_for_unique():
+    match = cdc.toxprofile_for("Acrylamide")
+    assert isinstance(match, dict)
+    assert match.get("cas") == "79-06-1"

tests/test_ctx_resolve.py

@@ -0,0 +1,55 @@
+import asyncio
+
+import httpx
+
+from core.sources import ctx
+
+
+def test_resolve_dtxsid_prefers_cas(monkeypatch):
+    async def fake_cas(q, http):
+        return "DTXSID1234567"
+
+    async def fake_name(q, http):
+        return None
+
+    async def fake_pc(q, http):
+        return "DTXSID0000000"
+
+    async def fake_dash(q, http):
+        return None
+
+    monkeypatch.setattr(ctx, "_resolve_from_cas", fake_cas)
+    monkeypatch.setattr(ctx, "_resolve_from_name", fake_name)
+    monkeypatch.setattr(ctx, "_resolve_dtxsid_via_pubchem", fake_pc)
+    monkeypatch.setattr(ctx, "_resolve_dtxsid_via_dashboard", fake_dash)
+
+    async def run():
+        async with httpx.AsyncClient() as client:
+            return await ctx.resolve_dtxsid("50-00-0", client)
+
+    assert asyncio.run(run()) == "DTXSID1234567"
+
+
+def test_resolve_dtxsid_pubchem_fallback(monkeypatch):
+    async def fake_cas(q, http):
+        return None
+
+    async def fake_name(q, http):
+        return None
+
+    async def fake_pc(q, http):
+        return "DTXSID7654321"
+
+    async def fake_dash(q, http):
+        return None
+
+    monkeypatch.setattr(ctx, "_resolve_from_cas", fake_cas)
+    monkeypatch.setattr(ctx, "_resolve_from_name", fake_name)
+    monkeypatch.setattr(ctx, "_resolve_dtxsid_via_pubchem", fake_pc)
+    monkeypatch.setattr(ctx, "_resolve_dtxsid_via_dashboard", fake_dash)
+
+    async def run():
+        async with httpx.AsyncClient() as client:
+            return await ctx.resolve_dtxsid("formaldehyde", client)
+
+    assert asyncio.run(run()) == "DTXSID7654321"

tests/test_iarc_render.py

@@ -0,0 +1,19 @@
+from app import render_iarc_block
+
+
+def test_iarc_render_url():
+    md = render_iarc_block({"ok": True, "url": "https://example.com"})
+    assert "Search IARC Monographs" in md
+    assert "https://example.com" in md
+
+
+def test_iarc_render_results_list():
+    md = render_iarc_block(
+        {
+            "ok": True,
+            "results": [
+                {"title": "IARC Monographs - Foo", "url": "https://example.com/foo", "year": "2020"}
+            ],
+        }
+    )
+    assert "- [IARC Monographs - Foo](https://example.com/foo) (2020)" in md

tests/test_rate_limit.py

@@ -0,0 +1,18 @@
+from core import rate_limit
+from core.config import settings
+
+
+def test_rate_limit_in_memory(monkeypatch):
+    monkeypatch.setattr(settings, "max_ai_summaries_per_day", 2)
+    monkeypatch.setattr(settings, "redis_url", "")
+    rate_limit._state["date"] = None
+    rate_limit._state["count"] = 0
+
+    allowed1, info1 = rate_limit.check_and_increment_global_ai_cap()
+    allowed2, info2 = rate_limit.check_and_increment_global_ai_cap()
+    allowed3, info3 = rate_limit.check_and_increment_global_ai_cap()
+
+    assert allowed1 is True
+    assert allowed2 is True
+    assert allowed3 is False
+    assert info3["limit"] == 2