diff options
Diffstat (limited to 'rest_framework/tests/relations_pk.py')
| -rw-r--r-- | rest_framework/tests/relations_pk.py | 35 | 
1 files changed, 17 insertions, 18 deletions
diff --git a/rest_framework/tests/relations_pk.py b/rest_framework/tests/relations_pk.py index e5391f1b..c2e61279 100644 --- a/rest_framework/tests/relations_pk.py +++ b/rest_framework/tests/relations_pk.py @@ -38,7 +38,7 @@ class ForeignKeySource(models.Model):  class ForeignKeyTargetSerializer(serializers.ModelSerializer): -    sources = serializers.ManyPrimaryKeyRelatedField(read_only=True) +    sources = serializers.ManyPrimaryKeyRelatedField()      class Meta:          model = ForeignKeyTarget @@ -235,24 +235,23 @@ class PKForeignKeyTests(TestCase):          ]          self.assertEquals(serializer.data, expected) -    # TODO: See #511 -    # def test_reverse_foreign_key_create(self): -    #     data = {'id': 3, 'name': u'target-3', 'sources': [1, 3]} -    #     serializer = ForeignKeyTargetSerializer(data=data) -    #     self.assertTrue(serializer.is_valid()) -    #     obj = serializer.save() -    #     self.assertEquals(serializer.data, data) -    #     self.assertEqual(obj.name, u'target-3') +    def test_reverse_foreign_key_create(self): +        data = {'id': 3, 'name': u'target-3', 'sources': [1, 3]} +        serializer = ForeignKeyTargetSerializer(data=data) +        self.assertTrue(serializer.is_valid()) +        obj = serializer.save() +        self.assertEquals(serializer.data, data) +        self.assertEqual(obj.name, u'target-3') -    #     # Ensure target 4 is added, and everything else is as expected -    #     queryset = ForeignKeyTarget.objects.all() -    #     serializer = ForeignKeyTargetSerializer(queryset) -    #     expected = [ -    #         {'id': 1, 'name': u'target-1', 'sources': [1, 2, 3]}, -    #         {'id': 2, 'name': u'target-2', 'sources': []}, -    #         {'id': 3, 'name': u'target-3', 'sources': [1, 3]}, -    #     ] -    #     self.assertEquals(serializer.data, expected) +        # Ensure target 4 is added, and everything else is as expected +        queryset = ForeignKeyTarget.objects.all() +        serializer = ForeignKeyTargetSerializer(queryset) +        expected = [ +            {'id': 1, 'name': u'target-1', 'sources': [2]}, +            {'id': 2, 'name': u'target-2', 'sources': []}, +            {'id': 3, 'name': u'target-3', 'sources': [1, 3]}, +        ] +        self.assertEquals(serializer.data, expected)      def test_foreign_key_update_with_invalid_null(self):          data = {'id': 1, 'name': u'source-1', 'target': None}  | 
