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_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