def test_get_tag_class(self):
"""Should return a css class when tag has a defined class"""
self.assertEqual(filters.get_tag_class("WIP"), "btn-warning")
self.assertEqual(filters.get_tag_class("dummy"), "btn-danger")
specific_tags = ["Node", "Node1", "NodeSpecial3", "Node*", "Node_"]
for specific_tag in specific_tags:
self.assertEqual(filters.get_tag_class(specific_tag), "btn-info")
def test_get_tag_class_no_class(self):
"""Should return an empty string when tag has no defined class"""
undefined_tags = ["foo", "Nod", "DUMMY", "Dummy", "wip", "WiP",
"node", "NoDe", "", "12", "", "_-_"]
for undefined_tag in undefined_tags:
self.assertEqual(filters.get_tag_class(undefined_tag), "")