def test_nonvideo_wrapper_classes(self, mock_find_embed): mock_find_embed.return_value = { "html": '<iframe src="www.example.com"></iframe>', "type": "not a video", } finder = CustomOEmbedFinder() finder.extra_wrapper_classes = ["video_wrapper_class"] result = finder.find_embed("www.example.com") soup = BeautifulSoup(result["html"], "html.parser") self.assertEqual(soup.find("div"), None)
def test_nonvideo_output_html(self, mock_find_embed): mock_find_embed.return_value = { "html": '<iframe src="www.example.com"></iframe>', "type": "not a video", "title": "Something else", } finder = CustomOEmbedFinder() finder.extra_classes = ["foo", "bar"] finder.extra_wrapper_classes = ["baz", "qux"] result = finder.find_embed("www.example.com") self.assertEqual( result["html"], '<iframe src="www.example.com"></iframe>', )