Add test for departments view

api/data_workspace/tests/test_views.py

@@ -3,6 +3,8 @@
 
 from api.parties.enums import PartyType
 from test_helpers.clients import DataTestClient
+from api.teams.tests.factories import TeamFactory
+from api.teams.models import Department
 
 
 class DataWorkspaceTests(DataTestClient):
@@ -103,3 +105,19 @@ def test_teams(self):
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         options = response.json()["actions"]["OPTIONS"]
         self.assertEqual(tuple(options.keys()), expected_fields)
+
+    def test_departments(self):
+        team = TeamFactory()
+        url = reverse("data_workspace:dw-departments-list")
+        response = self.client.get(url)
+        payload = response.json()
+
+        # Ensure we get departments and not sth else
+        deps_ids = [d["id"] for d in payload["results"]]
+        assert str(team.department.id) in deps_ids
+        assert not str(team.id) in deps_ids
+        assert len(deps_ids) == Department.objects.count()
+
+        # Ensure we get some expected fields
+        expected_fields = {"id", "name"}
+        assert set(payload["results"][0].keys()) == expected_fields