test: 添加资产和运行服务测试用例

server/tests/test_agent_asset_spreadsheet_import.py

@@ -0,0 +1,41 @@
+from io import BytesIO
+
+from openpyxl import Workbook, load_workbook
+
+from app.services.agent_asset_spreadsheet import AgentAssetSpreadsheetManager
+
+
+def test_rebuild_from_uploaded_content_preserves_sheet_values() -> None:
+    source = Workbook()
+    first = source.active
+    first.title = "差旅标准"
+    first.append(["城市", "住宿费"])
+    first.append(["北京", 500])
+    second = source.create_sheet("补贴标准")
+    second.append(["区域", "餐补"])
+    second.append(["直辖市", 75])
+
+    source_buffer = BytesIO()
+    source.save(source_buffer)
+
+    rebuilt = AgentAssetSpreadsheetManager.rebuild_from_uploaded_content(
+        source_buffer.getvalue()
+    )
+    workbook = load_workbook(BytesIO(rebuilt), data_only=False)
+
+    assert workbook.sheetnames == ["差旅标准", "补贴标准"]
+    assert workbook["差旅标准"]["A2"].value == "北京"
+    assert workbook["差旅标准"]["B2"].value == "500"
+    assert workbook["补贴标准"]["A2"].value == "直辖市"
+    assert workbook["补贴标准"]["B2"].value == "75"
+
+
+def build_workbook_bytes(rows: list[list[object]], *, sheet_name: str = "规则表") -> bytes:
+    workbook = Workbook()
+    sheet = workbook.active
+    sheet.title = sheet_name
+    for row in rows:
+        sheet.append(row)
+    buffer = BytesIO()
+    workbook.save(buffer)
+    return buffer.getvalue()