import unittest

from app.utils.lab_storage_code_task_set import (
    BC_MSC_BUILTIN_SET_NAME,
    HSC_BUILTIN_SET_NAME,
    UC_MSC_BUILTIN_SET_NAME,
    match_builtin_task_set_name_by_lab_storage_code,
)


class TestMatchBuiltinTaskSetName(unittest.TestCase):
    def test_bc_before_b(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("BC20260101"),
            BC_MSC_BUILTIN_SET_NAME,
        )

    def test_b_only(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("B20260101"),
            HSC_BUILTIN_SET_NAME,
        )

    def test_uc_msc(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("UC2026010"),
            UC_MSC_BUILTIN_SET_NAME,
        )

    def test_uc_case_insensitive(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("uc2026016"),
            UC_MSC_BUILTIN_SET_NAME,
        )

    def test_ubmc(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("UBMC001"),
            "UBMC冻存（内置模版）",
        )

    def test_pbmc(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("pbmc001"),
            "PBMC冻存（内置模版）",
        )

    def test_empty_and_unknown(self):
        self.assertIsNone(match_builtin_task_set_name_by_lab_storage_code(""))
        self.assertIsNone(match_builtin_task_set_name_by_lab_storage_code("XY123"))

    def test_bc_case_insensitive(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("bc99"),
            BC_MSC_BUILTIN_SET_NAME,
        )


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