def test_get_soup(self): self.assertIsInstance(get_soup('index', mock_client), BeautifulSoup) self.assertIsInstance(get_soup('flag', mock_client), BeautifulSoup)
def test_correct_output(self): # There is only one article about half staff flags self.assertListEqual([json.loads(message) for message in self.task.run()], [get_flag_data(get_soup('flag', mock_client))])
def setUp(self): self.maxDiff = None self.flag_soup = get_soup('flag', mock_client) self.not_flag_soup = get_soup('nonsense', mock_client)