Files
Atlas/internal/domain/provider_compat_test.go
T
renjue 4009214cbc
CI / docker (push) Successful in 7m4s
实现 Atlas AI 平台一期能力并完成品牌化改造。
引入用户端对话/生图与管理端 Provider 配置,统一 invoke 编排、Gateway、请求日志与模型管理;前端按设计规范拆分布局,Go module 更名为 github.com/rose_cat707/Atlas;生图长任务 SSE 推送;管理端登录防爆破;协议与服务类型联动;Replicate 模型同步修复;用户端创作台 UI 重构;管理端创作历史替代请求日志。

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-06-29 21:08:57 +08:00

37 lines
947 B
Go

package domain
import "testing"
func TestValidateProviderBinding(t *testing.T) {
cases := []struct {
protocol string
serviceType string
ok bool
}{
{"openai", "chat", true},
{"replicate", "text_to_image", true},
{"replicate", "image_to_image", true},
{"openai", "text_to_image", false},
{"replicate", "chat", false},
{"unknown", "chat", false},
}
for _, tc := range cases {
err := ValidateProviderBinding(tc.protocol, tc.serviceType)
if tc.ok && err != nil {
t.Fatalf("%s/%s expected ok, got %v", tc.protocol, tc.serviceType, err)
}
if !tc.ok && err == nil {
t.Fatalf("%s/%s expected error", tc.protocol, tc.serviceType)
}
}
}
func TestDefaultBindings(t *testing.T) {
if got := DefaultProtocol(ServiceTextToImage); got != "replicate" {
t.Fatalf("expected replicate, got %s", got)
}
if got := DefaultServiceType("openai"); got != ServiceChat {
t.Fatalf("expected chat, got %s", got)
}
}