def test_widget_to_json_no_data(self): """Using to_json with no data should return None for data""" data = None widget = WidgetField('Title', [TestWidgetSub]) self.assertEqual(widget.to_json(data), None)
def test_get_all_widgets(self): """Get all the widgets associated with a field""" register.widget(TestWidget) register.widget(TestWidget2) register.widget(TestWidget3) # Create article and image for testing article = DispatchTestHelpers.create_article(self.client) image = DispatchTestHelpers.create_image(self.client) widget1 = TestWidget() widget2 = TestWidget2() widget3 = TestWidget3() testfield = WidgetField('Title', [TestWidgetSub]) field_data = { 'id': 'test-widget', 'data': { 'title': 'test title', 'description': 'test description', 'article': article.data['id'], 'image': image.data['id'], 'widget': { 'id': 'test-widget-2', 'data': { 'title': 'test title', 'description': 'test description', 'article': article.data['id'], 'image': image.data['id'], 'widget': { 'id': 'test-widget-3', 'data': { 'title': 'Test widget 3' } } } } } } json = testfield.to_json(field_data) prepared_data = testfield.prepare_data(field_data).data self.assertEqual(json['id'], 'test-widget') self.assertEqual(json['data']['widget']['id'], 'test-widget-2') self.assertEqual(json['data']['widget']['data']['widget']['id'], 'test-widget-3') self.assertEqual(prepared_data['widget']['id'], 'test-widget-2') self.assertEqual(prepared_data['widget']['data']['widget']['id'], 'test-widget-3')
def test_widget_field_to_json(self): """Should be able to get to_json from field""" register.widget(TestWidget) # Create article and image for testing article = DispatchTestHelpers.create_article(self.client) image = DispatchTestHelpers.create_image(self.client) testfield = WidgetField('Title', [TestWidgetSub]) field_data = { 'id': 'test-widget', 'data': { 'title': 'test title', 'description': 'test description', 'article': article.data['id'], 'image': image.data['id'] } } json = testfield.to_json(field_data) self.assertEqual(json['id'], 'test-widget')