""" 标准库匹配规则单元测试 - 内存处理版本 测试文档中定义的5种匹配情况 """ import sys import os # 添加项目根目录到 Python 路径 current_dir = os.path.dirname(os.path.abspath(__file__)) project_root = os.path.dirname(os.path.dirname(current_dir)) if project_root not in sys.path: sys.path.insert(0, project_root) import pytest import asyncio from utils_test.standard_new_Test.standard_service import ( StandardRepository, StandardMatcher, StandardMatchingService, MatchResultCode, ValidityStatus, StandardRecord ) @pytest.fixture def mock_repository(): """创建模拟的数据仓库""" repo = StandardRepository() mock_data = [ # 情况1: 正常现行标准 {"id": 1, "standard_name": "铁路桥涵设计规范", "standard_number": "TB 10002-2017", "validity": "XH"}, {"id": 2, "standard_name": "铁路工程抗震设计规范", "standard_number": "GB 50111-2006", "validity": "XH"}, {"id": 3, "standard_name": "铁路混凝土工程施工质量验收标准", "standard_number": "TB 10424-2018", "validity": "XH"}, # 情况4: 不匹配（年份错误）- 输入2023，实际2024 {"id": 4, "standard_name": "公路水运危险性较大工程专项施工方案编制审查规程", "standard_number": "JT/T 1495-2024", "validity": "XH"}, # 情况2: 被替代（废止+有现行替代） {"id": 5, "standard_name": "起重机 钢丝绳 保养、维护、检验和报废", "standard_number": "GB/T 5972-2016", "validity": "FZ"}, {"id": 6, "standard_name": "起重机 钢丝绳 保养、维护、检验和报废", "standard_number": "GB/T 5972-2023", "validity": "XH"}, # 情况3: 废止无替代 {"id": 7, "standard_name": "缆索起重机", "standard_number": "GB/T 28756-2012", "validity": "FZ"}, {"id": 8, "standard_name": "电力高处作业防坠器", "standard_number": "DL/T 1147-2009", "validity": "FZ"}, ] repo.load_data(mock_data) return repo @pytest.fixture def matcher(mock_repository): """创建匹配器实例""" return StandardMatcher(mock_repository) @pytest.fixture def mock_service(): """创建使用Mock数据的服务""" service = StandardMatchingService(db_pool=None) # 同步初始化（测试时使用event loop） loop = asyncio.get_event_loop() loop.run_until_complete(service.initialize()) return service class TestStandardRepository: """测试数据仓库""" def test_load_data(self, mock_repository): """测试数据加载""" assert len(mock_repository.get_all_records()) == 8 def test_find_by_number_exact(self, mock_repository): """测试精确匹配标准号""" result = mock_repository.find_by_number_exact("TB 10002-2017") assert result is not None assert result.standard_name == "铁路桥涵设计规范" def test_find_by_name_exact(self, mock_repository): """测试精确匹配名称""" result = mock_repository.find_by_name_exact("铁路桥涵设计规范") assert result is not None assert result.standard_number == "TB 10002-2017" def test_find_current_by_name(self, mock_repository): """测试查询现行标准""" results = mock_repository.find_current_by_name("起重机 钢丝绳 保养、维护、检验和报废") assert len(results) == 1 assert results[0].standard_number == "GB/T 5972-2023" class TestStandardMatcher: """测试匹配器""" def test_case1_ok(self, matcher): """情况1: 状态正常""" result = matcher.match(1, "铁路桥涵设计规范", "TB 10002-2017") assert result.status_code == MatchResultCode.OK.value assert result.final_result == "无问题" def test_case2_substituted(self, matcher): """情况2: 被替代""" result = matcher.match(1, "起重机 钢丝绳 保养、维护、检验和报废", "GB/T 5972-2016") assert result.status_code == MatchResultCode.SUBSTITUTED.value assert result.substitute_number == "GB/T 5972-2023" assert "已废止" in result.final_result def test_case3_abolished(self, matcher): """情况3: 废止无替代""" result = matcher.match(1, "缆索起重机", "GB/T 28756-2012") assert result.status_code == MatchResultCode.ABOLISHED.value assert "无现行状态" in result.final_result def test_case4_mismatch(self, matcher): """情况4: 不匹配""" result = matcher.match(1, "公路水运危险性较大工程专项施工方案编制审查规程", "JT/T 1495-2023") assert result.status_code == MatchResultCode.MISMATCH.value assert result.substitute_number == "JT/T 1495-2024" def test_case5_not_found(self, matcher): """情况5: 不存在""" result = matcher.match(1, "不存在的标准", "GB/T 99999-9999") assert result.status_code == MatchResultCode.NOT_FOUND.value class TestStandardMatchingService: """测试服务类""" def test_check_single(self, mock_service): """测试单个检查""" result = mock_service.check_single(1, "铁路桥涵设计规范", "TB 10002-2017") assert result.status_code == MatchResultCode.OK.value def test_check_standards_batch(self, mock_service): """测试批量检查""" standards = [ {"standard_name": "铁路桥涵设计规范", "standard_number": "TB 10002-2017"}, {"standard_name": "缆索起重机", "standard_number": "GB/T 28756-2012"}, {"standard_name": "不存在的标准", "standard_number": "GB/T 99999-9999"}, ] results = mock_service.check_standards(standards) assert len(results) == 3 assert results[0].status_code == MatchResultCode.OK.value assert results[1].status_code == MatchResultCode.ABOLISHED.value assert results[2].status_code == MatchResultCode.NOT_FOUND.value class TestDocumentExamples: """测试文档中列出的7个示例""" @pytest.fixture def doc_matcher(self): """文档测试专用匹配器""" repo = StandardRepository() mock_data = [ {"id": 1, "standard_name": "铁路桥涵设计规范", "standard_number": "TB 10002-2017", "validity": "XH"}, {"id": 2, "standard_name": "铁路工程抗震设计规范", "standard_number": "GB 50111-2006", "validity": "XH"}, {"id": 3, "standard_name": "铁路混凝土工程施工质量验收标准", "standard_number": "TB 10424-2018", "validity": "XH"}, {"id": 4, "standard_name": "公路水运危险性较大工程专项施工方案编制审查规程", "standard_number": "JT/T 1495-2024", "validity": "XH"}, {"id": 5, "standard_name": "起重机 钢丝绳 保养、维护、检验和报废", "standard_number": "GB/T 5972-2016", "validity": "FZ"}, {"id": 6, "standard_name": "起重机 钢丝绳 保养、维护、检验和报废", "standard_number": "GB/T 5972-2023", "validity": "XH"}, {"id": 7, "standard_name": "缆索起重机", "standard_number": "GB/T 28756-2012", "validity": "FZ"}, {"id": 8, "standard_name": "电力高处作业防坠器", "standard_number": "DL/T 1147-2009", "validity": "FZ"}, ] repo.load_data(mock_data) return StandardMatcher(repo) def test_example_1_railway_bridge(self, doc_matcher): """（1）《铁路桥涵设计规范》(TB 10002-2017) - 正常""" result = doc_matcher.match(1, "铁路桥涵设计规范", "TB 10002-2017") assert result.status_code == MatchResultCode.OK.value def test_example_2_seismic_design(self, doc_matcher): """（2）《铁路工程抗震设计规范》(GB 50111-2006) - 正常""" result = doc_matcher.match(1, "铁路工程抗震设计规范", "GB 50111-2006") assert result.status_code == MatchResultCode.OK.value def test_example_3_concrete(self, doc_matcher): """（3）《铁路混凝土工程施工质量验收标准》(TB 10424-2018) - 正常""" result = doc_matcher.match(1, "铁路混凝土工程施工质量验收标准", "TB 10424-2018") assert result.status_code == MatchResultCode.OK.value def test_example_4_highway_mismatch(self, doc_matcher): """（4）年份不匹配 - 应为2024""" result = doc_matcher.match(1, "公路水运危险性较大工程专项施工方案编制审查规程", "JT/T 1495-2023") assert result.status_code == MatchResultCode.MISMATCH.value assert result.substitute_number == "JT/T 1495-2024" def test_example_5_crane_wire_substituted(self, doc_matcher): """（5）被替代 GB/T 5972-2016 -> GB/T 5972-2023""" result = doc_matcher.match(1, "起重机 钢丝绳 保养、维护、检验和报废", "GB/T 5972-2016") assert result.status_code == MatchResultCode.SUBSTITUTED.value assert result.substitute_number == "GB/T 5972-2023" def test_example_6_cable_crane_abolished(self, doc_matcher): """（6）废止无替代""" result = doc_matcher.match(1, "缆索起重机", "GB/T 28756-2012") assert result.status_code == MatchResultCode.ABOLISHED.value def test_example_7_safety_device_abolished(self, doc_matcher): """（7）废止无替代""" result = doc_matcher.match(1, "电力高处作业防坠器", "DL/T 1147-2009") assert result.status_code == MatchResultCode.ABOLISHED.value if __name__ == "__main__": pytest.main([__file__, "-v"])