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)
