def test_missed_room(service: UpdateRoomClose, context: Context, house):
context.house = house
context.room_id = None
result = service.select_room(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_room
def test_check_reservation_ok(service: UpdateRoomClose, context: Context,
house, reservation):
context.house = house
context.source = reservation
result = service.check_reservation(context)
assert is_successful(result)
def test_check_room_is_free_ok(service: UpdateRoomClose, context: Context,
house, room, reservation):
service._reservations_repo = Mock(is_room_busy=Mock(return_value=False))
context.house = house
context.room = room
context.source = reservation
result = service.check_room_is_free(context)
assert is_successful(result), result.failure()
def test_check_reservation_not_close_room(service: UpdateRoomClose,
context: Context, house,
reservation):
context.house = house
context.source = attr.evolve(reservation, status=ReservationStatuses.NEW)
result = service.check_reservation(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_reservation
def test_accept_reservation_fail(service: UpdateRoomClose, context: Context,
house, room, reservation):
service._reservations_repo = Mock(accept=Mock(return_value=Nothing))
context.house = house
context.room = room
context.reservation = reservation
result = service.accept_reservation(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.save
assert result.failure().error.startswith("Error accept Reservation ID=")
def test_accept_reservation_ok(service: UpdateRoomClose, context: Context,
house, room, reservation):
reservation = attr.evolve(reservation, is_verified=True)
service._reservations_repo = Mock(accept=Mock(
return_value=Some(reservation)))
context.house = house
context.room = room
context.reservation = attr.evolve(reservation, is_verified=False)
result = service.accept_reservation(context)
assert is_successful(result)
assert result.unwrap().reservation == reservation
def test_accept_reservation_error(service: UpdateRoomClose, context: Context,
house, room, reservation):
service._reservations_repo = Mock(accept=Mock(
side_effect=RuntimeError("ERR")))
context.house = house
context.room = room
context.reservation = reservation
result = service.accept_reservation(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == "ERR"
def test_check_room_is_free_error(service: UpdateRoomClose, context: Context,
house, room, reservation):
service._reservations_repo = Mock(is_room_busy=Mock(
side_effect=RuntimeError("ERR")))
context.house = house
context.room = room
context.source = reservation
result = service.check_room_is_free(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == "ERR"
def test_save_reservation_ok(service: UpdateRoomClose, context: Context, house,
room, reservation):
_reservation = attr.evolve(reservation,
close_reason=RoomCloseReasons.TMP_CLOSE)
service._reservations_repo = Mock(save=Mock(
return_value=(Some(_reservation), True)))
context.house = house
context.room = room
context.reservation = reservation
result = service.save_reservation(context)
assert is_successful(result)
assert result.unwrap().reservation == _reservation
def test_select_room_fail(service: UpdateRoomClose, context: Context, house):
service._rooms_repo = Mock(get=Mock(return_value=Nothing))
context.house = house
result = service.select_room(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_room
def test_select_reservation_ok(service: UpdateRoomClose, context: Context,
house, reservation):
service._reservations_repo = Mock(get=Mock(return_value=Some(reservation)))
context.house = house
result = service.select_reservation(context)
assert is_successful(result)
assert result.unwrap().source == reservation
def test_select_room_ok(service: UpdateRoomClose, context: Context, house,
room):
service._rooms_repo = Mock(get=Mock(return_value=Some(room)))
context.house = house
result = service.select_room(context)
assert is_successful(result)
assert result.unwrap().room == room
def test_select_room_error(service: UpdateRoomClose, context: Context, house):
service._rooms_repo = Mock(get=Mock(side_effect=RuntimeError("ERR")))
context.house = house
result = service.select_room(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == "ERR"
def context(house, room, user) -> Context:
return Context(
pk=111,
house_id=house.id,
room_id=room.id,
start_date=datetime.date.today(),
end_date=datetime.date.today() + datetime.timedelta(days=2),
reason=RoomCloseReasons.MAINTENANCE,
user=user,
notes="Comment",
)
def test_make_reservation_from_data_change_period(service: UpdateRoomClose,
context: Context, house,
room, reservation):
context.house = house
context.room = room
context.source = reservation
context.start_date = reservation.checkin + datetime.timedelta(days=10)
context.end_date = reservation.checkout + datetime.timedelta(days=10)
result = service.make_reservation_from_data(context)
assert is_successful(result), result.failure()
_reservation = result.unwrap().reservation
assert _reservation.id == reservation.id
assert _reservation.checkin == context.start_date
assert _reservation.checkout == context.end_date
assert _reservation.status == ReservationStatuses.CLOSE
assert _reservation.close_reason == RoomCloseReasons.MAINTENANCE
assert len(_reservation.rooms) == 1
_room = _reservation.rooms[0]
assert _room.id == reservation.rooms[0].id
assert _room.checkin == context.start_date
assert _room.checkout == context.end_date
assert _room.notes_info == "Comment"
assert len(_room.day_prices) == 2
price = _room.day_prices[0]
assert price.id is None
assert price.day == context.start_date
assert price.roomtype_id == room.roomtype_id
assert price.room_id == room.id
price = _room.day_prices[1]
assert price.id is None
assert price.day == context.start_date + datetime.timedelta(days=1)
assert price.roomtype_id == room.roomtype_id
assert price.room_id == room.id
