def test_without_labels(self):
"""
Test: Call _get_categories_from_labels with a card with no labels.
Expected result: No categories will be returned.
"""
#Mock a card object:
class MockCard:
json_obj = {"labels" : []}
categories = logic._get_categories_from_labels(MockCard())
self.assertEqual(categories, [])
def test_without_labels(self):
"""
Test: Call _get_categories_from_labels with a card with labels.
Expected result: Categories will be returned both based on color and name.
"""
#Mock a card and label object:
name = "a name"
color = "a color"
class MockCard:
json_obj = {"labels" : [{"name" : name}, {"color" : color}]}
categories = logic._get_categories_from_labels(MockCard())
self.assertTrue(name in categories)
self.assertTrue(color in categories)
