mirror of
https://github.com/django/django.git
synced 2025-10-28 16:16:12 +00:00
Fixed #36062 -- Handled serialization of CompositePrimaryKeys.
This commit is contained in:
@@ -17,7 +17,7 @@ from django.db.models import CompositePrimaryKey
|
||||
from django.forms import modelform_factory
|
||||
from django.test import TestCase
|
||||
|
||||
from .models import Comment, Post, Tenant, User
|
||||
from .models import Comment, Post, Tenant, TimeStamped, User
|
||||
|
||||
|
||||
class CommentForm(forms.ModelForm):
|
||||
@@ -224,6 +224,13 @@ class CompositePKFixturesTests(TestCase):
|
||||
self.assertEqual(post_2.tenant_id, 2)
|
||||
self.assertEqual(post_2.pk, (post_2.tenant_id, post_2.id))
|
||||
|
||||
def assert_deserializer(self, format, users, serialized_users):
|
||||
deserialized_user = list(serializers.deserialize(format, serialized_users))[0]
|
||||
self.assertEqual(deserialized_user.object.email, users[0].email)
|
||||
self.assertEqual(deserialized_user.object.id, users[0].id)
|
||||
self.assertEqual(deserialized_user.object.tenant, users[0].tenant)
|
||||
self.assertEqual(deserialized_user.object.pk, users[0].pk)
|
||||
|
||||
def test_serialize_user_json(self):
|
||||
users = User.objects.filter(pk=(1, 1))
|
||||
result = serializers.serialize("json", users)
|
||||
@@ -241,6 +248,7 @@ class CompositePKFixturesTests(TestCase):
|
||||
}
|
||||
],
|
||||
)
|
||||
self.assert_deserializer(format="json", users=users, serialized_users=result)
|
||||
|
||||
def test_serialize_user_jsonl(self):
|
||||
users = User.objects.filter(pk=(1, 2))
|
||||
@@ -257,6 +265,7 @@ class CompositePKFixturesTests(TestCase):
|
||||
},
|
||||
},
|
||||
)
|
||||
self.assert_deserializer(format="jsonl", users=users, serialized_users=result)
|
||||
|
||||
@unittest.skipUnless(HAS_YAML, "No yaml library detected")
|
||||
def test_serialize_user_yaml(self):
|
||||
@@ -276,6 +285,7 @@ class CompositePKFixturesTests(TestCase):
|
||||
},
|
||||
],
|
||||
)
|
||||
self.assert_deserializer(format="yaml", users=users, serialized_users=result)
|
||||
|
||||
def test_serialize_user_python(self):
|
||||
users = User.objects.filter(pk=(2, 4))
|
||||
@@ -294,6 +304,13 @@ class CompositePKFixturesTests(TestCase):
|
||||
},
|
||||
],
|
||||
)
|
||||
self.assert_deserializer(format="python", users=users, serialized_users=result)
|
||||
|
||||
def test_serialize_user_xml(self):
|
||||
users = User.objects.filter(pk=(1, 1))
|
||||
result = serializers.serialize("xml", users)
|
||||
self.assertIn('<object model="composite_pk.user" pk=\'["1", "1"]\'>', result)
|
||||
self.assert_deserializer(format="xml", users=users, serialized_users=result)
|
||||
|
||||
def test_serialize_post_uuid(self):
|
||||
posts = Post.objects.filter(pk=(2, "11111111-1111-1111-1111-111111111111"))
|
||||
@@ -311,3 +328,27 @@ class CompositePKFixturesTests(TestCase):
|
||||
},
|
||||
],
|
||||
)
|
||||
|
||||
def test_serialize_datetime(self):
|
||||
result = serializers.serialize("json", TimeStamped.objects.all())
|
||||
self.assertEqual(
|
||||
json.loads(result),
|
||||
[
|
||||
{
|
||||
"model": "composite_pk.timestamped",
|
||||
"pk": [1, "2022-01-12T05:55:14.956"],
|
||||
"fields": {
|
||||
"id": 1,
|
||||
"created": "2022-01-12T05:55:14.956",
|
||||
},
|
||||
},
|
||||
],
|
||||
)
|
||||
|
||||
def test_invalid_pk_extra_field(self):
|
||||
json = (
|
||||
'[{"fields": {"email": "user0001@example.com", "id": 1, "tenant": 1}, '
|
||||
'"pk": [1, 1, "extra"], "model": "composite_pk.user"}]'
|
||||
)
|
||||
with self.assertRaises(serializers.base.DeserializationError):
|
||||
next(serializers.deserialize("json", json))
|
||||
|
||||
Reference in New Issue
Block a user