encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-23 01:57:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make nested tests consistent with pk and hyperlink tests

Browse Source
This commit is contained in:
Tom Christie 2012-12-21 09:15:28 +00:00
parent 37acea5240
commit 04119245ec
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
rest_framework/tests/relations_nested.py
View File

@ -35,18 +35,17 @@ class ReverseForeignKeyTests(TestCase):
for idx in range(1, 4): for idx in range(1, 4):
source = ForeignKeySource(name='source-%d' % idx, target=target) source = ForeignKeySource(name='source-%d' % idx, target=target)
source.save() source.save()
self.target_data = {'id': 1, 'name': u'target-1', 'sources': [
def test_reverse_foreign_key_retrieve(self):
queryset = ForeignKeyTarget.objects.all()
serializer = ForeignKeyTargetSerializer(queryset)
expected = [
{'id': 1, 'name': u'target-1', 'sources': [
{'id': 1, 'name': u'source-1', 'target': 1}, {'id': 1, 'name': u'source-1', 'target': 1},
{'id': 2, 'name': u'source-2', 'target': 1}, {'id': 2, 'name': u'source-2', 'target': 1},
{'id': 3, 'name': u'source-3', 'target': 1}, {'id': 3, 'name': u'source-3', 'target': 1},
]},
{'id': 2, 'name': u'target-2', 'sources': [
]} ]}
self.new_target_data = {'id': 2, 'name': u'target-2', 'sources': []} ]
self.data = [self.target_data, self.new_target_data] self.assertEquals(serializer.data, expected)
def check_serialized_targets(self, data):
queryset = ForeignKeyTarget.objects.all()
serializer = ForeignKeyTargetSerializer(queryset)
self.assertEquals(serializer.data, data)
def test_reverse_foreign_key_retrieve(self):
self.check_serialized_targets(self.data)