Fix bug in test_search_tokens.py

Police-Data-Accessibility-Project · Jun 2, 2024 · 24a9a57 · 24a9a57
1 parent 1c0a8a0
commit 24a9a57
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration/test_search_tokens.py b/tests/integration/test_search_tokens.py
@@ -22,8 +22,8 @@ def test_search_tokens_get(
     assert response.status_code == 200
     data = response.json.get("data")
     assert (
-        data["count"] == 1
+        len(data) == 1
     ), "Quick Search endpoint response should return only one entry"
-    entry = data["data"][0]
+    entry = data[0]
     assert entry["agency_name"] == "Agency A"
     assert entry["airtable_uid"] == "SOURCE_UID_1"