From c36e2eabe9e9984e56135dda1c3cac7109e285ca Mon Sep 17 00:00:00 2001
From: Rob Golding <rob@robgolding.com>
Date: Sun, 28 Jan 2018 20:58:08 +0000
Subject: [PATCH] Add `ForeignKeyTargetWithSourcesSerializer` to relations/pk
 tests

This serializer nests the reverse `sources` relation, to test that the
reverse relation is iterated over correctly.
---
 tests/test_relations_pk.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/tests/test_relations_pk.py b/tests/test_relations_pk.py
index 3317d6251..7cba9d42d 100644
--- a/tests/test_relations_pk.py
+++ b/tests/test_relations_pk.py
@@ -26,16 +26,24 @@ class ManyToManySourceSerializer(serializers.ModelSerializer):
 
 
 # ForeignKey
+class ForeignKeySourceSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ForeignKeySource
+        fields = ('id', 'name', 'target')
+
+
 class ForeignKeyTargetSerializer(serializers.ModelSerializer):
     class Meta:
         model = ForeignKeyTarget
         fields = ('id', 'name', 'sources')
 
 
-class ForeignKeySourceSerializer(serializers.ModelSerializer):
+class ForeignKeyTargetWithSourcesSerializer(serializers.ModelSerializer):
+    sources = ForeignKeySourceSerializer(many=True)
+
     class Meta:
-        model = ForeignKeySource
-        fields = ('id', 'name', 'target')
+        model = ForeignKeyTarget
+        fields = ('id', 'name', 'sources')
 
 
 # Nullable ForeignKey