def test_get_video_link(self): soup = get_local_soup('video_thumbnail_examples.html') articles = soup.find_all("article") expected = [ '/videos/chroniques/434958/speed-game-live-any-majora-s-mask-fini-en-moins-de-1h35.htm', '/videos/chroniques/433900/speed-game-hotline-miami-2-en-moins-de-40-minutes.htm', '/videos/435083/une-petite-dose-de-gameplay-pour-fallout-4.htm', '/videos/435143/rayman-adventures-les-10-premieres-minutes-de-gameplay.htm', '/videos/gaming-live/435104/batgirl-une-affaire-de-famille-qui-tourne-a-la-debandade.htm', '/gaming-live/0001/00014554/trauma-center-under-the-knife-nintendo-ds-video-3-00000258.htm', ] actual = [_get_video_link(article).get('href') for article in articles] self.assertEqual(actual, expected)
def test_get_video_link_no_link(self): article = BeautifulSoup("<article/>", 'html.parser') expected = None actual = _get_video_link(article) self.assertEqual(actual, expected)