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 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 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_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 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(reservationId): result = ReservationTDG.find(reservationId) if not result: return None else: # must make a reference to timeslottable and create a timeslot object room = RoomMapper.find(result[0][1]) holder = UserMapper.find(result[0][3]) timeslot = TimeslotMapper.find(result[0][4]) equipment = EquipmentMapper.find(result[0][5]) return Reservation(room, holder, timeslot, result[0][2], equipment, result[0][0])
def findAll(): result = ReservationTDG.findAll() allReservations = [] if not result: return [] else: for index, r in enumerate(result): room = RoomMapper.find(result[index][1]) holder = UserMapper.find(result[index][3]) timeslot = TimeslotMapper.find(result[index][4]) equipment = EquipmentMapper.find(result[index][5]) reservation = Reservation(room, holder, timeslot, result[index][2], equipment, result[index][0]) allReservations.append(reservation) return allReservations
def test_find_found_in_id_map_found_in_DB(monkeypatch): # Test Data unexpected = Reservation('room', User('name', 'password'), 'time', 'description', Equipment("trcyvu"), 1) expected = Reservation('room1', User('name2', 'password3'), 'time1', 'description2', Equipment("fgvkgas"), 2) # Mock def id_find(_, __): return expected def tdg_find(_): return [[unexpected.getId(), unexpected.getRoom()]] monkeypatch.setattr(IdMap, 'find', id_find) monkeypatch.setattr(ReservationTDG, 'find', tdg_find) # Execute val = ReservationMapper.find("tvyub") # 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_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 makeNew(room, user, timeslot, description, equipment, reservationId): reservation = Reservation(room, user, timeslot, description, equipment, reservationId) UnitOfWork.registerNew(reservation) return reservation