def test_select_occupancy_no_discounts(service: CalculateNewReservation,
context: Context, house, room_type,
rate):
service._occupancy_repo = Mock(get=Mock(side_effect=RuntimeError("ERR")))
context.house = house
context.room_type = room_type
context.rate = rate
context.discounts = {DiscountTypes.AVAILABILITY: []}
result = service.select_occupancies(context)
assert is_successful(result)
assert result.unwrap().occupancies == {}
def test_select_occupancy_error(service: CalculateNewReservation,
context: Context, house, room_type, rate,
availability_discount):
service._occupancy_repo = Mock(get=Mock(side_effect=RuntimeError("ERR")))
context.house = house
context.room_type = room_type
context.rate = rate
context.discounts = {DiscountTypes.AVAILABILITY: [availability_discount]}
result = service.select_occupancies(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == "ERR"
def test_select_occupancy_ok(service: CalculateNewReservation,
context: Context, house, room_type, rate,
availability_discount):
occupancies = {
context.start_date: 1,
context.start_date + datetime.timedelta(days=1): 2
}
service._occupancy_repo = Mock(get=Mock(return_value=occupancies))
context.house = house
context.room_type = room_type
context.rate = rate
context.discounts = {DiscountTypes.AVAILABILITY: [availability_discount]}
result = service.select_occupancies(context)
assert is_successful(result)
assert result.unwrap().occupancies == occupancies