def test_get_tags_no_data(self):
tag_getter = TagGetter(db_gateway=Mock(), presenter=Mock())
tag_getter.repo.get_tags.return_value = []
tag_df = tag_getter._get_tags()
self.assertTrue(tag_df.empty)
self.assertEqual(['tag', 'tag_id'], tag_df.columns.to_list())
def test_get_tags(self):
tag_getter = TagGetter(db_gateway=Mock(), presenter=Mock())
tag_getter.repo.get_tags.return_value = [self.tag_1, self.tag_2]
expected_df = pd.DataFrame(data={
'tag': ['tag_1', 'tag_2'],
'tag_id': [None, None]
})
tag_df = tag_getter._get_tags()
assert_frame_equal(expected_df, tag_df)