rwadurian/backend/services/blockchain-service/scripts/quick-test.sh

120 lines
3.4 KiB
Bash

#!/bin/bash
# 快速测试脚本 - 在本地环境快速验证核心功能
set -e
echo "🚀 开始快速测试 Blockchain Service..."
echo ""
# 颜色定义
GREEN='\033[0;32m'
RED='\033[0;31m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color
BASE_URL="http://localhost:3012"
# 测试结果统计
PASS=0
FAIL=0
# 测试函数
test_api() {
local test_name=$1
local method=$2
local endpoint=$3
local data=$4
local expected_status=$5
echo -n "Testing: $test_name ... "
if [ -n "$data" ]; then
response=$(curl -s -w "\n%{http_code}" -X $method "$BASE_URL$endpoint" \
-H "Content-Type: application/json" \
-d "$data")
else
response=$(curl -s -w "\n%{http_code}" -X $method "$BASE_URL$endpoint" \
-H "Content-Type: application/json")
fi
status=$(echo "$response" | tail -n1)
body=$(echo "$response" | head -n-1)
if [ "$status" -eq "$expected_status" ]; then
echo -e "${GREEN}✓ PASS${NC}"
PASS=$((PASS + 1))
if command -v jq &> /dev/null && [ -n "$body" ]; then
echo "$body" | jq '.' 2>/dev/null || echo "$body"
else
echo "$body"
fi
else
echo -e "${RED}✗ FAIL${NC} (Expected: $expected_status, Got: $status)"
FAIL=$((FAIL + 1))
echo "$body"
fi
echo ""
}
# 1. 健康检查
echo -e "${YELLOW}=== 1. 健康检查 ===${NC}"
test_api "Health Check" "GET" "/health" "" 200
test_api "Ready Check" "GET" "/health/ready" "" 200
# 2. 余额查询测试
echo -e "${YELLOW}=== 2. 余额查询 ===${NC}"
# 使用一个已知的测试地址 (Binance Hot Wallet)
TEST_ADDRESS="0x8894E0a0c962CB723c1976a4421c95949bE2D4E3"
test_api "Query KAVA Balance" "GET" "/balance?chainType=KAVA&address=$TEST_ADDRESS" "" 200
test_api "Query Multi-Chain Balance" "GET" "/balance/multi-chain?address=$TEST_ADDRESS" "" 200
# 3. 地址派生测试
echo -e "${YELLOW}=== 3. 地址派生测试 ===${NC}"
# 测试用压缩公钥 (仅用于测试)
TEST_PUBLIC_KEY="0x02b4632d08485ff1df2db55b9dafd23347d1c47a457072a1e87be26896549a8737"
TEST_USER_ID="999999"
test_api "Derive Address" "POST" "/internal/derive-address" \
"{\"userId\": \"$TEST_USER_ID\", \"publicKey\": \"$TEST_PUBLIC_KEY\"}" \
201
# 4. 获取用户地址
echo -e "${YELLOW}=== 4. 获取用户地址 ===${NC}"
test_api "Get User Addresses" "GET" "/internal/user/$TEST_USER_ID/addresses" "" 200
# 5. 错误场景测试
echo -e "${YELLOW}=== 5. 错误场景测试 ===${NC}"
# 无效地址格式
test_api "Invalid Address Format" "GET" "/balance?chainType=KAVA&address=invalid" "" 400
# 无效链类型
test_api "Invalid Chain Type" "GET" "/balance?chainType=INVALID&address=$TEST_ADDRESS" "" 400
# 无效公钥格式
test_api "Invalid Public Key" "POST" "/internal/derive-address" \
"{\"userId\": \"1\", \"publicKey\": \"invalid\"}" \
400
# 6. API 文档测试
echo -e "${YELLOW}=== 6. API 文档 ===${NC}"
test_api "Swagger API Docs" "GET" "/api" "" 200
# 总结
echo ""
echo -e "${YELLOW}======================================${NC}"
echo -e "${YELLOW}测试完成!${NC}"
echo -e "${GREEN}通过: $PASS${NC}"
echo -e "${RED}失败: $FAIL${NC}"
echo -e "${YELLOW}======================================${NC}"
if [ $FAIL -eq 0 ]; then
echo -e "${GREEN}✓ 所有测试通过!${NC}"
exit 0
else
echo -e "${RED}✗ 存在失败的测试!${NC}"
exit 1
fi