def test_find_found_in_id_map_found_in_DB(monkeypatch): # Test Data unexpected = Waiting('room', User('name', 'password'), 'time', 'description', Equipment("trcyvu"), 1) expected = Waiting('room1', User('name2', 'password3'), 'time1', 'description2', Equipment("fgvkgas"), 2) # Mock def id_find(_, __): return expected def tdg_find(_): return [[ unexpected.getId(), unexpected.getRoom(), unexpected.getUser(), unexpected.getDescription(), unexpected.getTimeslot() ]] monkeypatch.setattr(IdMap, 'find', id_find) monkeypatch.setattr(WaitingTDG, 'find', tdg_find) # Execute val = WaitingMapper.find(1) # Verify assert (val.getTimeslot() is expected.getTimeslot()) assert (val.getRoom() is expected.getRoom()) assert (val.getUser() is expected.getUser()) assert (val.getDescription() is expected.getDescription()) assert (val.getId() is expected.getId())
def test_equipment_getting_number_of_equipment_needed(): equipment7 = Equipment("equipmentID_ibiubi", laptops=2, projectors=1, whiteboards=4) assert 7 == len(equipment7) equipment1 = Equipment("equipmentID", 0, 1, 0) assert 1 == len(equipment1) equipment4 = Equipment("equipmentID_12313", 1, 2, 1) assert 4 == len(equipment4)
def getAllEquipmentAvailableAtTimeslot(self, timeslot): """ :rtype: Equipment """ maxEquipmentAvailable = Equipment("equipmentID_uvu", laptops=MAX_EQUIPMENT_NUMBER, projectors=MAX_EQUIPMENT_NUMBER, whiteboards=MAX_EQUIPMENT_NUMBER) # iterate through reservations list, get total equipment already reserved for r in self.getAllReservationsForTimeslot(timeslot): for type, quantity in r.getEquipment().equipment.iteritems(): amountAvailable = maxEquipmentAvailable.equipment[type] maxEquipmentAvailable.equipment[type] = int(amountAvailable) - int(quantity) return maxEquipmentAvailable
def test_find_found_in_id_map_not_found_in_DB(monkeypatch): # Test Data expected = Waiting("room", User("mammad", "password"), Timeslot(1, 2, "date", 1, "mammad", "timeID_ibuon"), 'description', Equipment('equipment'), "waitingID_vubion") # Mock def id_find(_, __): return expected def no_find(_): return monkeypatch.setattr(IdMap, 'find', id_find) monkeypatch.setattr(WaitingTDG, 'find', no_find) # Execute val = WaitingMapper.find("waitingID_vubion") # Verify assert (val.getTimeslot() is expected.getTimeslot()) assert (val.getRoom() is expected.getRoom()) assert (val.getUser() is expected.getUser()) assert (val.getDescription() is expected.getDescription()) assert (val.getId() is expected.getId())
def find(equipmentId): result = EquipmentTDG.find(equipmentId) if not result: return else: return Equipment(result[0][0], result[0][1], result[0][2], result[0][3])
def reservation_not_found(_): room = Room(1) user = User('buddy', 'boy') time = Timeslot(1, 2, datetime(2020, 01, 01), '', 1, 1) equipment = Equipment("EquipmentID_iionask") return Reservation(room, user, time, 'description', equipment, 'test')
def makeNew(laptops, projectors, whiteboards, equipmentId): equipment = Equipment(equipmentId, laptops=laptops, projectors=projectors, whiteboards=whiteboards) UnitOfWork.registerNew(equipment) return equipment
def reservation_create(*args, **kwargs): room = Room(1) user = User('buddy', 'boy') time = Timeslot(1, 2, datetime(2020, 01, 01), 1, "userID_tyvub", "timeslotID_ugvhbjk") return Reservation(room, user, time, 'description', Equipment("equipmentID_yvhjb"), "reservationID_vghjbk")
def reservations_found(): room = Room(1) user = User('buddy', 'boy') time = Timeslot(1, 2, datetime(2020, 01, 01), 1, 'buddy', 'timeslotID_7g8hij') return [ Reservation(room, user, time, 'description', Equipment("equipmentID_ionoi"), "reservationID") ]
def find_by_room(*args, **kwargs): room = Room(1) user = User('buddy', 'boy') time = Timeslot(1, 2, datetime(2020, 01, 01), 1, "userID_vubin", "timeslotID_hbijkn") return [ Reservation(room, user, time, 'description', Equipment("equipmentID_uyvbin"), "reservationID_ygvuhjbk") ]
def find_by_room(*args, **kwargs): room = Room(1) user = User('buddy', 'boy') time = Timeslot(1, 2, datetime(2020, 01, 01), 1, "userID_ibun", "timeslotID_vuhbjk") return [ Reservation(room, user, time, 'description', Equipment("equipmentID_vguhbikjn"), "reservationID_tcytvuhb") ]
def find_by_user(*args, **kwargs): room = Room(1) user = User('buddy', 'boy') time = Timeslot(1, 2, datetime(2020, 01, 01), 1, "userID_bijknklm", "timeslotID_ghvjbk") return [ Reservation(room, user, time, 'description', Equipment("equipmentID_hgcvjb"), "reservationID_vuhbiuj") ]
def test_find_not_found_in_id_map_found_in_DB(monkeypatch): # Test Data expected_timeslot = Timeslot(1, 2, 'date', 1, 'usernameid', 'timeid') expected_room = Room(1) expected_user = User('abe', 3) expected_equipment = Equipment("equipmentID_uvibonpm") expected = Waiting(expected_room, expected_user, expected_timeslot, 'joe', expected_equipment, 234) # Mock def no_find(_, __): return def yes_find(_): return [[ expected.getId(), expected.getRoom(), expected.getUser(), expected.getDescription(), expected_equipment.getId(), expected.getTimeslot().getId() ]] def timeslot_find(_): return expected_timeslot def room_find(_): return expected_room def user_find(_): return expected_user monkeypatch.setattr(IdMap, 'find', no_find) monkeypatch.setattr(WaitingTDG, 'find', yes_find) monkeypatch.setattr(TimeslotMapper, 'find', timeslot_find) monkeypatch.setattr(RoomMapper, 'find', room_find) monkeypatch.setattr(UserMapper, 'find', user_find) # Execute val = WaitingMapper.find("iyubon") # Verify assert (val.getTimeslot() is expected_timeslot) assert (val.getRoom() is expected_room) assert (val.getUser() is expected_user) assert (val.getDescription() is expected.getDescription()) assert (val.getId() is expected.getId())
def test_find_found_in_id_map_not_found_in_DB(monkeypatch): # Test Data expected = Reservation('room', User('name', 'password'), 'time', 'description',Equipment("dxhtcfjygv"), 1) # Mock def id_find(_, __): return expected def no_find(_): return monkeypatch.setattr(IdMap, 'find', id_find) monkeypatch.setattr(ReservationTDG, 'find', no_find) # Execute val = ReservationMapper.find("vuybi") # Verify assert(val.getRoom() is expected.getRoom()) assert(val.getDescription() is expected.getDescription()) assert(val.getTimeslot() is expected.getTimeslot()) assert(val.getId() is expected.getId())
def test_find_not_found_in_id_map_found_in_DB(monkeypatch): # Test Data expected_timeslot = Timeslot(1, 2, 'date', 1, 'usernameid','timeid') expected_room = Room(1) expected = Reservation(expected_room, User('name', 'password'), expected_timeslot, 'description', Equipment("3w4ex5rcy"), "reservationID####") # Mock def no_find(_, __): return def yes_find(_): return [[expected.getId(), expected_room.getId(), expected.getDescription(), expected.getUser().getId(), expected.getEquipment().getId(), expected_timeslot.getId()]] def timeslot_find(_): return expected_timeslot def room_find(_): return expected_room monkeypatch.setattr(IdMap, 'find', no_find) monkeypatch.setattr(ReservationTDG, 'find', yes_find) monkeypatch.setattr(TimeslotMapper, 'find', timeslot_find) monkeypatch.setattr(RoomMapper, 'find', room_find) # Execute val = ReservationMapper.find("trcyv") # Verify assert(val.getRoom().getId() is expected_room.getId()) assert(val.getDescription() is expected.getDescription()) assert(val.getTimeslot().getId() is expected_timeslot.getId()) assert(val.getId() is expected.getId())
def delete(equipmentId): UnitOfWork.registerDeleted(Equipment(equipmentId, 0, 0, 0))
def test_equipment_with_no_arguments_is_zero_length(): equipment = Equipment("equipmentID_uybino") assert 0 == len(equipment)