def test_check_los_discounts(service: CalculateNewReservation,
context: Context, house, room_type, rate,
los_discount):
period = [
context.start_date,
context.start_date + datetime.timedelta(days=1),
context.start_date + datetime.timedelta(days=2),
context.start_date + datetime.timedelta(days=3),
]
context.end_date = context.start_date + datetime.timedelta(days=4)
context.house = house
context.room_type = room_type
context.rate = rate
context.prices = {
period[0]: Decimal(100),
period[1]: Decimal(100),
period[2]: Decimal(100),
period[3]: Decimal(100),
}
context.discounts = {DiscountTypes.LOS: [los_discount]}
result = service.check_los_discounts(context)
assert is_successful(result)
assert result.unwrap().prices == {
period[0]: Decimal(85),
period[1]: Decimal(85),
period[2]: Decimal(85),
period[3]: Decimal(85),
}
def test_missed_plan_id(service: CalculateNewReservation, context: Context,
house):
context.house = house
context.plan_id = None
result = service.select_rate_plan(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_rateplan
def test_select_rate_plan_ok(service: CalculateNewReservation,
context: Context, house, rate_plan):
service._prices_repo = Mock(get_plan=Mock(return_value=Some(rate_plan)))
context.house = house
result = service.select_rate_plan(context)
assert is_successful(result)
assert result.unwrap().rate_plan == rate_plan
def test_select_rate_plan_fail(service: CalculateNewReservation,
context: Context, house):
service._prices_repo = Mock(get_plan=Mock(return_value=Nothing))
context.house = house
result = service.select_rate_plan(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_rateplan
def test_select_room_type_ok(service: CalculateNewReservation,
context: Context, house, room_type):
service._roomtypes_repo = Mock(get=Mock(return_value=Some(room_type)))
context.house = house
result = service.select_room_type(context)
assert is_successful(result), result.failure()
assert result.unwrap().room_type == room_type
def test_select_rate_plan_error(service: CalculateNewReservation,
context: Context, house):
service._prices_repo = Mock(get_plan=Mock(side_effect=RuntimeError("ERR")))
context.house = house
result = service.select_rate_plan(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == "ERR"
def test_select_rates_ok(service: CalculateNewReservation, context: Context,
house, room_type, rate_plan, rate):
service._prices_repo = Mock(select_rates=Mock(return_value=[rate]))
context.house = house
context.room_type = room_type
context.rate_plan = rate_plan
result = service.select_rates(context)
assert is_successful(result)
assert result.unwrap().rates == [rate]
def test_select_rate_by_guest_count_fail(service: CalculateNewReservation,
context: Context, house, room_type,
rate_plan, rate):
context.house = house
context.room_type = room_type
context.rate_plan = rate_plan
context.rates = []
result = service.select_rate_by_guest_count(context)
assert is_successful(result)
assert result.unwrap().rate is None
def test_missed_rate_id(service: CalculateNewReservation, context: Context,
house, room_type, rate_plan, rate):
context.house = house
context.room_type = room_type
context.rate_plan = rate_plan
context.rate_id = None
context.rates = [rate]
result = service.select_rate_by_id(context)
assert is_successful(result)
assert result.unwrap().rate is None
def test_select_daily_prices_error(service: CalculateNewReservation,
context: Context, house, room_type, rate):
service._prices_repo = Mock(select_prices=Mock(
side_effect=RuntimeError("ERR")))
context.house = house
context.room_type = room_type
context.rate = rate
result = service.select_daily_prices(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == "ERR"
def test_select_rate_by_guest_count_only_less(service: CalculateNewReservation,
context: Context, house,
room_type, rate_plan, rate):
rate_with_1 = attr.evolve(rate, id=401, occupancy=1)
context.house = house
context.room_type = room_type
context.rate_plan = rate_plan
context.rates = [rate_with_1]
result = service.select_rate_by_guest_count(context)
assert is_successful(result)
assert result.unwrap().rate == rate_with_1
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_daily_prices_fail(service: CalculateNewReservation,
context: Context, house, room_type, rate):
service._prices_repo = Mock(select_prices=Mock(return_value={}))
context.house = house
context.room_type = room_type
context.rate = rate
result = service.select_daily_prices(context)
assert is_successful(result)
assert result.unwrap().prices == {
datetime.date.today(): None,
datetime.date.today() + datetime.timedelta(days=1): None,
}
def test_check_los_discounts_empty(service: CalculateNewReservation,
context: Context, house, room_type, rate):
period = [
context.start_date, context.start_date + datetime.timedelta(days=1)
]
context.house = house
context.room_type = room_type
context.rate = rate
context.prices = {period[0]: Decimal(100), period[1]: Decimal(110)}
context.discounts = {DiscountTypes.LOS: []}
result = service.check_los_discounts(context)
assert is_successful(result)
assert result.unwrap().prices == context.prices
def test_select_daily_prices_ok(service: CalculateNewReservation,
context: Context, house, room_type, rate):
prices = {
context.start_date: Decimal(100),
context.start_date + datetime.timedelta(days=1): Decimal(110)
}
service._prices_repo = Mock(select_prices=Mock(return_value=prices))
context.house = house
context.room_type = room_type
context.rate = rate
result = service.select_daily_prices(context)
assert is_successful(result)
assert result.unwrap().prices == prices
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
def test_select_discounts_no_rate(service: CalculateNewReservation,
context: Context, house, room_type,
rate_plan):
service._discounts_repo = Mock(select=Mock(
side_effect=RuntimeError("ERR")))
context.house = house
context.room_type = room_type
context.rate = None
context.rate_plan = rate_plan
context.prices = {
context.start_date: None,
context.start_date + datetime.timedelta(days=1): None
}
result = service.select_discounts(context)
assert is_successful(result)
assert result.unwrap().discounts == {}
def test_select_discounts_missed_rate_plan(service: CalculateNewReservation,
context: Context, house, room_type,
rate, rate_plan, last_min_discount):
service._discounts_repo = Mock(select=Mock(
return_value={DiscountTypes.LAST_MINUTE: [last_min_discount]}))
context.house = house
context.room_type = room_type
context.rate = rate
context.rate_plan = attr.evolve(rate_plan, id=999)
context.prices = {
context.start_date: Decimal(100),
context.start_date + datetime.timedelta(days=1): Decimal(110)
}
result = service.select_discounts(context)
assert is_successful(result)
assert result.unwrap().discounts == {DiscountTypes.LAST_MINUTE: []}
def test_select_discounts_error(service: CalculateNewReservation,
context: Context, house, room_type, rate,
rate_plan):
service._discounts_repo = Mock(select=Mock(
side_effect=RuntimeError("ERR")))
context.house = house
context.room_type = room_type
context.rate = rate
context.rate_plan = rate_plan
context.prices = {
context.start_date: Decimal(100),
context.start_date + datetime.timedelta(days=1): Decimal(110)
}
result = service.select_discounts(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == "ERR"
def test_check_availability_discounts(service: CalculateNewReservation,
context: Context, house, room_type, rate,
availability_discount):
period = [
context.start_date, context.start_date + datetime.timedelta(days=1)
]
context.house = house
context.room_type = room_type
context.rate = rate
context.prices = {period[0]: Decimal(100), period[1]: Decimal(110)}
context.discounts = {DiscountTypes.AVAILABILITY: [availability_discount]}
context.occupancies = {period[0]: 3, period[1]: 1}
result = service.check_availability_discounts(context)
assert is_successful(result)
assert result.unwrap().prices == {
period[0]: Decimal(85),
period[1]: Decimal(110)
}
def test_check_last_minute_discounts(service: CalculateNewReservation,
context: Context, house, room_type, rate,
last_min_discount):
period = [
context.start_date + datetime.timedelta(days=5),
context.start_date + datetime.timedelta(days=6)
]
context.house = house
context.room_type = room_type
context.rate = rate
context.prices = {period[0]: Decimal(100), period[1]: Decimal(110)}
context.discounts = {DiscountTypes.LAST_MINUTE: [last_min_discount]}
result = service.check_last_minute_discounts(context)
assert is_successful(result)
assert result.unwrap().prices == {
period[0]: Decimal(85),
period[1]: Decimal(110)
}
def test_check_min_prices(service: CalculateNewReservation, context: Context,
house, room_type, rate_plan, rate):
period = [
datetime.date.today(),
datetime.date.today() + datetime.timedelta(days=1)
]
context.house = house
context.room_type = room_type
context.rate = rate
context.prices = {period[0]: Decimal(80), period[1]: Decimal(110)}
context.price_restrictions = {
period[0]: Decimal(100),
period[1]: Decimal(100)
}
result = service.check_min_prices(context)
assert is_successful(result)
assert result.unwrap().prices == {
period[0]: Decimal(100),
period[1]: Decimal(110)
}
def test_select_price_restrictions_ok(service: CalculateNewReservation,
context: Context, house, room_type,
rate_plan, rate):
period = [
datetime.date.today(),
datetime.date.today() + datetime.timedelta(days=1)
]
service._prices_repo = Mock(select_restrictions=Mock(return_value={
period[0]: Decimal(50),
period[1]: Decimal(50)
}))
context.house = house
context.room_type = room_type
context.rate_plan = rate_plan
context.rate = rate
result = service.select_price_restrictions(context)
assert is_successful(result)
assert result.unwrap().price_restrictions == {
period[0]: Decimal(50),
period[1]: Decimal(50)
}