def test_select_house_fail(service: AttachContactToReservation,
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
def test_select_house_ok(service: AttachContactToReservation, context: Context,
house):
service._houses_repo = Mock(get=Mock(return_value=Some(house)))
result = service.select_house(context)
assert is_successful(result), result.failure()
assert result.unwrap().house == house
def test_check_reservation_ok(service: AttachContactToReservation,
context: Context, house, reservation):
context.house = house
context.reservation = reservation
result = service.check_reservation(context)
assert is_successful(result)
def test_missed_house_id(service: AttachContactToReservation,
context: Context):
context.house_id = None
result = service.select_house(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_house
def test_select_reservation_ok(service: RegisterReservationInOdoo, 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().reservation == reservation
def test_check_reservation_ok(service: CancelReservation, context: Context,
house, reservation):
context.house = house
context.source = reservation
result = service.check_reservation(context)
assert is_successful(result), result.failure()
def test_check_reservation_verified(service: AcceptReservationChanges,
context: Context, house, reservation):
context.house = house
context.source = attr.evolve(reservation, is_verified=True)
result = service.check_reservation(context)
assert is_successful(result)
def test_select_reservation_missed_pk(service: RegisterReservationInOdoo, context: Context, house):
context.house = house
context.pk = None
result = service.select_reservation(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_reservation
def test_success(service: ShowPricesForm, house, reservation, user, rate_plan,
room_type, room):
policy_1 = Policy(id=101, house_id=house.id, name='P1')
policy_2 = Policy(id=102, house_id=house.id, name='P2')
service._houses_repo = Mock(get=Mock(return_value=Some(house)))
service._reservations_repo = Mock(get=Mock(return_value=Some(reservation)))
service._prices_repo = Mock(select_plans=Mock(return_value=[rate_plan]))
service._roomtypes_repo = Mock(select=Mock(return_value=[room_type]))
service._rooms_repo = Mock(select=Mock(return_value=[room]))
service._policies_repo = Mock(select=Mock(
return_value=[policy_1, policy_2]))
result = service.execute(house.id, reservation.id, reservation.rooms[0].id,
user)
assert is_successful(result)
ctx = result.unwrap()
assert isinstance(ctx, ReservationPricesUpdateContext)
assert ctx.house == house
assert ctx.reservation == reservation
assert ctx.reservation_room == reservation.rooms[0]
assert ctx.rate_plans == [rate_plan]
assert ctx.room_types == [room_type]
assert ctx.rooms == [room]
assert ctx.policies == [policy_1, policy_2]
def test_select_house_ok(service: AcceptReservationChanges, 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_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_room_type(service: CreateReservation, context: Context, house):
context.house = house
context.request.roomtype_id = None
result = service.select_room_type(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_roomtype
def test_success(service: CreateReservation, context: Context, house,
room_type, rate_plan, rate, cancellation_policy):
reservation = Reservation(
id=111,
house_id=house.id,
connection_id=None,
source=ReservationSources.MANUAL,
channel=None,
channel_id='',
checkin=context.request.checkin,
checkout=context.request.checkout,
booked_at=timezone.now(),
status=ReservationStatuses.HOLD,
is_verified=True,
)
service._houses_repo = Mock(get=Mock(return_value=Some(house)))
service._roomtypes_repo = Mock(get=Mock(return_value=Some(room_type)))
service._prices_repo = Mock(get_plan=Mock(return_value=Some(rate_plan)),
select_rates=Mock(return_value=[rate]))
service._reservations_repo = Mock(
save=Mock(
return_value=(Some(attr.evolve(reservation, is_verified=False)),
True)),
accept=Mock(return_value=Some(reservation)),
)
service._policies_repo = Mock(get=Mock(
return_value=Some(cancellation_policy)))
result = service.execute(house.id, context.request, context.user)
assert is_successful(result), result.failure()
assert result.unwrap() == attr.evolve(reservation, is_verified=True)
def get(self, object_id, survey_id):
astro_object = self._get_object(object_id, survey_id)
if is_successful(astro_object):
aid = astro_object.unwrap()["aid"]
non_detections = self._get_non_detections(aid)
if is_successful(non_detections):
return non_detections
else:
return Failure(
ClientErrorException(
ObjectNotFound(object_id=object_id,
survey_id=survey_id)))
else:
return astro_object
def test_check_reservation_ok(service: AcceptReservationChanges,
context: Context, house, reservation):
context.house = house
context.source = reservation
result = service.check_reservation(context)
assert is_successful(result)
def test_select_house_error(service: RegisterReservationInOdoo, 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_path_to_missing_key_is_ok():
"""Test that missing keys are okay."""
dictionary = {'key': ['val1']}
test = ['val2', dictionary, ['bob']]
t_result = set_value_in_dict(*test)
assert is_successful(t_result)
assert dictionary['bob'] == 'val2' # type: ignore
def test_select_reservation_fail(service: RegisterReservationInOdoo, context: Context, house):
service._reservations_repo = Mock(get=Mock(return_value=Nothing))
context.house = house
result = service.select_reservation(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.missed_reservation
def get(self, object_id, survey_id):
astro_object = self._get_object(object_id, survey_id)
if is_successful(astro_object):
aid = astro_object.unwrap()["aid"]
detections = self._get_detections(aid)
if is_successful(detections) and len(detections.unwrap()) > 0:
return detections
else:
raise Failure(
ClientErrorException(
ObjectNotFound(object_id=object_id,
survey_id=self.survey_id)))
else:
return astro_object
def test_ioresulte():
"""Ensures that IOResultE correctly typecast."""
container: IOResult[float, Exception] = _function(1)
assert container == IOSuccess(10.0)
container = _function(0)
assert is_successful(container) is False
def test_custom_readerresult_types_resolve(
thing: CustomReaderResult, ) -> None:
"""Ensures that type aliases are resolved correctly."""
real_result = thing(RequiresContextResultE.empty)
if is_successful(real_result):
assert isinstance(real_result.unwrap(), int)
else:
assert isinstance(real_result.failure(), str)
def test_select_houses_all_ok(service, context, house):
house2 = attr.evolve(house, id=111)
service._houses_repo = Mock(select=Mock(return_value=[house, house2]))
context.house_id = None
result = service.select_houses(context)
assert is_successful(result)
assert result.unwrap().houses == [house, house2]
def test_invalid_alias_required(iterable_validator):
"""Test that alias key is required."""
test: dict = {'path': ['date']}
validate_result = iterable_validator(test)
assert not is_successful(validate_result)
assert isinstance(validate_result.failure(), ValueError)
assert isinstance(validate_result.failure().args[0][0], ValidationError)
assert "'alias' is a required property" in str(validate_result.failure())
def test_select_roomtypes_one_ok(service, context, house, user, room_type):
service._roomtypes_repo = Mock(get=Mock(return_value=Some(room_type)))
context.houses = [house]
context.users = {house.company.id: user}
result = service.select_roomtypes(context)
assert is_successful(result)
assert result.unwrap().room_types == {house.id: [room_type]}
def test_select_houses_all_error(service, context):
service._houses_repo = Mock(select=Mock(side_effect=RuntimeError("ERR")))
context.house_id = None
result = service.select_houses(context)
assert not is_successful(result)
assert result.failure().failure == OccupancyErrors.error
assert str(result.failure().exc) == "ERR"
def test_select_roomtypes_one_fail(service, context, house, user):
service._roomtypes_repo = Mock(get=Mock(return_value=Nothing))
context.houses = [house]
context.users = {house.company.id: user}
result = service.select_roomtypes(context)
assert not is_successful(result)
assert result.failure().failure == OccupancyErrors.missed_roomtype
def test_select_roomtypes_no_houses(service, context):
context.houses = []
context.users = {}
context.roomtype_id = None
result = service.select_roomtypes(context)
assert is_successful(result)
assert result.unwrap().room_types == {}
def test_select_room_types_ok(service: UpdateReservationPrices,
context: Context, house, room_type):
service._roomtypes_repo = Mock(select=Mock(return_value=[room_type]))
context.house = house
result = service.select_room_types(context)
assert is_successful(result)
assert result.unwrap().room_types == {room_type.id: room_type}
def test_select_rate_plan_ok(service: UpdateReservationPrices,
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: UpdateReservationPrices,
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
