from decimal import Decimal from app.domain.price import ( filter_prices_by_parcel_type, filter_and_sort_prices, filter_valid_prices, normalize_delivery_request, sort_prices_by_price, ) from app.schemas.request import DeliveryCalculationRequest, DeliveryEntity, ParcelType from app.schemas.response import DeliveryPrice def _make_price(**overrides) -> DeliveryPrice: payload = { "provider": "provider-a", "service_name": "standard", "price": Decimal("100.00"), "currency": "RUB", "delivery_days_min": 2, "delivery_days_max": 4, } payload.update(overrides) return DeliveryPrice.model_construct(**payload) def _make_request(**overrides) -> DeliveryCalculationRequest: payload = { "entity": DeliveryEntity.INDIVIDUAL, "from_city": 1, "to_city": 2, "weight_kg": 1.235, "length_cm": 10.04, "width_cm": 20.05, "height_cm": 30.06, } payload.update(overrides) return DeliveryCalculationRequest(**payload) def test_normalize_delivery_request_applies_rules_without_side_effects() -> None: request = _make_request() normalized = normalize_delivery_request(request, weight_round_scale=2) assert normalized is not request assert normalized.from_city == 1 assert normalized.to_city == 2 assert normalized.weight_kg == Decimal("1.24") assert normalized.length_cm == Decimal("10.0") assert normalized.width_cm == Decimal("20.1") assert normalized.height_cm == Decimal("30.1") assert request.from_city == 1 assert request.to_city == 2 def test_normalize_delivery_request_handles_normalization_boundaries() -> None: request = _make_request( weight_kg=0.0001, length_cm=0.01, width_cm=0.04, height_cm=0.15, ) low_scale = normalize_delivery_request(request, weight_round_scale=-5) high_scale = normalize_delivery_request(request, weight_round_scale=99) assert low_scale.weight_kg == Decimal("0.01") assert low_scale.length_cm == Decimal("0.1") assert low_scale.width_cm == Decimal("0.1") assert low_scale.height_cm == Decimal("0.2") assert high_scale.weight_kg == Decimal("0.01") def test_filter_valid_prices_handles_empty_input() -> None: assert filter_valid_prices([]) == [] assert filter_and_sort_prices([]) == [] def test_filter_prices_by_parcel_type_matches_document_markers_case_insensitively() -> None: document_ru = _make_price(service_name="Срочный Документ") document_en = _make_price(service_name="DOCUMENT EXPRESS") parcel = _make_price(service_name="Parcel locker") result = filter_prices_by_parcel_type( [document_ru, parcel, document_en], parcel_type=ParcelType.DOC, ) assert [price.service_name for price in result] == [ "Срочный Документ", "DOCUMENT EXPRESS", ] def test_filter_prices_by_parcel_type_returns_non_document_tariffs_for_parcel() -> None: document = _make_price(service_name="document delivery") parcel = _make_price(service_name="Economy parcel") standard = _make_price(service_name="Express") result = filter_prices_by_parcel_type( [document, parcel, standard], parcel_type=ParcelType.PARCEL, ) assert [price.service_name for price in result] == ["Economy parcel", "Express"] def test_filter_prices_by_parcel_type_keeps_prices_with_bypass_flag_regardless_of_type() -> None: bypass_document = _make_price( provider="cdek", service_name="Документ курьером", bypass_parcel_type_filter=True, ) bypass_parcel = _make_price( provider="cdek", service_name="Посылка склад-склад", bypass_parcel_type_filter=True, ) other_document = _make_price(provider="other", service_name="Document") other_parcel = _make_price(provider="other", service_name="Parcel") doc_result = filter_prices_by_parcel_type( [bypass_document, bypass_parcel, other_document, other_parcel], parcel_type=ParcelType.DOC, ) parcel_result = filter_prices_by_parcel_type( [bypass_document, bypass_parcel, other_document, other_parcel], parcel_type=ParcelType.PARCEL, ) assert [price.service_name for price in doc_result] == [ "Документ курьером", "Посылка склад-склад", "Document", ] assert [price.service_name for price in parcel_result] == [ "Документ курьером", "Посылка склад-склад", "Parcel", ] def test_filter_prices_by_parcel_type_returns_all_prices_when_type_is_missing() -> None: prices = [ _make_price(service_name="Документ"), _make_price(service_name="Parcel"), ] result = filter_prices_by_parcel_type(prices, parcel_type=None) assert result == prices def test_filter_valid_prices_applies_multiplier_and_rounds_half_up() -> None: low = _make_price(provider="cdek", price=Decimal("100.40")) high = _make_price(provider="boxberry", price=Decimal("100.50")) result = filter_valid_prices( [low, high], price_multiplier=Decimal("1.1"), ) assert [price.price for price in result] == [Decimal("110"), Decimal("111")] def test_filter_valid_prices_excludes_invalid_records() -> None: valid = _make_price(provider=" cdek ", service_name=" express ", currency="rub") invalid_provider = _make_price(provider=" ") invalid_price = _make_price(price=Decimal("-1.00")) invalid_currency = _make_price(currency="RUR1") invalid_range = _make_price(delivery_days_min=5, delivery_days_max=3) result = filter_valid_prices( [ valid, invalid_provider, invalid_price, invalid_currency, invalid_range, {"not": "a price"}, ] ) assert len(result) == 1 assert result[0].provider == "cdek" assert result[0].service_name == "express" assert result[0].currency == "RUB" def test_filter_valid_prices_excludes_non_finite_prices() -> None: valid = _make_price(provider="cdek", price=Decimal("199.00")) invalid_nan = _make_price(price=Decimal("NaN")) invalid_inf = _make_price(price=Decimal("Infinity")) invalid_neg_inf = _make_price(price=Decimal("-Infinity")) result = filter_valid_prices([invalid_nan, invalid_inf, invalid_neg_inf, valid]) assert len(result) == 1 assert result[0].provider == "cdek" def test_filter_valid_prices_excludes_none_provider_and_service() -> None: valid = _make_price(provider="cdek", service_name="express") invalid_provider_none = _make_price(provider=None) invalid_service_none = _make_price(service_name=None) result = filter_valid_prices([invalid_provider_none, invalid_service_none, valid]) assert len(result) == 1 assert result[0].provider == "cdek" assert result[0].service_name == "express" def test_filter_valid_prices_excludes_prices_rounded_to_zero_after_multiplier() -> None: low_price = _make_price(provider="cdek", price=Decimal("0.40")) result = filter_valid_prices( [low_price], price_multiplier=Decimal("0.5"), ) assert result == [] def test_sort_prices_by_price_keeps_equal_price_order_stable() -> None: price_a = _make_price(provider="a", price=Decimal("150.00")) price_b = _make_price(provider="b", price=Decimal("150.00")) price_c = _make_price(provider="c", price=Decimal("99.99")) sorted_prices = sort_prices_by_price([price_a, price_b, price_c]) assert [price.provider for price in sorted_prices] == ["c", "a", "b"] def test_filter_and_sort_prices_combines_domain_steps() -> None: price_a = _make_price(provider="a", price=Decimal("220.00")) price_b = _make_price(provider="b", price=Decimal("100.00")) invalid = _make_price(provider="", price=Decimal("50.00")) result = filter_and_sort_prices( [price_a, invalid, price_b], price_multiplier=Decimal("1.3"), ) assert [price.provider for price in result] == ["b", "a"] assert [price.price for price in result] == [Decimal("130"), Decimal("286")] def test_filter_and_sort_prices_applies_parcel_type_filter_before_sorting() -> None: document = _make_price(provider="a", price=Decimal("220.00"), service_name="Document") parcel = _make_price(provider="b", price=Decimal("100.00"), service_name="Parcel") document_ru = _make_price( provider="c", price=Decimal("150.00"), service_name="Документ курьером", ) result = filter_and_sort_prices( [document, parcel, document_ru], parcel_type=ParcelType.DOC, ) assert [price.provider for price in result] == ["c", "a"]