def test_success(service: UpdateReservationCache, house, reservation,
rate_plan, user):
service._houses_repo = Mock(get=Mock(return_value=Some(house)))
service._reservations_repo = Mock(select=Mock(return_value=[reservation]))
service._cache_repo = Mock(save=Mock(return_value=True),
delete=Mock(return_value=None))
service._prices_repo = Mock(select_plans=Mock(return_value=[rate_plan]))
service._members_repo = Mock(get_bot_user=Mock(return_value=Some(user)))
service.get_rpc_api = Mock(return_value=Mock(
select_invoices_for_order=Mock(return_value=[
Mock(id=987, amount_total=50.0, amount_residual=20.0)
])))
result = service.execute(house_id=house.id)
assert is_successful(result)
def test_save_cache_ok(service: UpdateReservationCache, context: Context,
house, room_type):
service._cache_repo = Mock(save=Mock(return_value=True))
context.house = house
context.cached_reservations = [
CachedReservation(
pk='111-1',
reservation_id=111,
grid='roomtype',
grid_id=room_type.id,
checkin=datetime.datetime.combine(datetime.date.today(),
datetime.time(14)),
checkout=datetime.datetime.combine(
datetime.date.today() + datetime.timedelta(days=1),
datetime.time(11)),
)
]
result = service.save_cache(context)
assert is_successful(result)
def test_save_cache_error(service: UpdateReservationCache, context: Context,
house, room_type):
service._cache_repo = Mock(save=Mock(side_effect=RuntimeError('ERR')))
context.house = house
context.cached_reservations = [
CachedReservation(
pk='111-1',
reservation_id=111,
grid='roomtype',
grid_id=room_type.id,
checkin=datetime.datetime.combine(datetime.date.today(),
datetime.time(14)),
checkout=datetime.datetime.combine(
datetime.date.today() + datetime.timedelta(days=1),
datetime.time(11)),
)
]
result = service.save_cache(context)
assert not is_successful(result)
assert result.failure().failure == ReservationErrors.error
assert str(result.failure().exc) == 'ERR'
