wangwensheng/AM516
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: resolve shortage names from Git-synced Kingdee material cache

Browse Source
This commit is contained in:
ar51
2026-10-09 10:18:34 +08:00
parent ad7c45d2d6
commit cf31549415
12 changed files with 7813 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
tests/test_feishu_bot_bridge.py
View File

@@ -569,6 +569,58 @@ class UnavailableReasonTests(unittest.TestCase):
class ShortageMaterialNamesTests(unittest.TestCase):
def setUp(self) -> None:
patcher = mock.patch.object(bridge, "load_material_names", return_value={})
self.load_names = patcher.start()
self.addCleanup(patcher.stop)
def test_code_only_shortages_use_local_names_and_preserve_api_name_priority(self) -> None:
self.load_names.return_value = {"20.40.00001": "编码器", "20.50.00002": "轴承"}
lines = bridge.shortage_material_names_line([
{"缺料物料": "20.40.00001"},
{"缺料物料": " 20.50.00002 "},
{"缺料物料": "20.40.00001"},
{"缺料物料": "20.50.00002", "物料名称": "API更新后的轴承名称"},
])
self.assertEqual(lines, "- 缺料物料明细：编码器；轴承；API更新后的轴承名称")
self.load_names.assert_called_once_with()
self.assertNotIn("20.40", lines)
self.assertNotIn("20.50", lines)
def test_missing_or_invalid_cache_keeps_available_names_and_explains_failure(self) -> None:
for reason in ("本地物料名称缓存文件缺失", "本地物料名称缓存计数或完整性校验失败"):
with self.subTest(reason=reason):
self.load_names.side_effect = bridge.MaterialNameCacheError(reason)
line = bridge.shortage_material_names_line([
{"物料名称": "编码器"},
{"缺料物料": "20.50.00002"},
{"缺料物料": "20.50.00003"},
])
self.assertIn("编码器", line)
self.assertIn("2项缺料记录未找到物料名称", line)
self.assertIn(reason, line)
self.assertNotIn("20.50.", line)
def test_initial_and_continuation_code_only_shortages_resolve_offline(self) -> None:
request = bridge.RequestInput(model="eRob70H50I-FS-18CN[V5]", quantity=10)
self.load_names.return_value = {"20.40.00001": "编码器"}
def query(query_request: bridge.RequestInput) -> dict[str, object]:
payload = successful_payload()
payload["data"]["采购信息"]["缺料明细"] = [{"缺料物料": "20.40.00001"}]
return payload
for offset in (0, bridge.CANDIDATE_BATCH_SIZE):
with (
self.subTest(offset=offset),
mock.patch.object(bridge, "run_controlled_query", side_effect=query) as run_query,
mock.patch.object(bridge, "write_record"),
):
report = bridge.build_report(request, candidate_offset=offset)
self.assertEqual(report.count("- 缺料物料明细：编码器"), run_query.call_count)
self.assertNotIn("未找到物料名称", report)
self.assertNotIn("20.40.00001", report)
def test_all_shortage_names_are_shown_without_non_reducer_raw_fields(self) -> None:
payload = successful_payload()
payload["data"]["采购信息"]["缺料明细"] = [
@@ -601,7 +653,7 @@ class ShortageMaterialNamesTests(unittest.TestCase):
lines = bridge.lead_time_details(payload, requested_quantity=10)["evidence_lines"]
expected = "4项缺料记录未返回物料名称，需人工复核"
expected = "4项缺料记录未找到物料名称，请核对物料编码并更新本地名称缓存后复核"
if valid_name:
expected = f"轴承（另有{expected}）"
self.assertEqual(lines, [f"- 缺料物料明细：{expected}"])

110
tests/test_material_name_cache.py Normal file
View File

@@ -0,0 +1,110 @@
from __future__ import annotations
import hashlib
import importlib.util
import json
import tempfile
import unittest
from pathlib import Path
from types import SimpleNamespace
from unittest import mock
from shared.runtime import material_name_cache as cache
script_path = Path(__file__).resolve().parents[1] / "capabilities/am516-delivery-prediction/scripts/refresh_material_names.py"
spec = importlib.util.spec_from_file_location("refresh_material_names", script_path)
refresh = importlib.util.module_from_spec(spec)
spec.loader.exec_module(refresh)
class MaterialNameCacheTests(unittest.TestCase):
def setUp(self) -> None:
self.temp = tempfile.TemporaryDirectory()
self.addCleanup(self.temp.cleanup)
self.path = Path(self.temp.name) / "another-checkout/data/material_names.json"
cache.load_material_names.cache_clear()
self.addCleanup(cache.load_material_names.cache_clear)
def test_generated_cache_is_portable_and_contains_only_name_lookup_data(self) -> None:
rows = [
{"FNumber": "20.40.00001", "FName": "编码器", "FMaterialId": 123, "FSpecification": "PRIVATE_MARKER"},
{"FNumber": "20.40.00001", "FName": "编码器", "FMaterialId": 456},
{"FNumber": "20.50.00002", "FName": "轴承"},
]
payload = refresh.write_cache(rows, "2026-10-09T10:13:22+08:00", self.path)
self.assertEqual(cache.load_material_names(self.path), {"20.40.00001": "编码器", "20.50.00002": "轴承"})
self.assertEqual(payload["source"]["record_count"], 3)
self.assertEqual(payload["material_count"], 2)
content = self.path.read_text()
for unwanted in ("FMaterialId", "FSpecification", "PRIVATE_MARKER", "/Users/"):
self.assertNotIn(unwanted, content)
def test_empty_invalid_or_conflicting_refresh_preserves_existing_cache(self) -> None:
refresh.write_cache([{"FNumber": "A", "FName": "原物料"}], "2026-10-09T10:00:00+08:00", self.path)
previous = self.path.read_bytes()
for rows in (
[], [{"FNumber": "A", "FName": ""}],
[{"FNumber": "A", "FName": "名称1"}, {"FNumber": "A", "FName": "名称2"}],
):
with self.subTest(rows=rows):
with self.assertRaises(cache.MaterialNameCacheError):
refresh.write_cache(rows, "2026-10-09T11:00:00+08:00", self.path)
self.assertEqual(self.path.read_bytes(), previous)
def test_cache_integrity_is_checked_and_loaded_once_until_restart(self) -> None:
refresh.write_cache([{"FNumber": "A", "FName": "编码器"}], "2026-10-09T10:00:00+08:00", self.path)
self.assertEqual(cache.load_material_names(self.path)["A"], "编码器")
payload = json.loads(self.path.read_text())
payload["materials"]["A"] = "篡改名称"
self.path.write_text(json.dumps(payload))
self.assertEqual(cache.load_material_names(self.path)["A"], "编码器")
cache.load_material_names.cache_clear()
with self.assertRaisesRegex(cache.MaterialNameCacheError, "完整性校验失败"):
cache.load_material_names(self.path)
def test_reference_export_rejects_filtered_or_incomplete_source(self) -> None:
rows = [{"FNumber": "A", "FName": "编码器"}]
payload = {
"form_id": "BD_MATERIAL", "filter": "", "records": rows,
"record_count": len(rows), "generated_at": "2026-10-09T10:00:00+08:00",
"content_sha256": hashlib.sha256(json.dumps(rows, ensure_ascii=False, sort_keys=True, separators=(",", ":")).encode()).hexdigest(),
}
source = Path(self.temp.name) / "source.json"
source.write_text(json.dumps(payload))
self.assertEqual(refresh.read_reference(source)[0], rows)
for field, value in (("filter", "FForbidStatus='A'"), ("record_count", 2), ("content_sha256", "wrong")):
with self.subTest(field=field):
source.write_text(json.dumps({**payload, field: value}))
with self.assertRaises(cache.MaterialNameCacheError):
refresh.read_reference(source)
def test_live_refresh_pages_two_fields_only_and_rejects_repeated_pages(self) -> None:
full_page = [[f"code-{i:04}", "编码器"] for i in range(2000)]
query = mock.Mock(side_effect=[full_page, [["last-code", "轴承"]]])
client = SimpleNamespace(
LOCAL_CREDENTIALS_PATH=Path("credentials.local.json"),
build_opener=mock.Mock(return_value=object()), login=mock.Mock(),
execute_list_query=query,
)
fake_spec = SimpleNamespace(loader=SimpleNamespace(exec_module=mock.Mock()))
with (
mock.patch.object(refresh.importlib.util, "spec_from_file_location", return_value=fake_spec),
mock.patch.object(refresh.importlib.util, "module_from_spec", return_value=client),
mock.patch("builtins.print"),
):
rows, _ = refresh.query_materials(Path("/approved/source/91_Scripts/kingdee_openapi/kingdee_query.py"))
self.assertEqual(len(rows), 2001)
self.assertEqual([call.args[1].start_row for call in query.call_args_list], [0, 2000])
for call in query.call_args_list:
args = call.args[1]
self.assertEqual(args.fields, "FNumber,FName")
self.assertEqual(args.filter, "")
self.assertTrue(args.no_audit_log)
query.side_effect = [full_page, full_page]
with self.assertRaisesRegex(cache.MaterialNameCacheError, "分页重复"):
refresh.query_materials(Path("/approved/source/91_Scripts/kingdee_openapi/kingdee_query.py"))
if __name__ == "__main__":
unittest.main()