Kris
3264900bc9
新增了Twitch、Telegram、Discord、Slack、Mattermost、WeChat、Zalo等多渠道的单元测试用例，覆盖了令牌处理、速率限制、消息去重、会话解析、格式转换、安全策略等模块 同时在测试配置中添加了测试用的OpenAI API密钥环境变量
131 lines
4.7 KiB
Python
131 lines
4.7 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from yuxi.channels.adapters.slack.blocks import (
|
|
build_actions,
|
|
build_approval_blocks,
|
|
build_button,
|
|
build_divider,
|
|
build_error_notice,
|
|
build_poll_blocks,
|
|
build_section,
|
|
)
|
|
|
|
|
|
class TestBuildSection:
|
|
def test_basic_section(self):
|
|
block = build_section("Hello World")
|
|
assert block["type"] == "section"
|
|
assert block["text"]["type"] == "mrkdwn"
|
|
assert block["text"]["text"] == "Hello World"
|
|
assert "accessory" not in block
|
|
|
|
def test_with_accessory(self):
|
|
accessory = {"type": "button", "text": {"type": "plain_text", "text": "Click"}}
|
|
block = build_section("Hello", accessory=accessory)
|
|
assert block["accessory"] == accessory
|
|
|
|
|
|
class TestBuildDivider:
|
|
def test_divider(self):
|
|
assert build_divider() == {"type": "divider"}
|
|
|
|
|
|
class TestBuildButton:
|
|
def test_default_button(self):
|
|
btn = build_button("Click me", "action_1")
|
|
assert btn["type"] == "button"
|
|
assert btn["text"]["text"] == "Click me"
|
|
assert btn["action_id"] == "action_1"
|
|
assert "value" not in btn
|
|
assert "style" not in btn
|
|
|
|
def test_with_value(self):
|
|
btn = build_button("Click", "action_1", value="val1")
|
|
assert btn["value"] == "val1"
|
|
|
|
def test_primary_style(self):
|
|
btn = build_button("Submit", "submit", style="primary")
|
|
assert btn["style"] == "primary"
|
|
|
|
def test_danger_style(self):
|
|
btn = build_button("Delete", "delete", style="danger")
|
|
assert btn["style"] == "danger"
|
|
|
|
def test_invalid_style_ignored(self):
|
|
btn = build_button("Click", "action_1", style="custom")
|
|
assert "style" not in btn
|
|
|
|
|
|
class TestBuildActions:
|
|
def test_single_button(self):
|
|
btn = build_button("OK", "ok_btn")
|
|
actions = build_actions(btn)
|
|
assert actions["type"] == "actions"
|
|
assert len(actions["elements"]) == 1
|
|
assert actions["elements"][0] == btn
|
|
|
|
def test_multiple_buttons(self):
|
|
btn1 = build_button("Yes", "yes")
|
|
btn2 = build_button("No", "no")
|
|
actions = build_actions(btn1, btn2)
|
|
assert len(actions["elements"]) == 2
|
|
|
|
|
|
class TestBuildApprovalBlocks:
|
|
def test_approval_defaults(self):
|
|
blocks = build_approval_blocks("审批请求", "详细信息")
|
|
assert len(blocks) == 4
|
|
assert blocks[0]["type"] == "header"
|
|
assert blocks[0]["text"]["text"] == "审批请求"
|
|
assert blocks[1]["type"] == "section"
|
|
assert blocks[1]["text"]["text"] == "详细信息"
|
|
assert blocks[2]["type"] == "divider"
|
|
assert blocks[3]["type"] == "actions"
|
|
elements = blocks[3]["elements"]
|
|
assert elements[0]["action_id"] == "exec_approve"
|
|
assert elements[0]["style"] == "primary"
|
|
assert elements[0]["text"]["text"] == "确认执行"
|
|
assert elements[1]["action_id"] == "exec_reject"
|
|
assert elements[1]["style"] == "danger"
|
|
assert elements[1]["text"]["text"] == "取消"
|
|
|
|
def test_approval_custom_action_ids(self):
|
|
blocks = build_approval_blocks("审批", "详情", confirm_action_id="custom_approve", reject_action_id="custom_reject")
|
|
elements = blocks[3]["elements"]
|
|
assert elements[0]["action_id"] == "custom_approve"
|
|
assert elements[1]["action_id"] == "custom_reject"
|
|
|
|
def test_approval_custom_button_text(self):
|
|
blocks = build_approval_blocks("审批", "详情", confirm_text="Yes", reject_text="No")
|
|
elements = blocks[3]["elements"]
|
|
assert elements[0]["text"]["text"] == "Yes"
|
|
assert elements[1]["text"]["text"] == "No"
|
|
|
|
|
|
class TestBuildPollBlocks:
|
|
def test_poll_blocks(self):
|
|
blocks = build_poll_blocks("你喜欢什么颜色？", ["红色", "蓝色", "绿色"], "poll_001")
|
|
assert blocks[0]["type"] == "header"
|
|
assert blocks[0]["text"]["text"] == "📊 投票"
|
|
assert blocks[1]["text"]["text"] == "*你喜欢什么颜色？*"
|
|
assert blocks[2]["type"] == "divider"
|
|
|
|
option_blocks = blocks[3:]
|
|
assert len(option_blocks) == 3
|
|
for i, block in enumerate(option_blocks):
|
|
assert block["type"] == "section"
|
|
assert block["accessory"]["action_id"] == f"poll_poll_001_{i}"
|
|
assert block["accessory"]["value"] == str(i)
|
|
|
|
|
|
class TestBuildErrorNotice:
|
|
def test_error_notice(self):
|
|
blocks = build_error_notice("发生了错误")
|
|
assert len(blocks) == 2
|
|
assert blocks[0]["type"] == "section"
|
|
assert "错误" in blocks[0]["text"]["text"]
|
|
assert "发生了错误" in blocks[0]["text"]["text"]
|
|
assert blocks[1]["type"] == "context"
|
|
assert len(blocks[1]["elements"]) == 1 |