def test_get_non_existant_widget(self):
"""Using get_widget with non-existant widget_id should fail"""
widget = WidgetField('Title', [TestWidgetSub])
try:
widget.get_widget('this_is_not_a_widget_id')
self.fail('WidgetNotFound Error should have been raised')
except WidgetNotFound:
pass
def test_get_widget_from_widget_field(self):
"""Should be able to validate and get widget from widgetfield"""
register.widget(TestWidget)
testwidget = TestWidget()
testfield = WidgetField('Title', [TestWidgetSub])
# Create article and image for testing
article = DispatchTestHelpers.create_article(self.client)
image = DispatchTestHelpers.create_image(self.client)
field_data = {
'id': testwidget.id,
'data': {
'title': 'test title',
'description': 'test description',
'article': article.data['id'],
'image': image.data['id']
}
}
widget = testfield.get_widget(field_data['id'])
self.assertEqual(type(widget), type(testwidget))
def test_get_widget_from_widget_field_no_id(self):
"""Test the case where None id is passed to 'get_widget' """
register.widget(TestWidget)
testwidget = TestWidget()
testfield = WidgetField('Title', [TestWidgetSub])
field_data = {'id': None}
widget = testfield.get_widget(field_data['id'])
self.assertEqual(widget, None)
