Fix test.

This commit is contained in:
Norman Jäckel 2014-09-03 16:49:36 +02:00
parent e64f422d45
commit 3fc1b13217

View File

@ -763,11 +763,11 @@ class ManyToManyTwoTests(TestCase):
null==True and blank==True. null==True and blank==True.
""" """
def setUp(self): def setUp(self):
class ManyToManySerializer(serializers.ModelSerializer): class ManyToManySerializerTwo(serializers.ModelSerializer):
class Meta: class Meta:
model = ManyToManyModelTwo model = ManyToManyModelTwo
self.serializer_class = ManyToManySerializer self.serializer_class = ManyToManySerializerTwo
# An anchor instance to use for the relationship # An anchor instance to use for the relationship
self.anchor = Anchor() self.anchor = Anchor()
@ -781,8 +781,7 @@ class ManyToManyTwoTests(TestCase):
serializer = self.serializer_class(data=data) serializer = self.serializer_class(data=data)
self.assertEqual(serializer.is_valid(), True) self.assertEqual(serializer.is_valid(), True)
instance = serializer.save() instance = serializer.save()
self.assertEqual(len(ManyToManyModelTwo.objects.all()), 1) self.assertEqual(ManyToManyModelTwo.objects.count(), 1)
self.assertEqual(instance.pk, 1)
self.assertEqual(list(instance.rel.all()), [self.anchor]) self.assertEqual(list(instance.rel.all()), [self.anchor])