tt/ttstd-infinite-canvas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(*): 稳定期治理 - 收敛核心架构、错误语义并修复跨项目数据竞态

Browse Source
This commit is contained in:
ddCat
2026-09-09 22:02:10 +08:00
parent bacd502f5c
commit b5f3413ded
298 changed files with 12681 additions and 26428 deletions
Download Patch File Download Diff File Expand all files Collapse all files

188
backend/internal/service/task_creation_test.go
View File

@@ -1,6 +1,7 @@
package service
import (
"encoding/json"
"strings"
"testing"
@@ -45,9 +46,18 @@ func TestResolveTaskModelSelectionAllowsExplicitSystemChannelWhenFrontendModelsE
t.Fatal(err)
}
channel := model.ModelChannel{ID: "channel-1", Scope: model.ChannelScopeSystem, Enabled: true, Name: "Agnes"}
capabilityJSON, err := json.Marshal(DefaultModelCapabilityConfigForModel(string(model.ChannelInterfaceNewAPIVideo), "agnes-video-2.5"))
if err != nil {
t.Fatal(err)
}
channelModel := model.ChannelModel{
ID: "channel-model-1", ChannelID: channel.ID, ModelKey: "agnes-video-2.5", Capability: "video",
Protocol: model.ChannelInterfaceNewAPIVideo, BillingMode: "per_second", PriceConfigured: true, Enabled: true,
CapabilityConfigJSON: string(capabilityJSON),
}
priceTier := model.ChannelModelPriceTier{
ID: "tier-1", ChannelModelID: channelModel.ID, SelectorKey: `{}`, SelectorJSON: `{}`,
BillingMode: "per_second", UnitPriceMicrocredits: 1, PriceConfigured: true, Enabled: true,
}
if err := db.Create(&channel).Error; err != nil {
t.Fatal(err)
@@ -55,6 +65,9 @@ func TestResolveTaskModelSelectionAllowsExplicitSystemChannelWhenFrontendModelsE
if err := db.Create(&channelModel).Error; err != nil {
t.Fatal(err)
}
if err := db.Create(&priceTier).Error; err != nil {
t.Fatal(err)
}
input, err := normalizeTaskInput(map[string]any{
"mode": "video",
"config": providerConfig{ChannelID: channel.ID, Model: channelModel.ModelKey, InterfaceType: string(channelModel.Protocol)},
@@ -101,3 +114,178 @@ func TestResolveTaskModelSelectionStillRequiresLogicalModelWithoutExplicitSystem
t.Fatalf("resolveTaskModelSelection() error = %v, want logicalModelId validation", err)
}
}
func TestResolveSystemChannelModelSelectionRebuildsAuthoritativeExecutionSpec(t *testing.T) {
profile := DefaultModelCapabilityConfigForModel(string(model.ChannelInterfaceGrokImage), "grok-image")
svc, _, channel, channelModel := createSystemChannelSelectionFixture(t, "image", model.ChannelInterfaceGrokImage, profile, []model.ChannelModelPriceTier{
newSelectionPriceTier("tier-1k", `{"operation":"text_to_image","quality":"1k"}`, "provider-image-1k", "fixed_request"),
newSelectionPriceTier("tier-2k", `{"operation":"text_to_image","quality":"2k"}`, "provider-image-2k", "fixed_request"),
})
input := map[string]any{
"mode": "image",
"config": map[string]any{
"channelId": channel.ID,
"model": channelModel.ModelKey,
"quality": "2k",
"priceTierId": "client-tier",
"providerModelKey": "client-provider-model",
"interfaceType": "runninghub-workflow-image",
"apiFormat": "client-format",
"baseUrl": "https://attacker.invalid/v1",
"apiKey": "client-api-key",
"secretKey": "client-secret",
"headers": []any{map[string]any{"name": "Authorization", "value": "client-token"}},
"capabilityConfig": map[string]any{"image": map[string]any{"maxOutputs": 99}},
},
// capabilityOptions 是创作端表达的最终规格，必须覆盖旧页面缓存到 config 的同名值。
"capabilityOptions": map[string]any{"quality": "1k", "size": "1:1"},
}
resolved, err := svc.resolveSystemChannelModelSelection(input, "canvas_image", "")
if err != nil {
t.Fatalf("resolveSystemChannelModelSelection() error = %v", err)
}
config := resolved["config"].(map[string]any)
for _, forbidden := range []string{"baseUrl", "apiKey", "secretKey", "headers", "capabilityConfig"} {
if _, exists := config[forbidden]; exists {
t.Fatalf("config[%q] must be removed from a system-channel request: %#v", forbidden, config[forbidden])
}
}
if got := config["quality"]; got != "1k" {
t.Fatalf("quality = %#v, want capabilityOptions value 1k", got)
}
if got := config["priceTierId"]; got != "tier-1k" {
t.Fatalf("priceTierId = %#v, want server-selected tier-1k", got)
}
if got := config["providerModelKey"]; got != "provider-image-1k" {
t.Fatalf("providerModelKey = %#v, want server-selected provider-image-1k", got)
}
if got := config["interfaceType"]; got != string(model.ChannelInterfaceGrokImage) {
t.Fatalf("interfaceType = %#v, want %q", got, model.ChannelInterfaceGrokImage)
}
if got := config["apiFormat"]; got != "openai" {
t.Fatalf("apiFormat = %#v, want openai", got)
}
options := resolved["capabilityOptions"].(map[string]any)
if options["quality"] != config["quality"] || options["size"] != config["size"] || options["count"] != config["count"] {
t.Fatalf("capabilityOptions and provider config diverged: options=%#v config=%#v", options, config)
}
}
func TestResolveSystemChannelModelSelectionAppliesServerDefaults(t *testing.T) {
profile := DefaultModelCapabilityConfigForModel(string(model.ChannelInterfaceGrokImage), "grok-image")
svc, _, channel, channelModel := createSystemChannelSelectionFixture(t, "image", model.ChannelInterfaceGrokImage, profile, []model.ChannelModelPriceTier{
newSelectionPriceTier("tier-default", `{"operation":"text_to_image","quality":"2k"}`, "provider-image-default", "fixed_request"),
})
resolved, err := svc.resolveSystemChannelModelSelection(map[string]any{
"mode": "image",
"config": map[string]any{"channelId": channel.ID, "model": channelModel.ModelKey},
}, "canvas_image", "")
if err != nil {
t.Fatalf("resolveSystemChannelModelSelection() error = %v", err)
}
config := resolved["config"].(map[string]any)
if config["size"] != "1:1" || config["quality"] != "2k" || config["count"] != 1 || config["transparentBackground"] != false {
t.Fatalf("server defaults were not applied: %#v", config)
}
if config["priceTierId"] != "tier-default" || config["providerModelKey"] != "provider-image-default" {
t.Fatalf("server price selection was not persisted: %#v", config)
}
}
func TestResolveSystemChannelModelSelectionRejectsUnsupportedRequest(t *testing.T) {
profile := DefaultModelCapabilityConfigForModel(string(model.ChannelInterfaceGrokImage), "grok-image")
svc, _, channel, channelModel := createSystemChannelSelectionFixture(t, "image", model.ChannelInterfaceGrokImage, profile, []model.ChannelModelPriceTier{
newSelectionPriceTier("tier-any", `{}`, "provider-image", "fixed_request"),
})
tests := []struct {
name string
input map[string]any
}{
{
name: "too many reference images",
input: map[string]any{
"mode": "image", "referenceImages": []any{"a", "b"},
"config": map[string]any{"channelId": channel.ID, "model": channelModel.ModelKey},
},
},
{
name: "unsupported quality",
input: map[string]any{
"mode": "image", "config": map[string]any{"channelId": channel.ID, "model": channelModel.ModelKey},
"capabilityOptions": map[string]any{"quality": "4k", "size": "1:1"},
},
},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
if _, err := svc.resolveSystemChannelModelSelection(test.input, "canvas_image", ""); err == nil {
t.Fatal("resolveSystemChannelModelSelection() should reject a request outside the server capability contract")
}
})
}
videoProfile := DefaultModelCapabilityConfigForModel(string(model.ChannelInterfaceNewAPIVideo), "video-model")
videoSvc, _, videoChannel, videoModel := createSystemChannelSelectionFixture(t, "video", model.ChannelInterfaceNewAPIVideo, videoProfile, []model.ChannelModelPriceTier{
newSelectionPriceTier("video-tier", `{}`, "provider-video", "per_second"),
})
if _, err := videoSvc.resolveSystemChannelModelSelection(map[string]any{
"mode": "video",
"config": map[string]any{"channelId": videoChannel.ID, "model": videoModel.ModelKey},
}, "canvas_video", "audio_to_video"); err == nil {
t.Fatal("resolveSystemChannelModelSelection() should reject an unsupported operation")
}
}
func TestResolveSystemChannelModelSelectionRejectsCorruptCapabilityConfig(t *testing.T) {
profile := DefaultModelCapabilityConfigForModel(string(model.ChannelInterfaceGrokImage), "grok-image")
svc, db, channel, channelModel := createSystemChannelSelectionFixture(t, "image", model.ChannelInterfaceGrokImage, profile, []model.ChannelModelPriceTier{
newSelectionPriceTier("tier-any", `{}`, "provider-image", "fixed_request"),
})
if err := db.Model(&model.ChannelModel{}).Where("id = ?", channelModel.ID).Update("capability_config_json", "{").Error; err != nil {
t.Fatal(err)
}
if _, err := svc.resolveSystemChannelModelSelection(map[string]any{
"mode": "image",
"config": map[string]any{"channelId": channel.ID, "model": channelModel.ModelKey},
}, "canvas_image", ""); err == nil {
t.Fatal("resolveSystemChannelModelSelection() should fail closed for corrupt persisted capability config")
}
}
func createSystemChannelSelectionFixture(t *testing.T, capability string, protocol model.ChannelInterfaceType, profile *ModelCapabilityConfig, tiers []model.ChannelModelPriceTier) (*Service, *gorm.DB, model.ModelChannel, model.ChannelModel) {
t.Helper()
svc, db := newChannelModelTestService(t)
channel := model.ModelChannel{ID: newID(), Scope: model.ChannelScopeSystem, Enabled: true, Name: "System Channel", APIFormat: "legacy"}
capabilityJSON, err := json.Marshal(profile)
if err != nil {
t.Fatal(err)
}
channelModel := model.ChannelModel{
ID: newID(), ChannelID: channel.ID, ModelKey: capability + "-model", ProviderModelKey: "provider-default",
Capability: capability, Protocol: protocol, Enabled: true, CapabilityConfigJSON: string(capabilityJSON),
}
if err := db.Create(&channel).Error; err != nil {
t.Fatal(err)
}
if err := db.Create(&channelModel).Error; err != nil {
t.Fatal(err)
}
for index := range tiers {
tiers[index].ChannelModelID = channelModel.ID
if err := db.Create(&tiers[index]).Error; err != nil {
t.Fatal(err)
}
}
return svc, db, channel, channelModel
}
func newSelectionPriceTier(id string, selector string, providerModelKey string, billingMode string) model.ChannelModelPriceTier {
return model.ChannelModelPriceTier{
ID: id, SelectorKey: selector, SelectorJSON: selector, ProviderModelKey: providerModelKey,
BillingMode: billingMode, UnitPriceMicrocredits: 10, PriceConfigured: true, Enabled: true,
}
}