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