Python UpdateReservationPrices.select_cancellation_policy Examples

Example #1

File: test_usecase_update_reservation_prices.py Project: pmisters/django-code-example

def test_select_cancellation_policy_fail(service: UpdateReservationPrices,
                                         context: Context, house, rate_plan,
                                         reservation):
    service._policies_repo = Mock(get=Mock(return_value=Nothing))
    context.house = house
    context.source = reservation
    context.reservation_room = reservation.rooms[0]
    context.rate_plan = rate_plan

    result = service.select_cancellation_policy(context)
    assert is_successful(result)
    assert result.unwrap().rate_plan.policy is None

Example #2

File: test_usecase_update_reservation_prices.py Project: pmisters/django-code-example

def test_select_cancellation_policy_for_plan_without_it(
        service: UpdateReservationPrices, context: Context, house, rate_plan,
        reservation):
    service._policies_repo = Mock(get=Mock(side_effect=RuntimeError('ERR')))
    context.house = house
    context.source = reservation
    context.reservation_room = reservation.rooms[0]
    context.rate_plan = attr.evolve(rate_plan, policy_id=0)

    result = service.select_cancellation_policy(context)
    assert is_successful(result)
    assert result.unwrap().rate_plan.policy is None

Example #3

File: test_usecase_update_reservation_prices.py Project: pmisters/django-code-example

def test_select_cancellation_policy_error(service: UpdateReservationPrices,
                                          context: Context, house, rate_plan,
                                          reservation):
    service._policies_repo = Mock(get=Mock(side_effect=RuntimeError('ERR')))
    context.house = house
    context.source = reservation
    context.reservation_room = reservation.rooms[0]
    context.rate_plan = rate_plan

    result = service.select_cancellation_policy(context)
    assert not is_successful(result)
    assert result.failure().failure == ReservationErrors.error
    assert str(result.failure().exc) == 'ERR'