diff --git a/api/data_workspace/tests/test_views.py b/api/data_workspace/tests/test_views.py index b13de23a7..61857da74 100644 --- a/api/data_workspace/tests/test_views.py +++ b/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