shanxia2.0/backend/tests/test_pl_delivery_times_aggregate.py

from datetime import date, datetime

from app.services.pl_plan_stop_service import collect_unique_delivery_date_strings


class _D:
    def __init__(self, delivery_time):
        self.delivery_time = delivery_time


def test_collect_unique_sorted_yyyy_mm_dd():
    rows = [
        _D(datetime(2026, 9, 14, 10, 0, 0)),
        _D(datetime(2026, 9, 12, 8, 0, 0)),
        _D(date(2026, 9, 12)),
        _D(None),
        _D(""),
        _D("2026-09-13"),
    ]
    assert collect_unique_delivery_date_strings(rows) == [
        "2026-09-12",
        "2026-09-13",
        "2026-09-14",
    ]


def test_collect_empty():
    assert collect_unique_delivery_date_strings([]) == []
    assert collect_unique_delivery_date_strings([_D(None)]) == []