89104bd0db
## 设计文档 - multi_role_permission_design: 多角色权限设计 (CONDITIONAL GO) - audit_log_enhancement_design: 审计日志增强 (CONDITIONAL GO) - routing_strategy_template_design: 路由策略模板 (CONDITIONAL GO) - sso_saml_technical_research: SSO/SAML调研 (CONDITIONAL GO) - compliance_capability_package_design: 合规能力包设计 (CONDITIONAL GO) ## TDD开发成果 - IAM模块: supply-api/internal/iam/ (111个测试) - 审计日志模块: supply-api/internal/audit/ (40+测试) - 路由策略模块: gateway/internal/router/ (33+测试) - 合规能力包: gateway/internal/compliance/ + scripts/ci/compliance/ ## 规范文档 - parallel_agent_output_quality_standards: 并行Agent产出质量规范 - project_experience_summary: 项目经验总结 (v2) - 2026-04-02-p1-p2-tdd-execution-plan: TDD执行计划 ## 评审报告 - 5个CONDITIONAL GO设计文档评审报告 - fix_verification_report: 修复验证报告 - full_verification_report: 全面质量验证报告 - tdd_module_quality_verification: TDD模块质量验证 - tdd_execution_summary: TDD执行总结 依据: Superpowers执行框架 + TDD规范
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
package strategy
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"lijiaoqiao/gateway/internal/adapter"
|
|
)
|
|
|
|
// TestStrategyTemplate_Interface 验证策略模板接口
|
|
func TestStrategyTemplate_Interface(t *testing.T) {
|
|
// 所有策略实现必须实现SelectProvider, Name, Type方法
|
|
|
|
// 创建策略实现示例
|
|
costBased := &CostBasedTemplate{
|
|
name: "CostBased",
|
|
}
|
|
|
|
aware := &CostAwareTemplate{
|
|
name: "CostAware",
|
|
}
|
|
|
|
// 验证实现了StrategyTemplate接口
|
|
var _ StrategyTemplate = costBased
|
|
var _ StrategyTemplate = aware
|
|
|
|
// 验证方法
|
|
assert.Equal(t, "CostBased", costBased.Name())
|
|
assert.Equal(t, "cost_based", costBased.Type())
|
|
|
|
assert.Equal(t, "CostAware", aware.Name())
|
|
assert.Equal(t, "cost_aware", aware.Type())
|
|
}
|
|
|
|
// TestStrategyTemplate_SelectProvider_Signature 验证SelectProvider方法签名
|
|
func TestStrategyTemplate_SelectProvider_Signature(t *testing.T) {
|
|
req := &RoutingRequest{
|
|
Model: "gpt-4",
|
|
UserID: "user123",
|
|
TenantID: "tenant1",
|
|
MaxCost: 1.0,
|
|
MaxLatency: 1000,
|
|
}
|
|
|
|
// 验证返回值 - 创建一个有providers的模板
|
|
template := &CostBasedTemplate{
|
|
name: "test",
|
|
maxCostPer1KTokens: 1.0,
|
|
providers: make(map[string]adapter.ProviderAdapter),
|
|
}
|
|
template.providers["test"] = &MockProvider{
|
|
name: "test",
|
|
costPer1KTokens: 0.5,
|
|
available: true,
|
|
models: []string{"gpt-4"},
|
|
}
|
|
|
|
decision, err := template.SelectProvider(context.Background(), req)
|
|
|
|
// 接口实现应该返回决策或错误
|
|
assert.NotNil(t, decision)
|
|
assert.Nil(t, err)
|
|
}
|