mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-07-18 20:17:09 +08:00
feat(feishu): add rich progress cards
This commit is contained in:
103
tests/test_feishu_progress_card.py
Normal file
103
tests/test_feishu_progress_card.py
Normal file
@@ -0,0 +1,103 @@
|
||||
from channel.feishu.feishu_progress_card import FeishuProgressState
|
||||
|
||||
|
||||
def _panels(card):
|
||||
return [
|
||||
element
|
||||
for element in card["body"]["elements"]
|
||||
if element.get("tag") == "collapsible_panel"
|
||||
]
|
||||
|
||||
|
||||
def test_running_card_has_status_reasoning_lane_and_stream_target():
|
||||
state = FeishuProgressState(started_at=100.0)
|
||||
|
||||
card = state.build_card(streaming=True, now=102.0)
|
||||
|
||||
assert card["header"]["template"] == "blue"
|
||||
assert card["header"]["title"]["content"] == "Working"
|
||||
assert card["config"]["streaming_mode"] is True
|
||||
assert _panels(card)[0]["header"]["title"]["content"] == "Reasoning"
|
||||
stream = next(
|
||||
element
|
||||
for element in card["body"]["elements"]
|
||||
if element.get("element_id") == "stream_md"
|
||||
)
|
||||
assert stream["content"] == "..."
|
||||
|
||||
|
||||
def test_tool_turn_populates_reasoning_and_tools_lanes():
|
||||
state = FeishuProgressState(started_at=100.0)
|
||||
state.consume({"type": "turn_start", "data": {"turn": 1}})
|
||||
state.consume({"type": "reasoning_update", "data": {"delta": "Need the latest files."}})
|
||||
state.consume({"type": "message_update", "data": {"delta": "Checking now."}})
|
||||
state.consume(
|
||||
{
|
||||
"type": "message_end",
|
||||
"data": {
|
||||
"tool_calls": [
|
||||
{"name": "read_file", "arguments": {"path": "README.md"}}
|
||||
]
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
card = state.build_card(streaming=True, now=103.0)
|
||||
panels = _panels(card)
|
||||
assert [panel["header"]["title"]["content"] for panel in panels] == [
|
||||
"Reasoning (1)",
|
||||
"Tools (1)",
|
||||
]
|
||||
assert panels[0]["elements"][0]["text"]["content"] == "Need the latest files."
|
||||
assert "read_file" in panels[1]["elements"][0]["text"]["content"]
|
||||
assert "running" in panels[1]["elements"][0]["text"]["content"]
|
||||
|
||||
state.consume({"type": "turn_start", "data": {"turn": 2}})
|
||||
done_card = state.build_card(streaming=True, now=104.0)
|
||||
assert "done" in _panels(done_card)[1]["elements"][0]["text"]["content"]
|
||||
|
||||
|
||||
def test_agent_end_finalizes_status_body_and_footer():
|
||||
state = FeishuProgressState(started_at=100.0)
|
||||
state.consume({"type": "turn_start", "data": {"turn": 1}})
|
||||
state.consume(
|
||||
{
|
||||
"type": "agent_end",
|
||||
"data": {"final_response": "**Finished**", "cancelled": False},
|
||||
}
|
||||
)
|
||||
|
||||
card = state.build_card(streaming=False, now=105.2)
|
||||
|
||||
assert card["header"]["template"] == "green"
|
||||
assert card["header"]["title"]["content"] == "Done"
|
||||
assert card["config"]["streaming_mode"] is False
|
||||
assert next(
|
||||
element["content"]
|
||||
for element in card["body"]["elements"]
|
||||
if element.get("element_id") == "stream_md"
|
||||
) == "**Finished**"
|
||||
footer = card["body"]["elements"][-1]
|
||||
assert footer["text_size"] == "notation"
|
||||
assert footer["content"] == "5.2s · 1 turn"
|
||||
|
||||
|
||||
def test_cancelled_agent_uses_partial_output_and_stopped_status():
|
||||
state = FeishuProgressState(started_at=100.0)
|
||||
state.consume({"type": "message_update", "data": {"delta": "partial"}})
|
||||
state.consume({"type": "agent_cancelled", "data": {}})
|
||||
state.consume(
|
||||
{
|
||||
"type": "agent_end",
|
||||
"data": {"final_response": "stale response", "cancelled": True},
|
||||
}
|
||||
)
|
||||
|
||||
card = state.build_card(streaming=False, now=101.0)
|
||||
|
||||
assert card["header"]["title"]["content"] == "Stopped"
|
||||
assert next(
|
||||
element["content"]
|
||||
for element in card["body"]["elements"]
|
||||
if element.get("element_id") == "stream_md"
|
||||
) == "partial"
|
||||
Reference in New Issue
Block a user