357 lines
11 KiB
Bash
Executable File
357 lines
11 KiB
Bash
Executable File
#!/bin/bash
|
||
# test-tmux-msg.sh — tmux-msg.sh / tmux-msg-notify.sh 单元测试
|
||
#
|
||
# 用法: bash scripts/test-tmux-msg.sh
|
||
#
|
||
# 通过 mock tmux 命令隔离测试，覆盖消息协议的 12 个场景
|
||
|
||
set -euo pipefail
|
||
|
||
MSG_SCRIPT="/workspace/scripts/tmux-msg.sh"
|
||
NOTIFY_SCRIPT="/workspace/scripts/tmux-msg-notify.sh"
|
||
LOG_SCRIPT="/workspace/scripts/tmux-msg-log.sh"
|
||
SEND_SCRIPT="/workspace/scripts/tmux-send-prompt.sh"
|
||
TEST_DIR=$(mktemp -d)
|
||
MOCK_DIR="${TEST_DIR}/mock-bin"
|
||
MSG_DIR="${TEST_DIR}/.tmux-messages"
|
||
LOG_FILE="${TEST_DIR}/test-output.log"
|
||
|
||
# 替代 /workspace/.tmux-messages 为测试目录
|
||
export TMUX_MSG_DIR_OVERRIDE="$MSG_DIR"
|
||
|
||
# ─── 颜色输出 ───
|
||
GREEN='\033[32m'
|
||
RED='\033[31m'
|
||
YELLOW='\033[33m'
|
||
RESET='\033[0m'
|
||
|
||
pass=0
|
||
fail=0
|
||
skip=0
|
||
|
||
result() {
|
||
local label="$1" ok="$2" detail="${3:-}"
|
||
if [[ "$ok" == "PASS" ]]; then
|
||
((pass++)) || true
|
||
echo -e " ${GREEN}✓ PASS${RESET} ${label}${detail:+ — ${detail}}"
|
||
elif [[ "$ok" == "SKIP" ]]; then
|
||
((skip++)) || true
|
||
echo -e " ${YELLOW}⊘ SKIP${RESET} ${label}${detail:+ — ${detail}}"
|
||
else
|
||
((fail++)) || true
|
||
echo -e " ${RED}✗ FAIL${RESET} ${label}${detail:+ — ${detail}}"
|
||
fi
|
||
}
|
||
|
||
cleanup() {
|
||
rm -rf "$TEST_DIR"
|
||
}
|
||
trap cleanup EXIT
|
||
|
||
# ─── 创建 mock tmux ───
|
||
create_mock_tmux() {
|
||
mkdir -p "$MOCK_DIR"
|
||
cat > "${MOCK_DIR}/tmux" <<'MOCK'
|
||
#!/bin/bash
|
||
LOG="$(dirname "$0")/../tmux-calls.log"
|
||
echo "$@" >> "$LOG"
|
||
if [[ "$1" == "capture-pane" ]]; then
|
||
echo "Some output"
|
||
echo "❯ "
|
||
fi
|
||
if [[ "$1" == "display-message" ]]; then
|
||
echo "✳ MockPane"
|
||
fi
|
||
if [[ "$1" == "send-keys" ]]; then
|
||
# 模拟发送成功
|
||
exit 0
|
||
fi
|
||
MOCK
|
||
chmod +x "${MOCK_DIR}/tmux"
|
||
}
|
||
|
||
clear_log() {
|
||
: > "${TEST_DIR}/tmux-calls.log"
|
||
}
|
||
|
||
read_log() {
|
||
cat "${TEST_DIR}/tmux-calls.log" 2>/dev/null || true
|
||
}
|
||
|
||
# 创建测试用的消息目录
|
||
setup_msg_dir() {
|
||
mkdir -p "${MSG_DIR}/prompts"
|
||
mkdir -p "${MSG_DIR}/$(date +%Y/%m/%d)"
|
||
: > "${MSG_DIR}/index.jsonl"
|
||
}
|
||
|
||
# ═══════════════════════════════════════════
|
||
echo ""
|
||
echo "═══ tmux-msg 协议单元测试 ═══"
|
||
echo ""
|
||
|
||
# ─── 1. send 文本字符串 ───
|
||
echo "── 1. send 文本字符串 ──"
|
||
create_mock_tmux
|
||
setup_msg_dir
|
||
clear_log
|
||
|
||
output=$(PATH="${MOCK_DIR}:${PATH}" TMUX_MSG_DIR="$MSG_DIR" TMUX_SEND_LOG="${TEST_DIR}/send.log" TMUX_MSG_NO_VERIFY=1 bash "$MSG_SCRIPT" send %2 "测试任务描述" --from %1 2>&1 || true)
|
||
msg_id=$(echo "$output" | grep -oE '^msg-[0-9]+-[0-9]+-[0-9a-f]+$' | head -1)
|
||
|
||
if [[ -n "$msg_id" ]]; then
|
||
result "send → 返回有效 msg_id" "PASS" "$msg_id"
|
||
else
|
||
result "send → 返回有效 msg_id" "FAIL" "实际: $msg_id"
|
||
fi
|
||
|
||
# 验证信封文件创建
|
||
envelope=$(find "$MSG_DIR" -name "${msg_id}.json" -type f 2>/dev/null | head -1)
|
||
if [[ -n "$envelope" ]]; then
|
||
result "send → 信封文件已创建" "PASS"
|
||
else
|
||
result "send → 信封文件已创建" "FAIL"
|
||
fi
|
||
|
||
# 验证索引更新
|
||
if [[ -f "${MSG_DIR}/index.jsonl" ]] && grep -q "$msg_id" "${MSG_DIR}/index.jsonl"; then
|
||
result "send → index.jsonl 已更新" "PASS"
|
||
else
|
||
result "send → index.jsonl 已更新" "FAIL"
|
||
fi
|
||
|
||
# ─── 2. send 包含回复指令 ───
|
||
echo ""
|
||
echo "── 2. send 包含回复指令 ──"
|
||
|
||
# 检查提示词文件包含 notify 指令
|
||
prompt_file="${MSG_DIR}/prompts/${msg_id}.md"
|
||
if [[ -f "$prompt_file" ]] && grep -q "tmux-msg-notify" "$prompt_file"; then
|
||
result "send → 提示词包含通知指令" "PASS"
|
||
else
|
||
result "send → 提示词包含通知指令" "FAIL"
|
||
fi
|
||
|
||
if [[ -f "$prompt_file" ]] && grep -q "跨面板任务" "$prompt_file"; then
|
||
result "send → 提示词包含任务头" "PASS"
|
||
else
|
||
result "send → 提示词包含任务头" "FAIL"
|
||
fi
|
||
|
||
if [[ -f "$prompt_file" ]] && grep -q "%1" "$prompt_file"; then
|
||
result "send → 提示词包含来源 pane" "PASS"
|
||
else
|
||
result "send → 提示词包含来源 pane" "FAIL"
|
||
fi
|
||
|
||
# ─── 3. 消息 ID 格式 ───
|
||
echo ""
|
||
echo "── 3. 消息 ID 格式 ──"
|
||
|
||
output2=$(PATH="${MOCK_DIR}:${PATH}" TMUX_MSG_DIR="$MSG_DIR" TMUX_SEND_LOG="${TEST_DIR}/send.log" TMUX_MSG_NO_VERIFY=1 bash "$MSG_SCRIPT" send %3 "第二个任务" 2>&1 || true)
|
||
msg_id2=$(echo "$output2" | grep -oE '^msg-[0-9]+-[0-9]+-[0-9a-f]+$' | head -1)
|
||
|
||
if [[ -n "$msg_id2" ]]; then
|
||
result "消息 ID 格式正确" "PASS" "$msg_id2"
|
||
else
|
||
result "消息 ID 格式正确" "FAIL" "实际: $msg_id2"
|
||
fi
|
||
|
||
if [[ "$msg_id" != "$msg_id2" ]]; then
|
||
result "消息 ID 唯一" "PASS"
|
||
else
|
||
result "消息 ID 唯一" "FAIL" "两次 send 返回相同 ID"
|
||
fi
|
||
|
||
# ─── 4. pane 标题解析 ───
|
||
echo ""
|
||
echo "── 4. pane 标题解析 ──"
|
||
|
||
if [[ -f "$envelope" ]]; then
|
||
from_title=$(python3 -c "import json; print(json.load(open('$envelope'))['from_title'])" 2>/dev/null)
|
||
if [[ "$from_title" == "✳ MockPane" ]]; then
|
||
result "pane 标题从 mock 正确解析" "PASS" "$from_title"
|
||
else
|
||
result "pane 标题从 mock 正确解析" "FAIL" "实际: $from_title"
|
||
fi
|
||
fi
|
||
|
||
# ─── 5. notify 成功 ───
|
||
echo ""
|
||
echo "── 5. notify 成功 ──"
|
||
setup_msg_dir
|
||
|
||
# 先手动创建一个消息信封用于 notify 测试
|
||
test_msg_id="msg-20260419-120000-test"
|
||
day_dir=$(date +%Y/%m/%d)
|
||
mkdir -p "${MSG_DIR}/${day_dir}"
|
||
cat > "${MSG_DIR}/${day_dir}/${test_msg_id}.json" <<ENVEOF
|
||
{"msg_id":"${test_msg_id}","direction":"request","from_pane":"%1","from_title":"PM","to_pane":"%2","to_title":"后端","task_summary":"测试任务","task_file":null,"status":"sent","created_at":"2026-04-19T12:00:00+08:00","delivered_at":"2026-04-19T12:00:01+08:00","replied_at":null,"reply_msg_id":null,"reply_summary":null,"error":null}
|
||
ENVEOF
|
||
echo "{\"msg_id\":\"${test_msg_id}\",\"direction\":\"request\",\"from\":\"%1\",\"to\":\"%2\",\"status\":\"sent\",\"summary\":\"测试任务\",\"ts\":\"2026-04-19T12:00:00+08:00\"}" >> "${MSG_DIR}/index.jsonl"
|
||
|
||
clear_log
|
||
output=$(PATH="${MOCK_DIR}:${PATH}" TMUX_MSG_DIR="$MSG_DIR" TMUX_SEND_LOG="${TEST_DIR}/send.log" TMUX_PANE="%2" bash "$NOTIFY_SCRIPT" "$test_msg_id" "%1" "已完成测试" 2>&1 || true)
|
||
|
||
# 验证原消息状态更新
|
||
updated_status=$(python3 -c "import json; print(json.load(open('${MSG_DIR}/${day_dir}/${test_msg_id}.json'))['status'])" 2>/dev/null)
|
||
if [[ "$updated_status" == "completed" ]]; then
|
||
result "notify → 原消息状态更新为 completed" "PASS"
|
||
else
|
||
result "notify → 原消息状态更新为 completed" "FAIL" "实际: $updated_status"
|
||
fi
|
||
|
||
# 验证回复信封创建
|
||
reply_envelope="${MSG_DIR}/${day_dir}/reply-${test_msg_id}.json"
|
||
if [[ -f "$reply_envelope" ]]; then
|
||
result "notify → 回复信封已创建" "PASS"
|
||
else
|
||
result "notify → 回复信封已创建" "FAIL"
|
||
fi
|
||
|
||
# 验证 notify 调用了 send-prompt 发送回复
|
||
log=$(read_log)
|
||
if echo "$log" | grep -q "send-keys"; then
|
||
result "notify → 调用了 tmux send-keys" "PASS"
|
||
else
|
||
result "notify → 调用了 tmux send-keys" "FAIL"
|
||
fi
|
||
|
||
# ─── 6. notify 未知 msg_id 容错 ───
|
||
echo ""
|
||
echo "── 6. notify 未知 msg_id ──"
|
||
clear_log
|
||
|
||
output=$(PATH="${MOCK_DIR}:${PATH}" TMUX_MSG_DIR="$MSG_DIR" TMUX_SEND_LOG="${TEST_DIR}/send.log" TMUX_PANE="%2" bash "$NOTIFY_SCRIPT" "msg-nonexistent" "%1" "测试" 2>&1 || true)
|
||
if echo "$output" | grep -q "未找到消息"; then
|
||
result "notify 未知 msg_id → 宽容处理" "PASS"
|
||
else
|
||
result "notify 未知 msg_id → 宽容处理" "FAIL" "输出: $output"
|
||
fi
|
||
|
||
# ─── 7. 目录结构 ───
|
||
echo ""
|
||
echo "── 7. 目录结构 ──"
|
||
|
||
today_dir=$(date +%Y/%m/%d)
|
||
if [[ -d "${MSG_DIR}/${today_dir}" ]]; then
|
||
result "日期分区目录已创建" "PASS" "${MSG_DIR}/${today_dir}"
|
||
else
|
||
result "日期分区目录已创建" "FAIL"
|
||
fi
|
||
|
||
if [[ -d "${MSG_DIR}/prompts" ]]; then
|
||
result "prompts 目录已创建" "PASS"
|
||
else
|
||
result "prompts 目录已创建" "FAIL"
|
||
fi
|
||
|
||
# ─── 8. index.jsonl 一致性 ───
|
||
echo ""
|
||
echo "── 8. index.jsonl 一致性 ──"
|
||
|
||
line_count=$(wc -l < "${MSG_DIR}/index.jsonl" | tr -d ' ')
|
||
if [[ "$line_count" -ge 1 ]]; then
|
||
result "index.jsonl 有内容" "PASS" "$line_count 行"
|
||
else
|
||
result "index.jsonl 有内容" "FAIL" "空文件"
|
||
fi
|
||
|
||
# 验证每行都是有效 JSON
|
||
invalid_lines=0
|
||
while IFS= read -r line; do
|
||
[[ -z "$line" ]] && continue
|
||
if ! echo "$line" | python3 -c "import sys,json; json.loads(sys.stdin.read())" 2>/dev/null; then
|
||
invalid_lines=$((invalid_lines + 1))
|
||
fi
|
||
done < "${MSG_DIR}/index.jsonl"
|
||
|
||
if [[ "$invalid_lines" -eq 0 ]]; then
|
||
result "index.jsonl 所有行都是有效 JSON" "PASS"
|
||
else
|
||
result "index.jsonl 所有行都是有效 JSON" "FAIL" "${invalid_lines} 行无效"
|
||
fi
|
||
|
||
# ─── 9. log 工具查询 ───
|
||
echo ""
|
||
echo "── 9. log 工具查询 ──"
|
||
|
||
output=$(TMUX_MSG_DIR="$MSG_DIR" bash "$LOG_SCRIPT" recent 2>&1)
|
||
if echo "$output" | grep -q "消息"; then
|
||
result "log recent → 返回消息列表" "PASS"
|
||
else
|
||
result "log recent → 返回消息列表" "FAIL"
|
||
fi
|
||
|
||
output=$(TMUX_MSG_DIR="$MSG_DIR" bash "$LOG_SCRIPT" show "$test_msg_id" 2>&1)
|
||
if echo "$output" | grep -q "$test_msg_id"; then
|
||
result "log show → 返回消息详情" "PASS"
|
||
else
|
||
result "log show → 返回消息详情" "FAIL"
|
||
fi
|
||
|
||
output=$(TMUX_MSG_DIR="$MSG_DIR" bash "$LOG_SCRIPT" stats 2>&1)
|
||
if echo "$output" | grep -q "总计"; then
|
||
result "log stats → 返回统计" "PASS"
|
||
else
|
||
result "log stats → 返回统计" "FAIL"
|
||
fi
|
||
|
||
# ─── 10. list 过滤 ───
|
||
echo ""
|
||
echo "── 10. list 过滤 ──"
|
||
|
||
output=$(TMUX_MSG_DIR="$MSG_DIR" bash "$LOG_SCRIPT" list --from %1 2>&1)
|
||
if echo "$output" | grep -q "%1"; then
|
||
result "list --from → 过滤生效" "PASS"
|
||
else
|
||
result "list --from → 过滤生效" "FAIL"
|
||
fi
|
||
|
||
output=$(TMUX_MSG_DIR="$MSG_DIR" bash "$LOG_SCRIPT" by-pane %2 2>&1)
|
||
if echo "$output" | grep -q "%2"; then
|
||
result "by-pane → 过滤生效" "PASS"
|
||
else
|
||
result "by-pane → 过滤生效" "FAIL"
|
||
fi
|
||
|
||
# ─── 11. send 参数校验 ───
|
||
echo ""
|
||
echo "── 11. send 参数校验 ──"
|
||
|
||
output=$(PATH="${MOCK_DIR}:${PATH}" TMUX_MSG_DIR="$MSG_DIR" TMUX_SEND_LOG="${TEST_DIR}/send.log" bash "$MSG_SCRIPT" send 2>&1 || true)
|
||
if echo "$output" | grep -q "用法\|需要"; then
|
||
result "send 无参数 → 提示用法" "PASS"
|
||
else
|
||
result "send 无参数 → 提示用法" "FAIL"
|
||
fi
|
||
|
||
# ─── 12. 信封字段完整性 ───
|
||
echo ""
|
||
echo "── 12. 信封字段完整性 ──"
|
||
|
||
required_fields="msg_id direction from_pane from_title to_pane to_title task_summary status created_at"
|
||
missing_fields=""
|
||
for field in $required_fields; do
|
||
if ! python3 -c "import json; d=json.load(open('$envelope')); assert '$field' in d" 2>/dev/null; then
|
||
missing_fields="${missing_fields} ${field}"
|
||
fi
|
||
done
|
||
|
||
if [[ -z "$missing_fields" ]]; then
|
||
result "信封字段完整" "PASS"
|
||
else
|
||
result "信封字段完整" "FAIL" "缺少:${missing_fields}"
|
||
fi
|
||
|
||
# ═══════════════════════════════════════════
|
||
echo ""
|
||
echo "════════════════════════════════════"
|
||
echo -e " ${GREEN}通过: ${pass}${RESET} ${RED}失败: ${fail}${RESET} ${YELLOW}跳过: ${skip}${RESET} 总计: $((pass + fail + skip))"
|
||
echo "════════════════════════════════════"
|
||
echo ""
|
||
|
||
if [[ "$fail" -gt 0 ]]; then
|
||
exit 1
|
||
fi
|