encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-09-17 09:42:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add test case for overriding default pk_field

Browse Source
This commit is contained in:
Ryan Thomson 2022-12-11 16:19:29 -05:00
parent 1393573718
commit 5c7440dbca
No known key found for this signature in database
GPG Key ID: 6D19CE34E3282000
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_model_serializer.py
View File

@ -754,6 +754,20 @@ class TestUUIDForeignKeyMapping(TestCase):
""") """)
self.assertEqual(repr(TestSerializer()), expected) self.assertEqual(repr(TestSerializer()), expected)
def test_uuid_pk_relation_override(self):
class TestSerializer(serializers.ModelSerializer):
class Meta:
model = UUIDForeignKeyModel
fields = '__all__'
extra_kwargs = {'foreign_key': {'pk_field': serializers.UUIDField(format='int')}}
expected = dedent("""
TestSerializer():
id = IntegerField(label='ID', read_only=True)
foreign_key = PrimaryKeyRelatedField(pk_field=UUIDField(format='int'), queryset=UUIDForeignKeyTarget.objects.all())
""")
self.assertEqual(repr(TestSerializer()), expected)
class DisplayValueTargetModel(models.Model): class DisplayValueTargetModel(models.Model):
name = models.CharField(max_length=100) name = models.CharField(max_length=100)