From c1cc3ada7b5790deb05f3169dd5d7147e2efd471 Mon Sep 17 00:00:00 2001 From: Rob Golding Date: Sun, 28 Jan 2018 21:01:15 +0000 Subject: [PATCH] Use `ForeignKeyTargetWithSourcesSerializer` in relations/pk tests When checking the expected data, we now must ensure that the nested models are serialized correctly also. --- tests/test_relations_pk.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/test_relations_pk.py b/tests/test_relations_pk.py index 7cba9d42d..5eb031c36 100644 --- a/tests/test_relations_pk.py +++ b/tests/test_relations_pk.py @@ -227,9 +227,13 @@ class PKForeignKeyTests(TestCase): def test_reverse_foreign_key_retrieve(self): queryset = ForeignKeyTarget.objects.all() - serializer = ForeignKeyTargetSerializer(queryset, many=True) + serializer = ForeignKeyTargetWithSourcesSerializer(queryset, many=True) expected = [ - {'id': 1, 'name': 'target-1', 'sources': [1, 2, 3]}, + {'id': 1, 'name': 'target-1', 'sources': [ + {'id': 1, 'name': 'source-1', 'target': 1}, + {'id': 2, 'name': 'source-2', 'target': 1}, + {'id': 3, 'name': 'source-3', 'target': 1}, + ]}, {'id': 2, 'name': 'target-2', 'sources': []}, ] with self.assertNumQueries(3): @@ -237,7 +241,7 @@ class PKForeignKeyTests(TestCase): def test_reverse_foreign_key_retrieve_prefetch_related(self): queryset = ForeignKeyTarget.objects.all().prefetch_related('sources') - serializer = ForeignKeyTargetSerializer(queryset, many=True) + serializer = ForeignKeyTargetWithSourcesSerializer(queryset, many=True) with self.assertNumQueries(2): serializer.data