Hopiu/django-model-utils
1
0
Fork 0
mirror of https://github.com/Hopiu/django-model-utils.git synced 2026-03-16 20:00:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove unnecessary model and simplify tests for prefetch_related

Browse source
This commit is contained in:
joeriddles 2023-01-26 14:25:09 -08:00
parent 9d466ffa74
commit 604c6f0b62
2 changed files with 8 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/models.py
View file

@ -324,14 +324,6 @@ class ModelTrackedFK(models.Model):
custom_tracker_without_id = ModelTracker(fields=['fk'])
class ModelTrackedPrefetchRelatedFK(models.Model):
fk = models.ForeignKey('ModelTrackedFK', on_delete=models.CASCADE)
tracker = ModelTracked()
custom_tracker = ModelTracker(fields=['fk_id'])
custom_tracker_without_id = ModelTracker(fields=['fk'])
class ModelTrackedNotDefault(models.Model):
name = models.CharField(max_length=20)
number = models.IntegerField()

17
tests/test_fields/test_field_tracker.py
View file

@ -15,7 +15,6 @@ from tests.models import (
ModelTrackedFK,
ModelTrackedMultiple,
ModelTrackedNotDefault,
ModelTrackedPrefetchRelatedFK,
Tracked,
TrackedAbstract,
TrackedFileField,
@ -539,27 +538,27 @@ class FieldTrackerForeignKeyTests(FieldTrackerTestCase):
self.assertCurrent(fk=self.instance.fk_id)
class ModelTrackedPrefetchRelatedFKTests(FieldTrackerTestCase):
class FieldTrackerForeignKeyPrefetchRelatedTests(FieldTrackerTestCase):
"""Test that using `prefetch_related` on a tracked field does raise a ValueError."""
tracked_class = ModelTrackedPrefetchRelatedFK
fk_class = Tracked
tracked_class = TrackedFK
def setUp(self):
model_tracked = ModelTracked.objects.create(name="", number=0)
model_tracked_fk = ModelTrackedFK.objects.create(fk=model_tracked)
self.instance = ModelTrackedPrefetchRelatedFK.objects.create(fk=model_tracked_fk)
model_tracked = self.fk_class.objects.create(name="", number=0)
self.instance = self.tracked_class.objects.create(fk=model_tracked)
def test_default(self):
self.tracker = self.instance.tracker
self.assertIsNotNone(list(self.tracked_class.objects.prefetch_related("fk__fk")))
self.assertIsNotNone(list(self.tracked_class.objects.prefetch_related("fk")))
def test_custom(self):
self.tracker = self.instance.custom_tracker
self.assertIsNotNone(list(self.tracked_class.objects.prefetch_related("fk__fk")))
self.assertIsNotNone(list(self.tracked_class.objects.prefetch_related("fk")))
def test_custom_without_id(self):
self.tracker = self.instance.custom_tracker_without_id
self.assertIsNotNone(list(self.tracked_class.objects.prefetch_related("fk__fk")))
self.assertIsNotNone(list(self.tracked_class.objects.prefetch_related("fk")))
class FieldTrackerTimeStampedTests(FieldTrackerTestCase):