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()
