Make annotate return instance of actual mockset class

stphivos · Sep 4, 2024 · 58c7b33 · 58c7b33
1 parent b8ac119
commit 58c7b33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/django_mock_queries/query.py b/django_mock_queries/query.py
@@ -127,7 +127,7 @@ def annotate(self, **kwargs):
                 row._annotated_fields.append(key)
                 setattr(row, key, get_attribute(row, value)[0])
 
-        return MockSet(*results, clone=self)
+        return self._mockset_class()(*results, clone=self)
 
     def aggregate(self, *args, **kwargs):
         result = {}