def test_resolve_room_categories(self): a_floor = MagicMock() should_be_called = MagicMock(return_value = True) cats = MagicMock(return_value = "cats_result") class_ns = "__main__.DXFRoomCatsResolver" with patch(class_ns+".get_room_categories_dict", cats): with patch(class_ns+"._resolve_room_categories_for_floor", should_be_called): # Chiamata 1 - un solo floor, passato come parametro r = DXFRoomCatsResolver.resolve_room_categories(MagicMock(), a_floor) self.assertEqual(r, 1) MagicMock.assert_called_once_with(cats) MagicMock.assert_called_once_with( should_be_called, a_floor, "cats_result", ) MagicMock.reset_mock(should_be_called) b = { "dxf": { "floors": ["firstfloor", "secondfloor", "thirdfloor"] } } r = DXFRoomCatsResolver.resolve_room_categories(b) for f in b["dxf"]["floors"]: MagicMock.assert_any_call( should_be_called, f, "cats_result" ) self.assertEqual(r, 3)
def before_callback(b): DXFRoomIdsResolver.resolve_rooms_id(b, new_floor, source_name=None) DXFRoomCatsResolver.resolve_room_categories(b, new_floor) # Ensure floor merging is performed AFTER DXF Room_id resolution merged = b.attributes_for_source("merged") merged["floors"] = DataMerger.merge_floors( b.get("edilizia"), b.get("easyroom"), b.get("dxf"))
def before_callback(b): DXFRoomIdsResolver.resolve_rooms_id( b, new_floor, source_name = None ) DXFRoomCatsResolver.resolve_room_categories(b, new_floor) # Ensure floor merging is performed AFTER DXF Room_id resolution merged = b.attributes_for_source("merged") merged["floors"] = DataMerger.merge_floors( b.get("edilizia"), b.get("easyroom"), b.get("dxf") )