zkxw/backend/tests/test_collection_regions_by_staff_user.py

import unittest

from app.crud.collection_region import map_staff_region_rows_to_list


class RegionsByStaffUserTests(unittest.TestCase):
    def test_map_preserves_order_and_fields(self):
        rows = [
            ("r2", "华东"),
            ("r1", "华南"),
        ]
        out = map_staff_region_rows_to_list(rows)
        self.assertEqual(
            out,
            [{"id": "r2", "regionName": "华东"}, {"id": "r1", "regionName": "华南"}],
        )

    def test_map_skips_empty_id(self):
        rows = [("", "无"), ("r1", "华南")]
        out = map_staff_region_rows_to_list(rows)
        self.assertEqual(out, [{"id": "r1", "regionName": "华南"}])


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