def test_plugin_returns_full_image_if_set_in_settings(self): picture_plugin = PicturePlugin() mock_picture = MagicMock() mock_picture.image = MagicMock() image_url = '/path/to/full/image.png' mock_picture.image.url = image_url self.assertEqual(image_url, picture_plugin.icon_src(mock_picture))
def test_plugin_returns_icon_by_default(self): if hasattr(settings, 'PICTURE_FULL_IMAGE_AS_ICON'): del settings.PICTURE_FULL_IMAGE_AS_ICON picture_plugin = PicturePlugin() mock_picture = Mock(spec=Picture) self.assertEqual(settings.STATIC_URL + u"cms/img/icons/plugins/image.png", picture_plugin.icon_src(mock_picture))