diff --git a/backend/app/tests/test_report_store.py b/backend/app/tests/test_report_store.py index ce9f254..63bbc06 100644 --- a/backend/app/tests/test_report_store.py +++ b/backend/app/tests/test_report_store.py @@ -39,7 +39,7 @@ class TestReportStore: store.add_report(_sample_report("test.com")) domains = store.get_domains() - assert "test.com" in domains + assert any(d == "test.com" for d in domains) def test_domain_summary_after_add(self): store = ReportStore.get_instance() @@ -72,7 +72,7 @@ class TestReportStore: assert store.delete_domain_with_cleanup("test.com") is True assert "test.com" not in store.get_domains() - assert "other.com" in store.get_domains() + assert any(d == "other.com" for d in store.get_domains()) def test_delete_nonexistent_domain(self): store = ReportStore.get_instance() diff --git a/backend/app/tests/test_reports_api.py b/backend/app/tests/test_reports_api.py index e14fe77..337b712 100644 --- a/backend/app/tests/test_reports_api.py +++ b/backend/app/tests/test_reports_api.py @@ -84,7 +84,7 @@ def test_upload_populates_domains_list(client: TestClient): response = client.get("/api/v1/reports/domains") assert response.status_code == 200 domains = response.json() - assert "example.com" in domains + assert any(d == "example.com" for d in domains) def test_reports_domains_empty(client: TestClient):