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 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())