update tests/unit/test_candidate_model.py

tests/unit/test_candidate_model.py

@@ -0,0 +1,33 @@
+from app.db_models import CandidateSource
+
+
+def test_candidate_source_defaults():
+    # SQLAlchemy models don't apply 'default' on __init__ without a session
+    # So we check the Column configuration instead
+    assert CandidateSource.status.default.arg == "pending"
+    assert CandidateSource.confidence_score.default.arg == 0
+    assert CandidateSource.proxies_found_count.default.arg == 0
+    assert CandidateSource.fail_count.default.arg == 0
+
+    candidate = CandidateSource(
+        url="http://example.com/proxies.txt", discovery_method="github"
+    )
+    # Explicitly check what we set
+    assert candidate.url == "http://example.com/proxies.txt"
+    assert candidate.discovery_method == "github"
+
+
+def test_candidate_source_fields():
+    candidate = CandidateSource(
+        url="http://test.com",
+        domain="test.com",
+        discovery_method="ai",
+        status="approved",
+        confidence_score=90,
+        meta_data={"summary": "found by ai"},
+    )
+
+    assert candidate.domain == "test.com"
+    assert candidate.status == "approved"
+    assert candidate.confidence_score == 90
+    assert candidate.meta_data == {"summary": "found by ai"}