.

Author: kingbuzzman

pytest_django/fixtures.py

@@ -610,7 +610,7 @@ def __init__(self) -> None:
 
         self.contexts = {alias: CaptureQueriesContext(connections[alias]) for alias in connections}
 
-    def __iter__(self) -> None:
+    def __iter__(self) -> Iterable[dict[str, Any]]:
         return iter(self.captured_queries)
 
     def __getitem__(self, index: int) -> dict[str, Any]:
@@ -620,7 +620,7 @@ def __len__(self) -> int:
         return len(self.captured_queries)
 
     @property
-    def captured_queries(self) -> dict[str, Any]:
+    def captured_queries(self) -> list[dict[str, Any]]:
         queries = []
         for context in self.contexts.values():
             queries.extend(context.captured_queries)

tests/test_fixtures.py

@@ -273,12 +273,12 @@ def test_django_assert_max_num_queries_all_connections(
     django_assert_max_num_queries_all_connections: DjangoAssertNumQueries,
 ) -> None:
     with nonverbose_config(request.config):
-        with django_assert_max_num_queries(2):
+        with django_assert_max_num_queries_all_connections(2):
             Item.objects.create(name="1-foo")
             Item.objects.using("second").create(name="2-bar")
 
         with pytest.raises(pytest.fail.Exception) as excinfo:  # noqa: PT012
-            with django_assert_max_num_queries(2) as captured:
+            with django_assert_max_num_queries_all_connections(2) as captured:
                 Item.objects.create(name="1-foo")
                 Item.objects.create(name="2-bar")
                 Item.objects.using("second").create(name="3-quux")