zkxw/backend/tests/test_production_task_set_route_steps.py

import unittest

from app.crud.production_task_set import (
    collect_missing_route_ids,
    collect_route_ids_without_steps,
)


class TestCollectRouteIdsWithoutSteps(unittest.TestCase):
    def test_empty_when_all_routes_have_steps(self):
        items = [{"route_id": "r1"}, {"route_id": "r2"}]
        self.assertEqual(
            collect_route_ids_without_steps(items, {"r1", "r2"}),
            [],
        )

    def test_reports_routes_without_steps_deduped(self):
        items = [
            {"route_id": "r1"},
            {"route_id": "r2"},
            {"route_id": "r1"},
            {"route_id": None},
            {"route_id": ""},
        ]
        self.assertEqual(
            collect_route_ids_without_steps(items, {"r2"}),
            ["r1"],
        )

    def test_missing_route_ids_unchanged(self):
        items = [{"route_id": "gone"}, {"route_id": "ok"}]
        self.assertEqual(
            collect_missing_route_ids(items, {"ok"}),
            ["gone"],
        )


if __name__ == "__main__":
    unittest.main()