def test_success(
    service: UpdateReservationPrices,
    context: Context,
    house,
    reservation,
    rate_plan,
    room_type,
    room,
    user,
    cancellation_policy,
):
    _reservation = attr.evolve(reservation, price_accepted=Decimal(3600))

    service._houses_repo = Mock(get=Mock(return_value=Some(house)))
    service._roomtypes_repo = Mock(select=Mock(return_value=[room_type]))
    service._rooms_repo = Mock(select=Mock(return_value=[room]))
    service._prices_repo = Mock(get_plan=Mock(return_value=Some(rate_plan)))
    service._reservations_repo = Mock(
        get=Mock(return_value=Some(reservation)),
        save=Mock(return_value=(Some(_reservation), True)))
    service._policies_repo = Mock(get=Mock(
        return_value=Some(cancellation_policy)))

    result = service.execute(house.id,
                             reservation.id,
                             reservation.rooms[0].id,
                             rate_plan.id,
                             user,
                             prices=context.prices)
    assert is_successful(result)
    assert result.unwrap() == _reservation
def test_select_house_ok(service: UpdateReservationPrices, context: Context,
                         house):
    service._houses_repo = Mock(get=Mock(return_value=Some(house)))

    result = service.select_house(context)
    assert is_successful(result)
    assert result.unwrap().house == house
def test_select_house_error(service: UpdateReservationPrices,
                            context: Context):
    service._houses_repo = Mock(get=Mock(side_effect=RuntimeError('ERR')))

    result = service.select_house(context)
    assert not is_successful(result)
    assert result.failure().failure == ReservationErrors.error
    assert str(result.failure().exc) == 'ERR'
def test_select_house_fail(service: UpdateReservationPrices, context: Context):
    service._houses_repo = Mock(get=Mock(return_value=Nothing))

    result = service.select_house(context)
    assert not is_successful(result)
    assert result.failure().failure == ReservationErrors.missed_house