Improve tests

Author: mikicz

pyproject.toml

@@ -32,6 +32,10 @@ classifiers = [
 [project.urls]
 "Repository" = "https://github.com/xelixdev/django-memoized-prefetch"
 
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
 [tool.ruff]
 line-length = 120
 target-version = "py39"

pytest.ini

@@ -3,3 +3,6 @@ DJANGO_SETTINGS_MODULE = tests.test_project.settings
 addopts = --strict-markers --cov=django_memoized_prefetch --cov-report term-missing:skip-covered
 pythonpath = .
 django_find_project = false
+
+filterwarnings =
+    error::seal.exceptions.UnsealedAttributeAccess

tests/test_memoized_prefetch.py

@@ -6,6 +6,7 @@
 
 from django_memoized_prefetch import MemoizedPrefetch, MemoizedPrefetchConfig
 from tests.test_project.test_app.factories import (
+    SomeChildModelFactory,
     SomeDifferentParentModelFactory,
     SomeModelFactory,
     SomeParentModelFactory,
@@ -115,6 +116,8 @@ def test_memoized_prefetch(
         with django_assert_num_queries(0):  # all parents already fetched
             memoized_prefetch.process_chunk(objects)
 
+        assert objects
+
         for obj in objects:
             # does not throw seal attribute -> fetched in process_chunk
             assert obj.some_parent_model is not None
@@ -151,6 +154,8 @@ def test_memoized_prefetch_many_to_many(
         with django_assert_num_queries(0):  # already fetched
             memoized_prefetch.process_chunk(objects)
 
+        assert objects
+
         for obj in objects:
             related_models = list(obj.some_related_models.all())
             assert len(related_models) == 2
@@ -178,22 +183,24 @@ def test_nullable_field(
     @pytest.fixture
     def child_models(self, objects_some_different_parent_a, objects_some_different_parent_b) -> list[SomeChildModel]:
         return [
-            SomeChildModel(some_model=some_model)
+            SomeChildModelFactory(some_model=some_model)
             for some_model in itertools.chain(objects_some_different_parent_a, objects_some_different_parent_b)
         ]
 
     def test_nested(self, child_models: list[SomeChildModel]):
         memoized_prefetch = MemoizedPrefetch(
             MemoizedPrefetchConfig(SomeParentModel, ["some_model__some_parent_model"], prefetch_all=True),
             MemoizedPrefetchConfig(
-                SomeDifferentParentModel, ["some_model__some_other_different_parent"]
+                SomeDifferentParentModel, ["some_model.some_other_different_parent"]
             ),  # support both . and __
         )
 
         objects = list(SomeChildModel.objects.select_related("some_model").seal())
 
         memoized_prefetch.process_chunk(objects)
 
+        assert objects
+
         for obj in objects:
             # does not throw seal attribute -> fetched in process_chunk
             assert obj.some_model.some_parent_model is not None

tests/test_project/test_app/factories.py

@@ -53,6 +53,7 @@ def some_related_models(self, create: bool, extracted: list[SomeRelatedModel]) -
 
     class Meta:
         model = SomeModel
+        skip_postgeneration_save = True
 
 
 class SomeChildModelFactory(factory.django.DjangoModelFactory):