zkxw/backend/tests/test_production_waste_output.py

"""生产废弃物列表：产出类型解析与关键字匹配语义。"""
import unittest

from app.crud.production_waste_output import (
    WASTE_OUTPUT_TYPE_CODE,
    WASTE_OUTPUT_TYPE_NAME,
    match_waste_output_keyword,
    pick_waste_output_type_ids,
)


class TestPickWasteOutputTypeIds(unittest.TestCase):
    def test_prefers_type_code_fqw(self):
        rows = [
            {"id": "a", "type_code": "YXW", "type_name": "有效物"},
            {"id": "b", "type_code": "FQW", "type_name": "废弃物"},
            {"id": "c", "type_code": "", "type_name": "废弃物"},
        ]
        self.assertEqual(pick_waste_output_type_ids(rows), {"b"})

    def test_fallback_type_name_when_no_fqw(self):
        rows = [
            {"id": "a", "type_code": "YXW", "type_name": "有效物"},
            {"id": "c", "type_code": "", "type_name": "废弃物"},
            {"id": "d", "type_code": None, "type_name": "废弃物"},
        ]
        self.assertEqual(pick_waste_output_type_ids(rows), {"c", "d"})

    def test_empty_when_no_match(self):
        self.assertEqual(pick_waste_output_type_ids([]), set())
        self.assertEqual(
            pick_waste_output_type_ids([{"id": "a", "type_code": "YXW", "type_name": "有效物"}]),
            set(),
        )


class TestMatchWasteOutputKeyword(unittest.TestCase):
    def test_matches_name_or_barcode(self):
        self.assertTrue(match_waste_output_keyword("废液", "废液A", "X1"))
        self.assertTrue(match_waste_output_keyword("X1", "废液", "BX1Y"))
        self.assertFalse(match_waste_output_keyword("zzz", "废液", "X1"))

    def test_blank_keyword_always_true(self):
        self.assertTrue(match_waste_output_keyword("", "废液", ""))
        self.assertTrue(match_waste_output_keyword(None, "", ""))


class TestConstants(unittest.TestCase):
    def test_constants(self):
        self.assertEqual(WASTE_OUTPUT_TYPE_CODE, "FQW")
        self.assertEqual(WASTE_OUTPUT_TYPE_NAME, "废弃物")


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