zkxw/backend/tests/test_qc_work_order_conclusion.py

"""质检工单列表：由任务结论派生工单级合格/不合格。"""

import unittest


class TestDeriveWorkOrderQcConclusion(unittest.TestCase):
    def test_empty_tasks(self):
        from app.services.qc_work_order_conclusion import derive_work_order_qc_conclusion

        self.assertEqual(derive_work_order_qc_conclusion([]), "")

    def test_any_fail_is_unqualified(self):
        from app.services.qc_work_order_conclusion import derive_work_order_qc_conclusion

        self.assertEqual(derive_work_order_qc_conclusion(["tg", "btg", ""]), "btg")

    def test_all_pass_is_qualified(self):
        from app.services.qc_work_order_conclusion import derive_work_order_qc_conclusion

        self.assertEqual(derive_work_order_qc_conclusion(["tg", "tg"]), "tg")

    def test_incomplete_returns_empty(self):
        from app.services.qc_work_order_conclusion import derive_work_order_qc_conclusion

        self.assertEqual(derive_work_order_qc_conclusion(["tg", ""]), "")
        self.assertEqual(derive_work_order_qc_conclusion(["tg", None]), "")

    def test_labels_qualified_unqualified(self):
        from app.services.qc_work_order_conclusion import (
            work_order_qc_conclusion_label,
        )

        self.assertEqual(work_order_qc_conclusion_label("tg"), "合格")
        self.assertEqual(work_order_qc_conclusion_label("btg"), "不合格")
        self.assertEqual(work_order_qc_conclusion_label(""), "")
        self.assertEqual(work_order_qc_conclusion_label("x"), "")


class TestBuildQcConclusionFields(unittest.TestCase):
    def test_build_fields(self):
        from app.services.qc_work_order_conclusion import build_qc_conclusion_fields

        self.assertEqual(
            build_qc_conclusion_fields(["tg", "tg"]),
            {"qcConclusion": "tg", "qcConclusionLabel": "合格"},
        )
        self.assertEqual(
            build_qc_conclusion_fields(["btg"]),
            {"qcConclusion": "btg", "qcConclusionLabel": "不合格"},
        )
        self.assertEqual(
            build_qc_conclusion_fields(["tg", ""]),
            {"qcConclusion": "", "qcConclusionLabel": ""},
        )


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