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), "")