def test_get_existing_tags_raises_missing(self):
repo = Mock()
repo.get_tags.return_value = [self.tag_1]
tag_getter = TagGetter(db_gateway=repo, presenter=Mock())
with self.assertRaises(ValueError) as context:
tag_getter.get_existing_tags(
names=[self.tag_1.name, self.tag_2.name])
self.assertEqual(str(context.exception),
"The following tag names don't exist: {'tag_2'}")
def test_get_existing_tags(self):
repo = Mock()
repo.get_tags.return_value = [self.tag_1]
tag_getter = TagGetter(db_gateway=repo, presenter=Mock())
tags = tag_getter.get_existing_tags(names=[self.tag_1.name])
self.assertEqual(tags, [self.tag_1])