From 834afc1f909cc2e29a70aacb6716c41bdb02ee97 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fran=C3=A7ois=20Freitag?= Date: Tue, 14 Mar 2023 17:23:18 +0100 Subject: [PATCH] Reuse existing models --- tests/djapp/models.py | 4 ---- tests/test_django.py | 14 +++++++------- 2 files changed, 7 insertions(+), 11 deletions(-) diff --git a/tests/djapp/models.py b/tests/djapp/models.py index b8a5d055..b7aa8794 100644 --- a/tests/djapp/models.py +++ b/tests/djapp/models.py @@ -18,10 +18,6 @@ class StandardModel(models.Model): foo = models.CharField(max_length=20) -class WithRelation(models.Model): - rel = models.ForeignKey(StandardModel, models.CASCADE) - - class NonIntegerPk(models.Model): foo = models.CharField(max_length=20, primary_key=True) bar = models.CharField(max_length=20, blank=True) diff --git a/tests/test_django.py b/tests/test_django.py index f0c5c9d1..df929846 100644 --- a/tests/test_django.py +++ b/tests/test_django.py @@ -998,25 +998,25 @@ def test_class_decorator_related_model_with_post_hook(self): """ @factory.django.mute_signals(signals.post_save) - class StandardFactory(factory.django.DjangoModelFactory): + class PointedFactory(factory.django.DjangoModelFactory): class Meta: - model = models.StandardModel + model = models.PointedModel @factory.post_generation def post_action(obj, create, extracted, **kwargs): pass - class WithRelationFactory(factory.django.DjangoModelFactory): - rel = factory.SubFactory(StandardFactory) + class PointerFactory(factory.django.DjangoModelFactory): + pointed = factory.SubFactory(PointedFactory) class Meta: - model = models.WithRelation + model = models.PointerModel - WithRelationFactory.create() + PointerFactory.create() self.handlers.post_save.assert_called_once_with( signal=mock.ANY, - sender=models.WithRelation, + sender=models.PointerModel, instance=mock.ANY, created=True, update_fields=None,