mirrors/wagtail
mirrors/wagtail
0
0
Fork 0
mirror of https://github.com/wagtail/wagtail.git synced 2024-11-30 19:20:56 +01:00
Code Issues Releases Wiki Activity

Test foreign key representation in streamfield

Browse Source
This commit is contained in:
Karl Hobley 2015-07-02 12:41:14 +01:00
parent 90cec15b3b
commit ff9f6d1215
1 changed files with 25 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
wagtail/contrib/wagtailapi/tests/test_pages.py
View File

@ -599,26 +599,42 @@ class TestPageDetailWithStreamField(TestCase):
fixtures = ['test.json'] fixtures = ['test.json']
def setUp(self): def setUp(self):
homepage = Page.objects.get(url_path='/home/') self.homepage = Page.objects.get(url_path='/home/')
self.stream_page = StreamPage(
title='stream page', slug='stream-page', def make_stream_page(self, body):
body='[{"type": "text", "value": "foo"}]') stream_page = StreamPage(
homepage.add_child(instance=self.stream_page) title='stream page',
slug='stream-page',
body=body
)
return self.homepage.add_child(instance=stream_page)
def test_can_fetch_streamfield_content(self): def test_can_fetch_streamfield_content(self):
response_url = reverse('wagtailapi_v1:pages:detail', args=(self.stream_page.id, )) stream_page = self.make_stream_page('[{"type": "text", "value": "foo"}]')
response_url = reverse('wagtailapi_v1:pages:detail', args=(stream_page.id, ))
response = self.client.get(response_url) response = self.client.get(response_url)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response['Content-type'], 'application/json') self.assertEqual(response['content-type'], 'application/json')
content = json.loads(response.content.decode('UTF-8')) content = json.loads(response.content.decode('utf-8'))
self.assertIn('id', content) self.assertIn('id', content)
self.assertEqual(content['id'], self.stream_page.id) self.assertEqual(content['id'], stream_page.id)
self.assertIn('body', content) self.assertIn('body', content)
self.assertEqual(content['body'], [{'type': 'text', 'value': 'foo'}]) self.assertEqual(content['body'], [{'type': 'text', 'value': 'foo'}])
def test_image_block(self):
stream_page = self.make_stream_page('[{"type": "image", "value": 1}]')
response_url = reverse('wagtailapi_v1:pages:detail', args=(stream_page.id, ))
response = self.client.get(response_url)
content = json.loads(response.content.decode('utf-8'))
# ForeignKeys in a StreamField shouldn't be translated into dictionary representation
self.assertEqual(content['body'], [{'type': 'image', 'value': 1}])
@override_settings( @override_settings(
WAGTAILFRONTENDCACHE={ WAGTAILFRONTENDCACHE={