From 2b908d2e74018bfb017dddcd2111a1d7ba4a4071 Mon Sep 17 00:00:00 2001 From: Ze'ev Gilovitz Date: Fri, 23 May 2014 11:50:13 +0800 Subject: [PATCH] using serialiser.save_object method now. --- rest_framework/tests/test_pk_related_creation.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/rest_framework/tests/test_pk_related_creation.py b/rest_framework/tests/test_pk_related_creation.py index bf2f3ade9..b3a392646 100644 --- a/rest_framework/tests/test_pk_related_creation.py +++ b/rest_framework/tests/test_pk_related_creation.py @@ -47,5 +47,9 @@ class TestPrimaryKeyRelatedRelation(TestCase): self.assertTrue(serializer.is_valid()) obj = serializer.object + serializer.save_object(obj) - self.assertEqual(obj.members, [person]) + self.assertEquals(obj.members.count(), 1) + + member = obj.members.all()[0] + self.assertEqual(member, person)