From 77c75efcf2cd3a4000a824c3a783bd9d212847c7 Mon Sep 17 00:00:00 2001 From: Tom Talbot Date: Tue, 10 Jun 2014 11:22:17 +0100 Subject: [PATCH] Fix embedly error tests --- wagtail/wagtailembeds/tests.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/wagtail/wagtailembeds/tests.py b/wagtail/wagtailembeds/tests.py index abb7bb0498..0fbaf5baf0 100644 --- a/wagtail/wagtailembeds/tests.py +++ b/wagtail/wagtailembeds/tests.py @@ -119,21 +119,30 @@ class TestEmbedly(TestCase): @unittest.skipIf(no_embedly, "Embedly is not installed") def test_embedly_403(self): with patch('embedly.Embedly.oembed') as oembed: - oembed.return_value['error_code'] = 403 + oembed.return_value = {'type': 'photo', + 'url': 'http://www.example.com', + 'error': True, + 'error_code': 403} self.assertRaises(AccessDeniedEmbedlyException, wagtail_embedly, 'http://www.example.com', key='foo') @unittest.skipIf(no_embedly, "Embedly is not installed") def test_embedly_404(self): with patch('embedly.Embedly.oembed') as oembed: - oembed.return_value['error_code'] = 404 + oembed.return_value = {'type': 'photo', + 'url': 'http://www.example.com', + 'error': True, + 'error_code': 404} self.assertRaises(EmbedNotFoundException, wagtail_embedly, 'http://www.example.com', key='foo') @unittest.skipIf(no_embedly, "Embedly is not installed") def test_embedly_other_error(self): with patch('embedly.Embedly.oembed') as oembed: - oembed.return_value['error_code'] = 999 + oembed.return_value = {'type': 'photo', + 'url': 'http://www.example.com', + 'error': True, + 'error_code': 999} self.assertRaises(EmbedlyException, wagtail_embedly, 'http://www.example.com', key='foo')