3db196e49d
基于 skill-template 脚手架开发，包含： - payroll_config（费率配置，可编辑）+ payslip_snapshots（解析历史，只读）2 张业务表 - 全国默认 + 北上广 4 套费率（社保/公积金/个税起征点/7 级税率表） - 5 个业务 Action：calculate-payslip-simple / calculate-payslip-full / explain-term / view-config / edit-config - 个税累计预扣法计算，支持多月历史累加 - 20 个薪资名词库（社保/公积金/个税/全勤/加班费等） - 完整市场文档（README/TUTORIAL/DEMO/CHANGELOG） - 296 测试全部通过
138 lines
4.7 KiB
Plaintext
138 lines
4.7 KiB
Plaintext
# -*- coding: utf-8 -*-
|
||
"""
|
||
可选桌面 E2E 示例（默认不执行：文件名后缀为 .sample）—— **附件上传 + 长流程场景**。
|
||
|
||
启用方式：
|
||
1. 复制为 test_desktop_smoke_with_attachment.py
|
||
2. 修改 SKILL_SLUG / PROMPT / 断言为你技能的真实业务
|
||
3. 准备样例附件，或动态生成（参考 _prepare_sample_attachment）
|
||
4. 运行：pytest tests/desktop/test_desktop_smoke_with_attachment.py -v -s
|
||
|
||
依赖：
|
||
- pytest / jiangchang_desktop_sdk / 匠厂桌面宿主已就绪
|
||
- 通过 conftest.py 自动 bootstrap，jiangchang_desktop_sdk.e2e_helpers 模块在 sys.path
|
||
|
||
【为什么用此模板而不是 test_desktop_smoke.py.sample】
|
||
- client.send_file() 在 v2.0.17+ 抛 NotImplementedError，必须用拖拽方式
|
||
- client.ask() 在 user_idx<0 时会锁死（已知 SDK 缺陷），本模板绕开 ask()，
|
||
在测试侧自己用 jiangchang_desktop_sdk.e2e_helpers 做等待
|
||
|
||
【字幕配合】
|
||
- 关键节点都打了 _logger.info("step: N/7 ...")，对应 record_screencast.py 的
|
||
SUBTITLE_SCRIPT 用 step: 关键词触发字幕
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
import os
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
|
||
from jiangchang_desktop_sdk import JiangchangDesktopClient
|
||
from jiangchang_desktop_sdk.e2e_helpers import (
|
||
drop_file_into_composer,
|
||
send_prompt_via_composer,
|
||
wait_for_agent_complete,
|
||
wait_for_attachment_ready,
|
||
wait_for_composer_enabled,
|
||
wait_for_user_message,
|
||
)
|
||
|
||
# ============================================================
|
||
# ↓↓↓ 复制后修改这些 ↓↓↓
|
||
# ============================================================
|
||
|
||
SKILL_SLUG = "query-payslip-default"
|
||
|
||
PROMPT = (
|
||
"请使用 /query-payslip-default 处理我刚才上传的文件。"
|
||
"完成后请用清单方式汇总：[改成你技能要求的字段]。"
|
||
)
|
||
|
||
# 业务断言：根据你的技能输出修改
|
||
EXPECTED_KEYWORDS_IN_REPLY = ["关键词1", "关键词2"]
|
||
|
||
# ============================================================
|
||
|
||
_logger = logging.getLogger("desktop-e2e-with-attachment-sample")
|
||
if not _logger.handlers:
|
||
_logger.setLevel(logging.INFO)
|
||
_h = logging.StreamHandler()
|
||
_h.setFormatter(logging.Formatter("[E2E] %(message)s"))
|
||
_logger.addHandler(_h)
|
||
|
||
_ARTIFACT_DIR = os.path.join(os.path.dirname(__file__), "artifacts")
|
||
|
||
|
||
def _prepare_sample_attachment(tmp_path: Path) -> Path:
|
||
"""在 tmp_path 下生成一个样例附件（按 skill 真实需要修改）。
|
||
|
||
示例：生成一个 5 行的 xlsx。你的 skill 可能需要 CSV / PDF / 图片等其他格式。
|
||
"""
|
||
from openpyxl import Workbook # noqa: WPS433
|
||
xlsx = tmp_path / "sample.xlsx"
|
||
wb = Workbook()
|
||
ws = wb.active
|
||
ws.append(["列1", "列2", "列3"])
|
||
for i in range(1, 6):
|
||
ws.append([f"值{i}A", f"值{i}B", i * 100])
|
||
wb.save(str(xlsx))
|
||
return xlsx
|
||
|
||
|
||
@pytest.fixture
|
||
def client(request):
|
||
c = JiangchangDesktopClient()
|
||
_logger.info("ensure_app_running")
|
||
c.ensure_app_running(wait_timeout_s=45)
|
||
c.bring_to_front()
|
||
c.wait_gateway_ready(timeout_ms=30000)
|
||
yield c
|
||
try:
|
||
rep = getattr(request.node, "rep_call", None)
|
||
if rep is not None and rep.failed:
|
||
os.makedirs(_ARTIFACT_DIR, exist_ok=True)
|
||
paths = c.snapshot(_ARTIFACT_DIR, tag=request.node.name)
|
||
_logger.error("failure snapshot: %s", paths)
|
||
finally:
|
||
c.disconnect()
|
||
|
||
|
||
def test_desktop_with_attachment(client: JiangchangDesktopClient, tmp_path: Path) -> None:
|
||
_logger.info("step: 1/7 准备样例附件")
|
||
attachment_path = _prepare_sample_attachment(tmp_path)
|
||
|
||
client.new_task()
|
||
page = client.get_page()
|
||
client.bring_to_front()
|
||
|
||
_logger.info("step: 2/7 拖拽附件到聊天框")
|
||
drop_file_into_composer(page, attachment_path)
|
||
|
||
_logger.info("step: 3/7 等附件 stage-buffer 完成")
|
||
wait_for_attachment_ready(page, attachment_path.name)
|
||
|
||
wait_for_composer_enabled(page)
|
||
client.bring_to_front()
|
||
_logger.info("step: 4/7 键入并发送指令")
|
||
send_prompt_via_composer(page, PROMPT)
|
||
|
||
_logger.info("step: 5/7 Main Agent 接收，等待路由到技能")
|
||
user_idx = wait_for_user_message(page, PROMPT, timeout_s=60.0)
|
||
|
||
_logger.info("step: 6/7 技能在后端执行（可能 2-3 分钟）")
|
||
client.bring_to_front()
|
||
reply = wait_for_agent_complete(
|
||
page, user_idx,
|
||
overall_timeout_s=1200.0,
|
||
stable_seconds=5.0,
|
||
)
|
||
|
||
_logger.info("step: 7/7 校验回复内容")
|
||
_logger.info("reply length=%d", len(reply))
|
||
|
||
assert reply, "assistant 回复为空"
|
||
for kw in EXPECTED_KEYWORDS_IN_REPLY:
|
||
assert kw in reply, f"期望关键词 {kw!r} 未出现在回复中；片段：{reply[:400]!r}"
|