diff --git a/tests/test_web.py b/tests/test_web.py index d72bec78..425bc151 100644 --- a/tests/test_web.py +++ b/tests/test_web.py @@ -1822,11 +1822,15 @@ class WebTest(TestCase): I hereby ☕ post """ - mock_get.return_value = requests_response(feed, headers={ - 'Content-Type': atom.CONTENT_TYPE, - 'Last-Modified': 'Sat, 01 Jan 2024 01:02:03 GMT', - 'ETag': '"abc123"', - }) + mock_get.side_effect = [ + requests_response(feed, headers={ + 'Content-Type': atom.CONTENT_TYPE, + 'Last-Modified': 'Sat, 01 Jan 2024 01:02:03 GMT', + 'ETag': '"abc123"', + }), + # fetch post to look for image + WEBMENTION_NO_REL_LINK, + ] got = self.post('/queue/poll-feed', data={'domain': 'user.com'}) self.assertEqual(200, got.status_code) @@ -1903,8 +1907,13 @@ class WebTest(TestCase): """ - mock_get.return_value = requests_response( - feed, headers={'Content-Type': rss.CONTENT_TYPE}) + mock_get.side_effect = [ + requests_response(feed, headers={'Content-Type': rss.CONTENT_TYPE}), + # fetch posts to look for images + WEBMENTION_NO_REL_LINK, + WEBMENTION_NO_REL_LINK, + WEBMENTION_NO_REL_LINK, + ] got = self.post('/queue/poll-feed', data={'domain': 'user.com'}) self.assertEqual(200, got.status_code) @@ -1964,8 +1973,12 @@ class WebTest(TestCase): I hereby ☕ post """ - mock_get.return_value = requests_response( - feed, headers={'Content-Type': 'application/xml; charset=utf-8'}) + mock_get.side_effect = [ + requests_response( + feed, headers={'Content-Type': 'application/xml; charset=utf-8'}), + # fetch post to look for image + WEBMENTION_NO_REL_LINK, + ] got = self.post('/queue/poll-feed', data={'domain': 'user.com'}) self.assertEqual(200, got.status_code) @@ -1990,8 +2003,11 @@ class WebTest(TestCase): I hereby ☕ post """ - mock_get.return_value = requests_response( - feed, headers={'Content-Type': atom.CONTENT_TYPE}) + mock_get.side_effect = [ + requests_response(feed, headers={'Content-Type': atom.CONTENT_TYPE}), + # fetch post to look for image + WEBMENTION_NO_REL_LINK, + ] got = self.post('/queue/poll-feed', data={'domain': 'user.com'}) self.assertEqual(200, got.status_code) @@ -2121,8 +2137,11 @@ class WebTest(TestCase): I hereby ☕ post """ - mock_get.return_value = requests_response( - feed, headers={'Content-Type': atom.CONTENT_TYPE}) + mock_get.side_effect = [ + requests_response(feed, headers={'Content-Type': atom.CONTENT_TYPE}), + # fetch post to look for image + WEBMENTION_NO_REL_LINK, + ] got = self.post('/queue/poll-feed', data={'domain': 'user.com'}) self.assertEqual(200, got.status_code) @@ -2263,8 +2282,12 @@ class WebTest(TestCase): """ - mock_get.return_value = requests_response( - feed, headers={'Content-Type': atom.CONTENT_TYPE}) + mock_get.side_effect = [ + requests_response(feed, headers={'Content-Type': atom.CONTENT_TYPE}), + # fetch posts to look for images + WEBMENTION_NO_REL_LINK, + WEBMENTION_NO_REL_LINK, + ] got = self.post('/queue/poll-feed', data={'domain': 'user.com'}) self.assertEqual(200, got.status_code)