feat: fix tests

chats/apps/api/v1/rooms/viewsets.py

@@ -2,7 +2,6 @@
 from datetime import timedelta
 
 from django.conf import settings
-from django.db import connection, reset_queries
 from django.db.models import (
     BooleanField,
     Case,
@@ -37,20 +36,6 @@
 from chats.apps.rooms.models import Room
 
 
-def database_debug(func, *args, **kwargs):
-    def inner_func(*args, **kwargs):
-        reset_queries()
-        results = func(*args, **kwargs)
-        query_info = connection.queries
-        print("function_name: {}".format(func.__name__))
-        print("query_count: {}".format(len(query_info)))
-        # queries = ["\n{}\n".format(query["sql"]) for query in query_info]
-        # print("queries: \n{}\n".format("".join(queries)))
-        return results
-
-    return inner_func
-
-
 class RoomViewset(
     mixins.ListModelMixin,
     mixins.RetrieveModelMixin,
@@ -71,10 +56,6 @@ class RoomViewset(
     pagination_class = CursorPagination
     pagination_class.page_size_query_param = "limit"
 
-    @database_debug
-    def list(self, request, *args, **kwargs):
-        return super().list(request, *args, **kwargs)
-
     def get_permissions(self):
         permission_classes = [permissions.IsAuthenticated]
         if self.action != "list":
@@ -110,9 +91,7 @@ def get_queryset(self):
                 output_field=BooleanField(),
             ),
         )
-        # import pdb
 
-        # pdb.set_trace()
         return qs
 
     def get_serializer_class(self):

chats/apps/rooms/tests/test_viewsets.py

@@ -110,14 +110,14 @@ def _ok_list_rooms(self, token, rooms: list, data: dict):
         response, results = self._request_list_rooms(token, data)
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
-        self.assertEqual(response.json().get("count"), len(rooms))
+        self.assertEqual(len(response.json().get("results")), len(rooms))
         for result in results:
             self.assertIn(result.get("uuid"), rooms)
 
     def _not_ok_list_rooms(self, token, data: dict):
         response, _ = self._request_list_rooms(token, data)
         self.assertEqual(response.status_code, status.HTTP_200_OK)
-        self.assertEqual(response.json().get("count"), 0)
+        self.assertEqual(len(response.json().get("results")), 0)
 
     def test_list_rooms_given_agents(self):
         self._ok_list_rooms(