提交

backend/tests/test_lab_storage_code_task_set.py

@@ -0,0 +1,43 @@
+import unittest
+
+from app.utils.lab_storage_code_task_set import 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"),
+            "干细胞培养（内置模版）",
+        )
+
+    def test_b_only(self):
+        self.assertEqual(
+            match_builtin_task_set_name_by_lab_storage_code("B20260101"),
+            "造血干样本处理（内置模版）",
+        )
+
+    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"),
+            "干细胞培养（内置模版）",
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()