diff --git a/wagtail/wagtailembeds/tests.py b/wagtail/wagtailembeds/tests.py index b6eb2a91b..d49c52420 100644 --- a/wagtail/wagtailembeds/tests.py +++ b/wagtail/wagtailembeds/tests.py @@ -290,6 +290,16 @@ class TestEmbedFilter(TestCase): self.assertEqual(result, '') + @unittest.expectedFailure + @patch('wagtail.wagtailembeds.embeds.get_embed') + def test_catches_embed_not_found(self, get_embed): + get_embed.side_effect = EmbedNotFoundException + + temp = template.Template('{% load wagtailembeds_tags %}{{ "http://www.youtube.com/watch/"|embed }}') + result = temp.render(template.Context()) + + self.assertEqual(result, '') + class TestEmbedBlock(TestCase): def test_deserialize(self): @@ -460,6 +470,18 @@ class TestMediaEmbedHandler(TestCase): self.assertIn('

Author: test author name

', result) self.assertIn('test title', result) + @unittest.expectedFailure + @patch('wagtail.wagtailembeds.embeds.get_embed') + def test_test_expand_db_attributes_for_editor_catches_embed_not_found(self, get_embed): + get_embed.side_effect = EmbedNotFoundException + + result = MediaEmbedHandler.expand_db_attributes( + {'url': 'http://www.youtube.com/watch/'}, + True + ) + + self.assertEqual(result, '') + @patch('wagtail.wagtailembeds.embeds.get_embed') def test_expand_db_attributes(self, get_embed): get_embed.return_value = Embed( @@ -480,3 +502,14 @@ class TestMediaEmbedHandler(TestCase): False ) self.assertIn('test html', result) + + @patch('wagtail.wagtailembeds.embeds.get_embed') + def test_expand_db_attributes_catches_embed_not_found(self, get_embed): + get_embed.side_effect = EmbedNotFoundException + + result = MediaEmbedHandler.expand_db_attributes( + {'url': 'http://www.youtube.com/watch/'}, + False + ) + + self.assertEqual(result, '')