def test_get_room_categories_dict(self): with patch("__main__.RoomCategory.get_cat_dict", self.room_cats_dict_mock): cats_dict = DXFRoomCatsResolver.get_room_categories_dict() self.assertTrue("AUL01" in cats_dict) self.assertTrue("AUL03" in cats_dict) self.assertTrue("WC01" in cats_dict) self.assertEqual(cats_dict["AUL01"], "Aula") self.assertEqual(cats_dict["AUL03"], "Aula Informatica") self.assertEqual(cats_dict["WC01"], "WC") self.assertEqual(len(cats_dict.keys()), 3)
def setUp(self): self.room_categories = { "AUL01" : { "group_name" : "", "description" : "Aula", "scope" : "didactic" }, "AUL03" : { "group_name" : "", "description" : "Aula Informatica", "scope" : "didactic" }, "WC01" : { "group_name" : "Bagno", "description" : "WC", "scope" : "WC" } } self.room_cats_dict_mock = MagicMock(return_value= self.room_categories) with patch("__main__.RoomCategory.get_cat_dict", self.room_cats_dict_mock): self.cats = DXFRoomCatsResolver.get_room_categories_dict()