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 findAll():
result = RoomTDG.findAll()
rooms = []
if not result:
return
else:
for index, r in enumerate(result):
room = Room(r[0])
rooms.append(room)
return rooms
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 test_find_found_in_id_map_not_found_in_DB(monkeypatch):
# Test Data
expected = Room(110)
# Mock
def id_find(_, __):
return expected
def no_find(_):
return
monkeypatch.setattr(IdMap, 'find', id_find)
monkeypatch.setattr(RoomTDG, 'find', no_find)
# Execute
val = RoomMapper.find(1)
# Verify
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_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 makeNew(roomId):
room = Room(roomId)
UnitOfWork.registerNew(room)
return room
def delete(roomId):
UnitOfWork.registerDeleted(Room(roomId))
def find(roomId):
result = RoomTDG.find(roomId)
if not result:
return
else:
return Room(result[0][0])
def room_find(_):
return Room(1)
def rooms_found():
return [Room(1), Room(2), Room(3)]
