hiagent-java-sdk/run-test.sh at main · volcengine/hiagent-java-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#!/bin/bash

# 读取 .env 文件（如果存在）
if [ -f .env ]; then
    echo "检测到 .env 文件，正在加载..."
    set -a
    source .env
    set +a
fi

# 检查参数
if [ -z "$1" ]; then
    echo "❌ 缺少测试用例类名参数"
    echo "用法: $0 <TestClassName>"
    echo "例如: $0 com.volcengine.hiagent.api.examples.TestCreateApiToken"
    exit 1
fi

TEST_CLASS=$1

# 检查环境变量
echo "检查环境变量..."
MISSING_VARS=""

if [ -z "$VOLC_ACCESSKEY" ]; then
    MISSING_VARS="$MISSING_VARS VOLC_ACCESSKEY"
fi

if [ -z "$VOLC_SECRETKEY" ]; then
    MISSING_VARS="$MISSING_VARS VOLC_SECRETKEY"
fi

if [ -n "$MISSING_VARS" ]; then
    echo "❌ 缺少以下环境变量: $MISSING_VARS"
    echo
    echo "请设置环境变量："
    echo "export VOLC_ACCESSKEY=\"your_access_key\""
    echo "export VOLC_SECRETKEY=\"your_secret_key\""
    echo "export HIAGENT_TOP_REGION=\"cn-beijing\"  # 可选，默认值"
    echo "export HIAGENT_TOP_ENDPOINT=\"observe.volcengineapi.com\"  # 可选，默认值"
    echo "export WORKSPACE_ID=\"your_workspace_id\""
    echo "export CUSTOM_APP_ID=\"your_custom_app_id\""
    echo
    echo "或者创建 .env 文件（如果支持的话）"
    exit 1
fi

echo "✓ 环境变量检查通过"
echo

# 设置默认值
if [ -z "$HIAGENT_TOP_REGION" ]; then
    export HIAGENT_TOP_REGION="cn-beijing"
    echo "使用默认 HIAGENT_TOP_REGION: $HIAGENT_TOP_REGION"
fi

if [ -z "$HIAGENT_TOP_ENDPOINT" ]; then
    export HIAGENT_TOP_ENDPOINT="observe.volcengineapi.com"
    echo "使用默认 HIAGENT_TOP_ENDPOINT: $HIAGENT_TOP_ENDPOINT"
fi

echo
echo "当前配置："
echo "  HIAGENT_TOP_REGION: $HIAGENT_TOP_REGION"
echo "  HIAGENT_TOP_ENDPOINT: $HIAGENT_TOP_ENDPOINT"
echo "  WORKSPACE_ID: $WORKSPACE_ID"
echo "  CUSTOM_APP_ID: $CUSTOM_APP_ID"
echo

# 根据类名确定模块
MODULE_DIR=""
if [[ $TEST_CLASS == com.volcengine.hiagent.api.* ]]; then
    MODULE_DIR="hiagent-api"
elif [[ $TEST_CLASS == com.volcengine.hiagent.observe.* ]]; then
    MODULE_DIR="hiagent-observe"
elif [[ $TEST_CLASS == com.volcengine.hiagent.components.* ]]; then
    MODULE_DIR="hiagent-components"
elif [[ $TEST_CLASS == com.volcengine.hiagent.eva.* ]]; then
    MODULE_DIR="hiagent-eva"
else
    echo "❌ 无法确定模块，类名: $TEST_CLASS"
    echo "支持的包前缀："
    echo "  - com.volcengine.hiagent.api.*"
    echo "  - com.volcengine.hiagent.observe.*"
    echo "  - com.volcengine.hiagent.components.*"
    echo "  - com.volcengine.hiagent.eva.*"
    exit 1
fi

# 进入正确的目录
echo "正在安装整个项目到本地仓库..."
mvn clean install -q -DskipTests

if [ $? -ne 0 ]; then
    echo "❌ 安装失败"
    exit 1
fi

echo "✓ 安装成功"
echo

echo "进入 $MODULE_DIR 目录..."
cd "$MODULE_DIR" || {
    echo "❌ 无法进入 $MODULE_DIR 目录"
    echo "请确保在项目根目录运行此脚本"
    exit 1
}

echo "正在运行 $TEST_CLASS..."
echo "=================================="

# 运行测试
mvn exec:java -Dexec.mainClass="$TEST_CLASS" -q

echo "=================================="
echo "测试完成"