From bd7925c181f52e31891198d8226c0daa9df75aaa Mon Sep 17 00:00:00 2001
From: Rob Golding <rob@robgolding.com>
Date: Sun, 28 Jan 2018 20:55:13 +0000
Subject: [PATCH] Give `ForeignKeySource` a custom manager

This change allows the tests to ensure that both forward and reverse
foreign key serialization works when the model has a custom manager.
---
 tests/models.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tests/models.py b/tests/models.py
index e5d49a0a5..40f69031f 100644
--- a/tests/models.py
+++ b/tests/models.py
@@ -3,6 +3,8 @@ from __future__ import unicode_literals
 import uuid
 
 from django.db import models
+from django.db.models.manager import BaseManager
+from django.db.models.query import QuerySet
 from django.utils.translation import ugettext_lazy as _
 
 
@@ -45,12 +47,18 @@ class UUIDForeignKeyTarget(RESTFrameworkModel):
     name = models.CharField(max_length=100)
 
 
+class ForeignKeySourceManager(BaseManager.from_queryset(QuerySet)):
+    pass
+
+
 class ForeignKeySource(RESTFrameworkModel):
     name = models.CharField(max_length=100)
     target = models.ForeignKey(ForeignKeyTarget, related_name='sources',
                                help_text='Target', verbose_name='Target',
                                on_delete=models.CASCADE)
 
+    objects = ForeignKeySourceManager()
+
 
 # Nullable ForeignKey
 class NullableForeignKeySource(RESTFrameworkModel):