fix unsafe test call

dymmond · Feb 6, 2025 · 4560d45 · 4560d45
1 parent 9e95d78
commit 4560d45
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/tests/factory/test_factory.py b/tests/factory/test_factory.py
@@ -6,7 +6,8 @@
 import edgy
 from edgy.testing import DatabaseTestClient
 from edgy.testing.exceptions import ExcludeValue
-from edgy.testing.factory import FactoryField, ModelFactory, ModelFactoryContext
+from edgy.testing.factory import FactoryField, ModelFactory
+from edgy.testing.factory.base import ModelFactoryContextImplementation
 from edgy.testing.factory.metaclasses import DEFAULT_MAPPING
 from tests.settings import DATABASE_URL
 
@@ -341,27 +342,33 @@ class Meta:
         }:
             DEFAULT_MAPPING[field_type_name](
                 ProductFactory.meta.fields["user"],
-                ModelFactoryContext(
+                ModelFactoryContextImplementation(
                     faker=ProductFactory.meta.faker,
                     exclude_autoincrement=ProductFactory.exclude_autoincrement,
+                    depth=0,
+                    callcounts={},
                 ),
                 {},
             )
         elif field_type_name == "ChoiceField":
             DEFAULT_MAPPING[field_type_name](
                 ProductFactory.meta.fields["type"],
-                ModelFactoryContext(
+                ModelFactoryContextImplementation(
                     faker=ProductFactory.meta.faker,
                     exclude_autoincrement=ProductFactory.exclude_autoincrement,
+                    depth=0,
+                    callcounts={},
                 ),
                 {},
             )
         elif field_type_name == "RefForeignKey":
             DEFAULT_MAPPING[field_type_name](
                 UserFactory.meta.fields["product_ref"],
-                ModelFactoryContext(
+                ModelFactoryContextImplementation(
                     faker=UserFactory.meta.faker,
                     exclude_autoincrement=UserFactory.exclude_autoincrement,
+                    depth=0,
+                    callcounts={},
                 ),
                 {},
             )
@@ -370,9 +377,11 @@ class Meta:
             if callback:
                 callback(
                     UserFactory.meta.fields["name"],
-                    ModelFactoryContext(
+                    ModelFactoryContextImplementation(
                         faker=UserFactory.meta.faker,
                         exclude_autoincrement=UserFactory.exclude_autoincrement,
+                        depth=0,
+                        callcounts={},
                     ),
                     {},
                 )