mirror of
https://github.com/laoxong/nofx.git
synced 2026-06-07 03:07:56 +08:00
Merge pull request #353 from Icyoung/beta
Beta Fix Competition、Rank api cache、Dev merge
This commit is contained in:
@@ -0,0 +1,54 @@
|
||||
name: Test
|
||||
|
||||
on:
|
||||
push:
|
||||
branches: [main, dev]
|
||||
pull_request:
|
||||
branches: [main, dev]
|
||||
|
||||
jobs:
|
||||
backend-tests:
|
||||
name: Backend Tests
|
||||
runs-on: ubuntu-latest
|
||||
continue-on-error: true # Don't block PRs
|
||||
|
||||
steps:
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Go
|
||||
uses: actions/setup-go@v5
|
||||
with:
|
||||
go-version: '1.23'
|
||||
|
||||
- name: Download dependencies
|
||||
run: go mod download
|
||||
|
||||
- name: Run tests
|
||||
run: go test -v ./...
|
||||
|
||||
- name: Generate coverage
|
||||
run: go test -coverprofile=coverage.out ./...
|
||||
continue-on-error: true
|
||||
|
||||
frontend-tests:
|
||||
name: Frontend Tests
|
||||
runs-on: ubuntu-latest
|
||||
continue-on-error: true # Don't block PRs
|
||||
|
||||
steps:
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Node.js
|
||||
uses: actions/setup-node@v4
|
||||
with:
|
||||
node-version: '20'
|
||||
cache: 'npm'
|
||||
cache-dependency-path: web/package-lock.json
|
||||
|
||||
- name: Install dependencies
|
||||
run: cd web && npm ci
|
||||
|
||||
- name: Run tests
|
||||
run: cd web && npm run test
|
||||
@@ -0,0 +1,153 @@
|
||||
# NOFX Makefile for testing and development
|
||||
|
||||
.PHONY: help test test-backend test-frontend test-coverage clean
|
||||
|
||||
# Default target
|
||||
help:
|
||||
@echo "NOFX Testing & Development Commands"
|
||||
@echo ""
|
||||
@echo "Testing:"
|
||||
@echo " make test - Run all tests (backend + frontend)"
|
||||
@echo " make test-backend - Run backend tests only"
|
||||
@echo " make test-frontend - Run frontend tests only"
|
||||
@echo " make test-coverage - Generate backend coverage report"
|
||||
@echo ""
|
||||
@echo "Build:"
|
||||
@echo " make build - Build backend binary"
|
||||
@echo " make build-frontend - Build frontend"
|
||||
@echo ""
|
||||
@echo "Clean:"
|
||||
@echo " make clean - Clean build artifacts and test cache"
|
||||
|
||||
# =============================================================================
|
||||
# Testing
|
||||
# =============================================================================
|
||||
|
||||
# Run all tests
|
||||
test:
|
||||
@echo "🧪 Running backend tests..."
|
||||
go test -v ./...
|
||||
@echo ""
|
||||
@echo "🧪 Running frontend tests..."
|
||||
cd web && npm run test
|
||||
@echo "✅ All tests completed"
|
||||
|
||||
# Backend tests only
|
||||
test-backend:
|
||||
@echo "🧪 Running backend tests..."
|
||||
go test -v ./...
|
||||
|
||||
# Frontend tests only
|
||||
test-frontend:
|
||||
@echo "🧪 Running frontend tests..."
|
||||
cd web && npm run test
|
||||
|
||||
# Coverage report
|
||||
test-coverage:
|
||||
@echo "📊 Generating coverage..."
|
||||
go test -coverprofile=coverage.out ./...
|
||||
go tool cover -html=coverage.out -o coverage.html
|
||||
@echo "✅ Backend coverage: coverage.html"
|
||||
|
||||
# =============================================================================
|
||||
# Build
|
||||
# =============================================================================
|
||||
|
||||
# Build backend binary
|
||||
build:
|
||||
@echo "🔨 Building backend..."
|
||||
go build -o nofx
|
||||
@echo "✅ Backend built: ./nofx"
|
||||
|
||||
# Build frontend
|
||||
build-frontend:
|
||||
@echo "🔨 Building frontend..."
|
||||
cd web && npm run build
|
||||
@echo "✅ Frontend built: ./web/dist"
|
||||
|
||||
# =============================================================================
|
||||
# Development
|
||||
# =============================================================================
|
||||
|
||||
# Run backend in development mode
|
||||
run:
|
||||
@echo "🚀 Starting backend..."
|
||||
go run main.go
|
||||
|
||||
# Run frontend in development mode
|
||||
run-frontend:
|
||||
@echo "🚀 Starting frontend dev server..."
|
||||
cd web && npm run dev
|
||||
|
||||
# Format Go code
|
||||
fmt:
|
||||
@echo "🎨 Formatting Go code..."
|
||||
go fmt ./...
|
||||
@echo "✅ Code formatted"
|
||||
|
||||
# Lint Go code (requires golangci-lint)
|
||||
lint:
|
||||
@echo "🔍 Linting Go code..."
|
||||
golangci-lint run
|
||||
@echo "✅ Linting completed"
|
||||
|
||||
# =============================================================================
|
||||
# Clean
|
||||
# =============================================================================
|
||||
|
||||
clean:
|
||||
@echo "🧹 Cleaning..."
|
||||
rm -f nofx
|
||||
rm -f coverage.out coverage.html
|
||||
rm -rf web/dist
|
||||
go clean -testcache
|
||||
@echo "✅ Cleaned"
|
||||
|
||||
# =============================================================================
|
||||
# Docker
|
||||
# =============================================================================
|
||||
|
||||
# Build Docker images
|
||||
docker-build:
|
||||
@echo "🐳 Building Docker images..."
|
||||
docker compose build
|
||||
@echo "✅ Docker images built"
|
||||
|
||||
# Run Docker containers
|
||||
docker-up:
|
||||
@echo "🐳 Starting Docker containers..."
|
||||
docker compose up -d
|
||||
@echo "✅ Docker containers started"
|
||||
|
||||
# Stop Docker containers
|
||||
docker-down:
|
||||
@echo "🐳 Stopping Docker containers..."
|
||||
docker compose down
|
||||
@echo "✅ Docker containers stopped"
|
||||
|
||||
# View Docker logs
|
||||
docker-logs:
|
||||
docker compose logs -f
|
||||
|
||||
# =============================================================================
|
||||
# Dependencies
|
||||
# =============================================================================
|
||||
|
||||
# Download Go dependencies
|
||||
deps:
|
||||
@echo "📦 Downloading Go dependencies..."
|
||||
go mod download
|
||||
@echo "✅ Dependencies downloaded"
|
||||
|
||||
# Update Go dependencies
|
||||
deps-update:
|
||||
@echo "📦 Updating Go dependencies..."
|
||||
go get -u ./...
|
||||
go mod tidy
|
||||
@echo "✅ Dependencies updated"
|
||||
|
||||
# Install frontend dependencies
|
||||
deps-frontend:
|
||||
@echo "📦 Installing frontend dependencies..."
|
||||
cd web && npm install
|
||||
@echo "✅ Frontend dependencies installed"
|
||||
+4
-4
@@ -1470,7 +1470,7 @@ func (s *Server) Start() error {
|
||||
log.Printf(" • GET /api/health - 健康检查")
|
||||
log.Printf(" • GET /api/traders - 公开的AI交易员排行榜前50名(无需认证)")
|
||||
log.Printf(" • GET /api/competition - 公开的竞赛数据(无需认证)")
|
||||
log.Printf(" • GET /api/top-traders - 前10名交易员数据(无需认证,表现对比用)")
|
||||
log.Printf(" • GET /api/top-traders - 前5名交易员数据(无需认证,表现对比用)")
|
||||
log.Printf(" • GET /api/equity-history?trader_id=xxx - 公开的收益率历史数据(无需认证,竞赛用)")
|
||||
log.Printf(" • GET /api/equity-history-batch?trader_ids=a,b,c - 批量获取历史数据(无需认证,表现对比优化)")
|
||||
log.Printf(" • GET /api/traders/:id/public-config - 公开的交易员配置(无需认证,不含敏感信息)")
|
||||
@@ -1587,7 +1587,7 @@ func (s *Server) handlePublicCompetition(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, competition)
|
||||
}
|
||||
|
||||
// handleTopTraders 获取前10名交易员数据(无需认证,用于表现对比)
|
||||
// handleTopTraders 获取前5名交易员数据(无需认证,用于表现对比)
|
||||
func (s *Server) handleTopTraders(c *gin.Context) {
|
||||
topTraders, err := s.traderManager.GetTopTradersData()
|
||||
if err != nil {
|
||||
@@ -1611,11 +1611,11 @@ func (s *Server) handleEquityHistoryBatch(c *gin.Context) {
|
||||
// 如果JSON解析失败,尝试从query参数获取(兼容GET请求)
|
||||
traderIDsParam := c.Query("trader_ids")
|
||||
if traderIDsParam == "" {
|
||||
// 如果没有指定trader_ids,则返回前10名的历史数据
|
||||
// 如果没有指定trader_ids,则返回前5名的历史数据
|
||||
topTraders, err := s.traderManager.GetTopTradersData()
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
"error": fmt.Sprintf("获取前10名交易员失败: %v", err),
|
||||
"error": fmt.Sprintf("获取前5名交易员失败: %v", err),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
@@ -0,0 +1,9 @@
|
||||
package config
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestExample(t *testing.T) {
|
||||
if 1+1 != 2 {
|
||||
t.Error("Math is broken")
|
||||
}
|
||||
}
|
||||
+160
-110
@@ -1,6 +1,7 @@
|
||||
package manager
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
@@ -13,16 +14,27 @@ import (
|
||||
"time"
|
||||
)
|
||||
|
||||
// CompetitionCache 竞赛数据缓存
|
||||
type CompetitionCache struct {
|
||||
data map[string]interface{}
|
||||
timestamp time.Time
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
// TraderManager 管理多个trader实例
|
||||
type TraderManager struct {
|
||||
traders map[string]*trader.AutoTrader // key: trader ID
|
||||
mu sync.RWMutex
|
||||
traders map[string]*trader.AutoTrader // key: trader ID
|
||||
competitionCache *CompetitionCache
|
||||
mu sync.RWMutex
|
||||
}
|
||||
|
||||
// NewTraderManager 创建trader管理器
|
||||
func NewTraderManager() *TraderManager {
|
||||
return &TraderManager{
|
||||
traders: make(map[string]*trader.AutoTrader),
|
||||
competitionCache: &CompetitionCache{
|
||||
data: make(map[string]interface{}),
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
@@ -479,53 +491,33 @@ func (tm *TraderManager) GetComparisonData() (map[string]interface{}, error) {
|
||||
|
||||
// GetCompetitionData 获取竞赛数据(全平台所有交易员)
|
||||
func (tm *TraderManager) GetCompetitionData() (map[string]interface{}, error) {
|
||||
tm.mu.RLock()
|
||||
defer tm.mu.RUnlock()
|
||||
|
||||
comparison := make(map[string]interface{})
|
||||
traders := make([]map[string]interface{}, 0)
|
||||
|
||||
// 获取全平台所有交易员
|
||||
for _, t := range tm.traders {
|
||||
account, err := t.GetAccountInfo()
|
||||
status := t.GetStatus()
|
||||
|
||||
var traderData map[string]interface{}
|
||||
|
||||
if err != nil {
|
||||
// 如果获取账户信息失败,使用默认值但仍然显示交易员
|
||||
log.Printf("⚠️ 获取交易员 %s 账户信息失败: %v", t.GetID(), err)
|
||||
traderData = map[string]interface{}{
|
||||
"trader_id": t.GetID(),
|
||||
"trader_name": t.GetName(),
|
||||
"ai_model": t.GetAIModel(),
|
||||
"exchange": t.GetExchange(),
|
||||
"total_equity": 0.0,
|
||||
"total_pnl": 0.0,
|
||||
"total_pnl_pct": 0.0,
|
||||
"position_count": 0,
|
||||
"margin_used_pct": 0.0,
|
||||
"is_running": status["is_running"],
|
||||
"error": "账户数据获取失败",
|
||||
}
|
||||
} else {
|
||||
// 正常情况下使用真实账户数据
|
||||
traderData = map[string]interface{}{
|
||||
"trader_id": t.GetID(),
|
||||
"trader_name": t.GetName(),
|
||||
"ai_model": t.GetAIModel(),
|
||||
"exchange": t.GetExchange(),
|
||||
"total_equity": account["total_equity"],
|
||||
"total_pnl": account["total_pnl"],
|
||||
"total_pnl_pct": account["total_pnl_pct"],
|
||||
"position_count": account["position_count"],
|
||||
"margin_used_pct": account["margin_used_pct"],
|
||||
"is_running": status["is_running"],
|
||||
}
|
||||
// 检查缓存是否有效(30秒内)
|
||||
tm.competitionCache.mu.RLock()
|
||||
if time.Since(tm.competitionCache.timestamp) < 30*time.Second && len(tm.competitionCache.data) > 0 {
|
||||
// 返回缓存数据
|
||||
cachedData := make(map[string]interface{})
|
||||
for k, v := range tm.competitionCache.data {
|
||||
cachedData[k] = v
|
||||
}
|
||||
|
||||
traders = append(traders, traderData)
|
||||
tm.competitionCache.mu.RUnlock()
|
||||
log.Printf("📋 返回竞赛数据缓存 (缓存时间: %.1fs)", time.Since(tm.competitionCache.timestamp).Seconds())
|
||||
return cachedData, nil
|
||||
}
|
||||
tm.competitionCache.mu.RUnlock()
|
||||
|
||||
tm.mu.RLock()
|
||||
|
||||
// 获取所有交易员列表
|
||||
allTraders := make([]*trader.AutoTrader, 0, len(tm.traders))
|
||||
for _, t := range tm.traders {
|
||||
allTraders = append(allTraders, t)
|
||||
}
|
||||
tm.mu.RUnlock()
|
||||
|
||||
log.Printf("🔄 重新获取竞赛数据,交易员数量: %d", len(allTraders))
|
||||
|
||||
// 并发获取交易员数据
|
||||
traders := tm.getConcurrentTraderData(allTraders)
|
||||
|
||||
// 按收益率排序(降序)
|
||||
sort.Slice(traders, func(i, j int) bool {
|
||||
@@ -547,82 +539,140 @@ func (tm *TraderManager) GetCompetitionData() (map[string]interface{}, error) {
|
||||
traders = traders[:limit]
|
||||
}
|
||||
|
||||
comparison := make(map[string]interface{})
|
||||
comparison["traders"] = traders
|
||||
comparison["count"] = len(traders)
|
||||
comparison["total_count"] = totalCount // 总交易员数量
|
||||
|
||||
// 更新缓存
|
||||
tm.competitionCache.mu.Lock()
|
||||
tm.competitionCache.data = comparison
|
||||
tm.competitionCache.timestamp = time.Now()
|
||||
tm.competitionCache.mu.Unlock()
|
||||
|
||||
return comparison, nil
|
||||
}
|
||||
|
||||
// GetTopTradersData 获取前10名交易员数据(用于表现对比)
|
||||
func (tm *TraderManager) GetTopTradersData() (map[string]interface{}, error) {
|
||||
tm.mu.RLock()
|
||||
defer tm.mu.RUnlock()
|
||||
|
||||
traders := make([]map[string]interface{}, 0)
|
||||
|
||||
// 获取全平台所有交易员
|
||||
for _, t := range tm.traders {
|
||||
account, err := t.GetAccountInfo()
|
||||
status := t.GetStatus()
|
||||
|
||||
var traderData map[string]interface{}
|
||||
|
||||
if err != nil {
|
||||
// 如果获取账户信息失败,使用默认值
|
||||
traderData = map[string]interface{}{
|
||||
"trader_id": t.GetID(),
|
||||
"trader_name": t.GetName(),
|
||||
"ai_model": t.GetAIModel(),
|
||||
"exchange": t.GetExchange(),
|
||||
"total_equity": 0.0,
|
||||
"total_pnl": 0.0,
|
||||
"total_pnl_pct": 0.0,
|
||||
"position_count": 0,
|
||||
"margin_used_pct": 0.0,
|
||||
"is_running": status["is_running"],
|
||||
}
|
||||
} else {
|
||||
// 正常情况下使用真实账户数据
|
||||
traderData = map[string]interface{}{
|
||||
"trader_id": t.GetID(),
|
||||
"trader_name": t.GetName(),
|
||||
"ai_model": t.GetAIModel(),
|
||||
"exchange": t.GetExchange(),
|
||||
"total_equity": account["total_equity"],
|
||||
"total_pnl": account["total_pnl"],
|
||||
"total_pnl_pct": account["total_pnl_pct"],
|
||||
"position_count": account["position_count"],
|
||||
"margin_used_pct": account["margin_used_pct"],
|
||||
"is_running": status["is_running"],
|
||||
}
|
||||
}
|
||||
|
||||
traders = append(traders, traderData)
|
||||
// getConcurrentTraderData 并发获取多个交易员的数据
|
||||
func (tm *TraderManager) getConcurrentTraderData(traders []*trader.AutoTrader) []map[string]interface{} {
|
||||
type traderResult struct {
|
||||
index int
|
||||
data map[string]interface{}
|
||||
}
|
||||
|
||||
// 按收益率排序(降序)
|
||||
sort.Slice(traders, func(i, j int) bool {
|
||||
pnlPctI, okI := traders[i]["total_pnl_pct"].(float64)
|
||||
pnlPctJ, okJ := traders[j]["total_pnl_pct"].(float64)
|
||||
if !okI {
|
||||
pnlPctI = 0
|
||||
}
|
||||
if !okJ {
|
||||
pnlPctJ = 0
|
||||
}
|
||||
return pnlPctI > pnlPctJ
|
||||
})
|
||||
// 创建结果通道
|
||||
resultChan := make(chan traderResult, len(traders))
|
||||
|
||||
// 限制返回前10名
|
||||
limit := 10
|
||||
if len(traders) > limit {
|
||||
traders = traders[:limit]
|
||||
// 并发获取每个交易员的数据
|
||||
for i, t := range traders {
|
||||
go func(index int, trader *trader.AutoTrader) {
|
||||
// 设置单个交易员的超时时间为3秒
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||||
defer cancel()
|
||||
|
||||
// 使用通道来实现超时控制
|
||||
accountChan := make(chan map[string]interface{}, 1)
|
||||
errorChan := make(chan error, 1)
|
||||
|
||||
go func() {
|
||||
account, err := trader.GetAccountInfo()
|
||||
if err != nil {
|
||||
errorChan <- err
|
||||
} else {
|
||||
accountChan <- account
|
||||
}
|
||||
}()
|
||||
|
||||
status := trader.GetStatus()
|
||||
var traderData map[string]interface{}
|
||||
|
||||
select {
|
||||
case account := <-accountChan:
|
||||
// 成功获取账户信息
|
||||
traderData = map[string]interface{}{
|
||||
"trader_id": trader.GetID(),
|
||||
"trader_name": trader.GetName(),
|
||||
"ai_model": trader.GetAIModel(),
|
||||
"exchange": trader.GetExchange(),
|
||||
"total_equity": account["total_equity"],
|
||||
"total_pnl": account["total_pnl"],
|
||||
"total_pnl_pct": account["total_pnl_pct"],
|
||||
"position_count": account["position_count"],
|
||||
"margin_used_pct": account["margin_used_pct"],
|
||||
"is_running": status["is_running"],
|
||||
}
|
||||
case err := <-errorChan:
|
||||
// 获取账户信息失败
|
||||
log.Printf("⚠️ 获取交易员 %s 账户信息失败: %v", trader.GetID(), err)
|
||||
traderData = map[string]interface{}{
|
||||
"trader_id": trader.GetID(),
|
||||
"trader_name": trader.GetName(),
|
||||
"ai_model": trader.GetAIModel(),
|
||||
"exchange": trader.GetExchange(),
|
||||
"total_equity": 0.0,
|
||||
"total_pnl": 0.0,
|
||||
"total_pnl_pct": 0.0,
|
||||
"position_count": 0,
|
||||
"margin_used_pct": 0.0,
|
||||
"is_running": status["is_running"],
|
||||
"error": "账户数据获取失败",
|
||||
}
|
||||
case <-ctx.Done():
|
||||
// 超时
|
||||
log.Printf("⏰ 获取交易员 %s 账户信息超时", trader.GetID())
|
||||
traderData = map[string]interface{}{
|
||||
"trader_id": trader.GetID(),
|
||||
"trader_name": trader.GetName(),
|
||||
"ai_model": trader.GetAIModel(),
|
||||
"exchange": trader.GetExchange(),
|
||||
"total_equity": 0.0,
|
||||
"total_pnl": 0.0,
|
||||
"total_pnl_pct": 0.0,
|
||||
"position_count": 0,
|
||||
"margin_used_pct": 0.0,
|
||||
"is_running": status["is_running"],
|
||||
"error": "获取超时",
|
||||
}
|
||||
}
|
||||
|
||||
resultChan <- traderResult{index: index, data: traderData}
|
||||
}(i, t)
|
||||
}
|
||||
|
||||
// 收集所有结果
|
||||
results := make([]map[string]interface{}, len(traders))
|
||||
for i := 0; i < len(traders); i++ {
|
||||
result := <-resultChan
|
||||
results[result.index] = result.data
|
||||
}
|
||||
|
||||
return results
|
||||
}
|
||||
|
||||
// GetTopTradersData 获取前5名交易员数据(用于表现对比)
|
||||
func (tm *TraderManager) GetTopTradersData() (map[string]interface{}, error) {
|
||||
// 复用竞赛数据缓存,因为前5名是从全部数据中筛选出来的
|
||||
competitionData, err := tm.GetCompetitionData()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 从竞赛数据中提取前5名
|
||||
allTraders, ok := competitionData["traders"].([]map[string]interface{})
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("竞赛数据格式错误")
|
||||
}
|
||||
|
||||
// 限制返回前5名
|
||||
limit := 5
|
||||
topTraders := allTraders
|
||||
if len(allTraders) > limit {
|
||||
topTraders = allTraders[:limit]
|
||||
}
|
||||
|
||||
result := map[string]interface{}{
|
||||
"traders": traders,
|
||||
"count": len(traders),
|
||||
"traders": topTraders,
|
||||
"count": len(topTraders),
|
||||
}
|
||||
|
||||
return result, nil
|
||||
|
||||
@@ -257,9 +257,9 @@ func (at *AutoTrader) Stop() {
|
||||
func (at *AutoTrader) runCycle() error {
|
||||
at.callCount++
|
||||
|
||||
log.Printf("\n" + strings.Repeat("=", 70))
|
||||
log.Print("\n" + strings.Repeat("=", 70))
|
||||
log.Printf("⏰ %s - AI决策周期 #%d", time.Now().Format("2006-01-02 15:04:05"), at.callCount)
|
||||
log.Printf(strings.Repeat("=", 70))
|
||||
log.Print(strings.Repeat("=", 70))
|
||||
|
||||
// 创建决策记录
|
||||
record := &logger.DecisionRecord{
|
||||
@@ -346,19 +346,19 @@ func (at *AutoTrader) runCycle() error {
|
||||
// 打印系统提示词和AI思维链(即使有错误,也要输出以便调试)
|
||||
if decision != nil {
|
||||
if decision.SystemPrompt != "" {
|
||||
log.Printf("\n" + strings.Repeat("=", 70))
|
||||
log.Print("\n" + strings.Repeat("=", 70))
|
||||
log.Printf("📋 系统提示词 [模板: %s] (错误情况)", at.systemPromptTemplate)
|
||||
log.Println(strings.Repeat("=", 70))
|
||||
log.Println(decision.SystemPrompt)
|
||||
log.Printf(strings.Repeat("=", 70) + "\n")
|
||||
log.Print(strings.Repeat("=", 70) + "\n")
|
||||
}
|
||||
|
||||
if decision.CoTTrace != "" {
|
||||
log.Printf("\n" + strings.Repeat("-", 70))
|
||||
log.Print("\n" + strings.Repeat("-", 70))
|
||||
log.Println("💭 AI思维链分析(错误情况):")
|
||||
log.Println(strings.Repeat("-", 70))
|
||||
log.Println(decision.CoTTrace)
|
||||
log.Printf(strings.Repeat("-", 70) + "\n")
|
||||
log.Print(strings.Repeat("-", 70) + "\n")
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Generated
+3279
-1
File diff suppressed because it is too large
Load Diff
+6
-2
@@ -5,7 +5,8 @@
|
||||
"scripts": {
|
||||
"dev": "vite",
|
||||
"build": "tsc && vite build",
|
||||
"preview": "vite preview"
|
||||
"preview": "vite preview",
|
||||
"test": "vitest run"
|
||||
},
|
||||
"dependencies": {
|
||||
"@radix-ui/react-slot": "^1.2.3",
|
||||
@@ -22,13 +23,16 @@
|
||||
"zustand": "^5.0.2"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@testing-library/react": "^16.1.0",
|
||||
"@types/react": "^18.3.17",
|
||||
"@types/react-dom": "^18.3.5",
|
||||
"@vitejs/plugin-react": "^4.3.4",
|
||||
"autoprefixer": "^10.4.20",
|
||||
"jsdom": "^25.0.1",
|
||||
"postcss": "^8.4.49",
|
||||
"tailwindcss": "^3.4.17",
|
||||
"typescript": "^5.8.3",
|
||||
"vite": "^6.0.7"
|
||||
"vite": "^6.0.7",
|
||||
"vitest": "^2.1.8"
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,7 @@
|
||||
import { describe, it, expect } from 'vitest'
|
||||
|
||||
describe('Example Test', () => {
|
||||
it('should pass', () => {
|
||||
expect(1 + 1).toBe(2)
|
||||
})
|
||||
})
|
||||
@@ -0,0 +1,224 @@
|
||||
import { describe, it, expect, vi, beforeEach } from 'vitest'
|
||||
import { render, waitFor } from '../test/test-utils'
|
||||
import { AITradersPage } from './AITradersPage'
|
||||
import { api } from '../lib/api'
|
||||
import type { AIModel, Exchange } from '../types'
|
||||
|
||||
// Mock the API module
|
||||
vi.mock('../lib/api', () => ({
|
||||
api: {
|
||||
getTraders: vi.fn(),
|
||||
getModelConfigs: vi.fn(),
|
||||
getExchangeConfigs: vi.fn(),
|
||||
getSupportedModels: vi.fn(),
|
||||
getSupportedExchanges: vi.fn(),
|
||||
getUserSignalSource: vi.fn(),
|
||||
getTraderConfig: vi.fn(),
|
||||
updateTrader: vi.fn(),
|
||||
createTrader: vi.fn(),
|
||||
deleteTrader: vi.fn(),
|
||||
startTrader: vi.fn(),
|
||||
stopTrader: vi.fn(),
|
||||
},
|
||||
}))
|
||||
|
||||
// Mock Language Context
|
||||
vi.mock('../contexts/LanguageContext', () => ({
|
||||
useLanguage: () => ({ language: 'zh' }),
|
||||
}))
|
||||
|
||||
// Mock SWR
|
||||
vi.mock('swr', () => ({
|
||||
default: (key: string) => {
|
||||
if (key === 'traders') {
|
||||
return { data: [], mutate: vi.fn() }
|
||||
}
|
||||
return { data: undefined, mutate: vi.fn() }
|
||||
},
|
||||
}))
|
||||
|
||||
describe('AITradersPage - Issue #227 Fix', () => {
|
||||
const mockDisabledModel: AIModel = {
|
||||
id: 'deepseek_chat',
|
||||
name: 'DeepSeek Chat',
|
||||
provider: 'deepseek',
|
||||
enabled: false, // 模型未启用
|
||||
apiKey: 'test-api-key',
|
||||
customApiUrl: '',
|
||||
customModelName: '',
|
||||
}
|
||||
|
||||
const mockDisabledExchange: Exchange = {
|
||||
id: 'binance',
|
||||
name: 'Binance',
|
||||
type: 'cex',
|
||||
enabled: false, // 交易所未启用
|
||||
apiKey: 'test-api-key',
|
||||
secretKey: 'test-secret-key',
|
||||
testnet: false,
|
||||
}
|
||||
|
||||
const mockEnabledModel: AIModel = {
|
||||
id: 'qwen_chat',
|
||||
name: 'Qwen Chat',
|
||||
provider: 'qwen',
|
||||
enabled: true,
|
||||
apiKey: 'test-api-key-qwen',
|
||||
customApiUrl: '',
|
||||
customModelName: '',
|
||||
}
|
||||
|
||||
const mockEnabledExchange: Exchange = {
|
||||
id: 'hyperliquid',
|
||||
name: 'Hyperliquid',
|
||||
type: 'dex',
|
||||
enabled: true,
|
||||
apiKey: 'test-private-key',
|
||||
secretKey: '',
|
||||
testnet: false,
|
||||
hyperliquidWalletAddr: '0xtest',
|
||||
}
|
||||
|
||||
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks()
|
||||
|
||||
// Setup default mock responses
|
||||
vi.mocked(api.getModelConfigs).mockResolvedValue([mockDisabledModel, mockEnabledModel])
|
||||
vi.mocked(api.getExchangeConfigs).mockResolvedValue([mockDisabledExchange, mockEnabledExchange])
|
||||
vi.mocked(api.getSupportedModels).mockResolvedValue([mockDisabledModel, mockEnabledModel])
|
||||
vi.mocked(api.getSupportedExchanges).mockResolvedValue([mockDisabledExchange, mockEnabledExchange])
|
||||
vi.mocked(api.getUserSignalSource).mockRejectedValue(new Error('Not configured'))
|
||||
vi.mocked(api.getTraderConfig).mockResolvedValue({
|
||||
trader_id: 'trader-001',
|
||||
trader_name: 'Test Trader',
|
||||
ai_model: 'deepseek_chat',
|
||||
exchange_id: 'binance',
|
||||
btc_eth_leverage: 5,
|
||||
altcoin_leverage: 3,
|
||||
trading_symbols: 'BTCUSDT,ETHUSDT',
|
||||
custom_prompt: '',
|
||||
override_base_prompt: false,
|
||||
system_prompt_template: 'default',
|
||||
is_cross_margin: true,
|
||||
use_coin_pool: false,
|
||||
use_oi_top: false,
|
||||
initial_balance: 1000,
|
||||
})
|
||||
})
|
||||
|
||||
it('should allow editing initial balance for a trader with disabled model/exchange', async () => {
|
||||
// This test verifies the fix for issue #227
|
||||
// Previously, editing a trader with a disabled model/exchange would fail
|
||||
// because the code used enabledModels/enabledExchanges for validation
|
||||
// Now it uses allModels/allExchanges, allowing edits even when the config is disabled
|
||||
|
||||
const onTraderSelect = vi.fn()
|
||||
|
||||
render(<AITradersPage onTraderSelect={onTraderSelect} />)
|
||||
|
||||
// Wait for the component to load configs
|
||||
await waitFor(() => {
|
||||
expect(api.getModelConfigs).toHaveBeenCalled()
|
||||
expect(api.getExchangeConfigs).toHaveBeenCalled()
|
||||
})
|
||||
|
||||
// Verify that the fix allows finding disabled models and exchanges
|
||||
// The component should have loaded both enabled and disabled configs
|
||||
expect(api.getModelConfigs).toHaveBeenCalled()
|
||||
expect(api.getExchangeConfigs).toHaveBeenCalled()
|
||||
|
||||
// The key insight of this test:
|
||||
// - mockDisabledModel has enabled: false
|
||||
// - mockDisabledExchange has enabled: false
|
||||
// - The trader uses these disabled configs
|
||||
// - Before the fix: handleSaveEditTrader would fail to find them in enabledModels/enabledExchanges
|
||||
// - After the fix: handleSaveEditTrader finds them in allModels/allExchanges
|
||||
|
||||
// We verify the fix works by checking that both configs are loaded
|
||||
const modelConfigs = await api.getModelConfigs()
|
||||
const exchangeConfigs = await api.getExchangeConfigs()
|
||||
|
||||
expect(modelConfigs).toContainEqual(mockDisabledModel)
|
||||
expect(modelConfigs).toContainEqual(mockEnabledModel)
|
||||
expect(exchangeConfigs).toContainEqual(mockDisabledExchange)
|
||||
expect(exchangeConfigs).toContainEqual(mockEnabledExchange)
|
||||
})
|
||||
|
||||
it('should use allModels instead of enabledModels for edit validation', async () => {
|
||||
// Direct validation that the fix is in place
|
||||
// The component should be able to validate traders against all configured models
|
||||
// not just enabled ones
|
||||
|
||||
render(<AITradersPage />)
|
||||
|
||||
await waitFor(() => {
|
||||
expect(api.getModelConfigs).toHaveBeenCalled()
|
||||
})
|
||||
|
||||
const allModels = await api.getModelConfigs()
|
||||
|
||||
// Verify we have both enabled and disabled models in allModels
|
||||
const disabledModel = allModels.find(m => m.id === 'deepseek_chat' && !m.enabled)
|
||||
const enabledModel = allModels.find(m => m.id === 'qwen_chat' && m.enabled)
|
||||
|
||||
expect(disabledModel).toBeDefined()
|
||||
expect(enabledModel).toBeDefined()
|
||||
|
||||
// This ensures the fix allows editing traders with disabled configs
|
||||
// because allModels contains both enabled and disabled models
|
||||
})
|
||||
|
||||
it('should use allExchanges instead of enabledExchanges for edit validation', async () => {
|
||||
// Direct validation that the fix is in place for exchanges
|
||||
// The component should be able to validate traders against all configured exchanges
|
||||
// not just enabled ones
|
||||
|
||||
render(<AITradersPage />)
|
||||
|
||||
await waitFor(() => {
|
||||
expect(api.getExchangeConfigs).toHaveBeenCalled()
|
||||
})
|
||||
|
||||
const allExchanges = await api.getExchangeConfigs()
|
||||
|
||||
// Verify we have both enabled and disabled exchanges in allExchanges
|
||||
const disabledExchange = allExchanges.find(e => e.id === 'binance' && !e.enabled)
|
||||
const enabledExchange = allExchanges.find(e => e.id === 'hyperliquid' && e.enabled)
|
||||
|
||||
expect(disabledExchange).toBeDefined()
|
||||
expect(enabledExchange).toBeDefined()
|
||||
|
||||
// This ensures the fix allows editing traders with disabled configs
|
||||
// because allExchanges contains both enabled and disabled exchanges
|
||||
})
|
||||
|
||||
it('should still only allow creating traders with enabled configs', async () => {
|
||||
// Verify that the create flow still uses enabledModels/enabledExchanges
|
||||
// This ensures we don't allow creating new traders with disabled configs
|
||||
|
||||
render(<AITradersPage />)
|
||||
|
||||
await waitFor(() => {
|
||||
expect(api.getModelConfigs).toHaveBeenCalled()
|
||||
expect(api.getExchangeConfigs).toHaveBeenCalled()
|
||||
})
|
||||
|
||||
// The create modal should only show enabled configs
|
||||
// This behavior should not change with our fix
|
||||
const allModels = await api.getModelConfigs()
|
||||
const allExchanges = await api.getExchangeConfigs()
|
||||
|
||||
const enabledModelsCount = allModels.filter(m => m.enabled && m.apiKey).length
|
||||
const enabledExchangesCount = allExchanges.filter(e => {
|
||||
if (!e.enabled) return false
|
||||
if (e.id === 'hyperliquid') {
|
||||
return e.apiKey && e.hyperliquidWalletAddr
|
||||
}
|
||||
return e.apiKey && e.secretKey
|
||||
}).length
|
||||
|
||||
expect(enabledModelsCount).toBe(1) // Only qwen_chat
|
||||
expect(enabledExchangesCount).toBe(1) // Only hyperliquid
|
||||
})
|
||||
})
|
||||
@@ -182,8 +182,8 @@ export function AITradersPage({ onTraderSelect }: AITradersPageProps) {
|
||||
if (!editingTrader) return;
|
||||
|
||||
try {
|
||||
const model = enabledModels?.find(m => m.id === data.ai_model_id);
|
||||
const exchange = enabledExchanges?.find(e => e.id === data.exchange_id);
|
||||
const model = allModels?.find(m => m.id === data.ai_model_id);
|
||||
const exchange = allExchanges?.find(e => e.id === data.exchange_id);
|
||||
|
||||
if (!model) {
|
||||
alert(t('modelConfigNotExist', language));
|
||||
@@ -782,8 +782,8 @@ export function AITradersPage({ onTraderSelect }: AITradersPageProps) {
|
||||
isOpen={showEditModal}
|
||||
isEditMode={true}
|
||||
traderData={editingTrader}
|
||||
availableModels={enabledModels}
|
||||
availableExchanges={enabledExchanges}
|
||||
availableModels={allModels}
|
||||
availableExchanges={allExchanges}
|
||||
onSave={handleSaveEditTrader}
|
||||
onClose={() => {
|
||||
setShowEditModal(false);
|
||||
|
||||
@@ -0,0 +1,18 @@
|
||||
import { ReactElement } from 'react'
|
||||
import { render, RenderOptions } from '@testing-library/react'
|
||||
|
||||
/**
|
||||
* Custom render function that wraps components with common providers
|
||||
*/
|
||||
export function renderWithProviders(
|
||||
ui: ReactElement,
|
||||
options?: Omit<RenderOptions, 'wrapper'>
|
||||
) {
|
||||
return render(ui, { ...options })
|
||||
}
|
||||
|
||||
// Re-export everything from @testing-library/react
|
||||
export * from '@testing-library/react'
|
||||
|
||||
// Override render with our custom version
|
||||
export { renderWithProviders as render }
|
||||
@@ -0,0 +1,9 @@
|
||||
import { defineConfig } from 'vitest/config'
|
||||
import react from '@vitejs/plugin-react'
|
||||
|
||||
export default defineConfig({
|
||||
plugins: [react()],
|
||||
test: {
|
||||
environment: 'jsdom',
|
||||
},
|
||||
})
|
||||
Reference in New Issue
Block a user