Update test_api_v1.py

lucyparsons · Dec 20, 2024 · 2f50bb9 · 2f50bb9
1 parent 9a1fe2f
commit 2f50bb9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/OpenOversight/tests/routes/test_api_v1.py b/OpenOversight/tests/routes/test_api_v1.py
@@ -26,7 +26,7 @@ def test_get_dept_attributes(client, session, department_id: int):
 
         # Get assignments
         expected_assignments = (
-            session.get(Assignment)
+            session.query(Assignment)
             .join(Assignment.base_officer)
             .filter(Officer.department_id == department_id)
             .options(contains_eager(Assignment.base_officer))
@@ -55,7 +55,7 @@ def test_get_dept_attributes(client, session, department_id: int):
         )
         incidents = json.loads(resp_incidents.data.decode(ENCODING_UTF_8))
 
-        assert resp_assignments.status_code == HTTPStatus.OK
+        assert resp_incidents.status_code == HTTPStatus.OK
         assert len(incidents) == expected_incidents
 
         # Get salaries
@@ -73,5 +73,5 @@ def test_get_dept_attributes(client, session, department_id: int):
         )
         salaries = json.loads(resp_salaries.data.decode(ENCODING_UTF_8))
 
-        assert resp_assignments.status_code == HTTPStatus.OK
+        assert resp_salaries.status_code == HTTPStatus.OK
         assert len(salaries) == expected_salaries