diff --git a/chats/apps/dashboard/tests/test_serializers.py b/chats/apps/dashboard/tests/test_serializers.py index 0b366462..e8add10c 100644 --- a/chats/apps/dashboard/tests/test_serializers.py +++ b/chats/apps/dashboard/tests/test_serializers.py @@ -1,6 +1,8 @@ from django.test import TestCase - -from chats.apps.api.v1.dashboard.serializers import dashboard_agents_data +from chats.apps.api.v1.dashboard.serializers import ( + dashboard_general_data, + dashboard_agents_data, +) from chats.apps.projects.models import Project @@ -20,6 +22,20 @@ class SerializerTests(TestCase): def setUp(self): self.project = Project.objects.get(pk="34a93b52-231e-11ed-861d-0242ac120002") + def test_active_chats_function_passing_sector(self): + serializer = dashboard_general_data( + context={"sector": "d49049f0-c601-4e05-a293-98c1dea5fe4f"}, + project=self.project, + ) + self.assertEqual(serializer["active_chats"], 0) + + def test_active_chats_function_without_filter(self): + serializer = dashboard_general_data( + context={}, + project=self.project, + ) + self.assertEqual(serializer["active_chats"], 1) + def test_returned_fields_from_dashboard_agent_serializer(self): project = Project.objects.get(uuid="34a93b52-231e-11ed-861d-0242ac120002") instance = dashboard_agents_data(