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)