Раис Юсупалиев
230 lines
7.5 KiB
Python
230 lines
7.5 KiB
Python
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_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"]
|