Issue 1811: Add tests to illustrate issue

This commit is contained in:
Adrien Brunet 2018-12-21 12:30:14 +01:00
parent 2d48dd4e4c
commit 3f95d60a82
2 changed files with 45 additions and 0 deletions

View File

@ -52,6 +52,13 @@ class ForeignKeySource(RESTFrameworkModel):
on_delete=models.CASCADE) on_delete=models.CASCADE)
class ForeignKeySourceWithLimitedChoices(RESTFrameworkModel):
target = models.ForeignKey(ForeignKeyTarget, help_text='Target',
verbose_name='Target',
limit_choices_to={"name__startswith": "limited-"},
on_delete=models.CASCADE)
# Nullable ForeignKey # Nullable ForeignKey
class NullableForeignKeySource(RESTFrameworkModel): class NullableForeignKeySource(RESTFrameworkModel):
name = models.CharField(max_length=100) name = models.CharField(max_length=100)

View File

@ -0,0 +1,38 @@
from django.test import TestCase
from rest_framework import serializers
from .models import (
ForeignKeySource,
ForeignKeyTarget,
ForeignKeySourceWithLimitedChoices,
)
class ForeignKeySourceWithLimitedChoicesSerializer(serializers.ModelSerializer):
class Meta:
model = ForeignKeySourceWithLimitedChoices
fields = ("id", "target")
class ForeignKeySourceSerializer(serializers.ModelSerializer):
class Meta:
model = ForeignKeySource
fields = ("id", "target")
class LimitedChoicesInQuerySetTests(TestCase):
def setUp(self):
for idx in range(1, 4):
limited_target = ForeignKeyTarget(name="limited-target-%d" % idx)
limited_target.save()
target = ForeignKeyTarget(name="target-%d" % idx)
target.save()
def test_queryset_size_without_limited_choices(self):
queryset = ForeignKeySourceSerializer().fields["target"].get_queryset()
assert len(queryset) == 6
def test_queryset_size_with_limited_choices(self):
queryset = ForeignKeySourceWithLimitedChoicesSerializer().fields["target"].get_queryset()
assert len(queryset) == 3