9b9a7f4a72
- 多维表格（canvas-batch-table-node/use-canvas-batch-table/canvas-batch-table 等）以 PR#530 实现为主 - 保留官方 main 全部新功能：banner/画布版本历史/提示词编辑/方舟 Agent Plan/支付导出 - 移植本地定制：ddcat 多参考图校验、万有引力 lxmone 连接错误提示、Description v20(v24)、自建插件(a6api/metaso-h3/antigravity-proxy/lxmone)、seccomp、Windows 脚本 - 迁移版本升至 v24（官方 v20-23 banner/版本历史 + 我们的 Description 列） - 修复官方缺失的 markdownPlainText import
66 lines
2.3 KiB
Go
66 lines
2.3 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestOpenAIImageRejectsInvalidSizeBeforeSubmission(t *testing.T) {
|
|
for _, size := range []string{"banana", "7:0", "0x1024", "-1x1024", "4097x1024", "1024x5000"} {
|
|
t.Run(size, func(t *testing.T) {
|
|
_, err := runImageTask(context.Background(), canvasGenerationInput{Mode: "image", Config: providerConfig{InterfaceType: "openai-image", BaseURL: "https://api.ddcat.pronhubcn.com/v1", Size: size}})
|
|
if err == nil || !strings.Contains(err.Error(), "尺寸") {
|
|
t.Fatalf("expected local size rejection: %v", err)
|
|
}
|
|
})
|
|
}
|
|
for _, size := range []string{"", "auto", "1:1", "16:9", "1024x1024", "4096x4096"} {
|
|
if err := validateOpenAIImageInput(canvasGenerationInput{Config: providerConfig{Size: size}}); err != nil {
|
|
t.Fatalf("valid size %q: %v", size, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestOpenAIImageRejectsMaskWithoutSourceBeforePlugin(t *testing.T) {
|
|
_, err := runImageTask(context.Background(), canvasGenerationInput{Mode: "image", Config: providerConfig{InterfaceType: "openai-image"}, Mask: &providerMedia{ID: "mask"}})
|
|
if err == nil || !strings.Contains(err.Error(), "源图片") {
|
|
t.Fatalf("expected missing source rejection: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestProviderConnectionErrorPreservesCause(t *testing.T) {
|
|
for _, cause := range []error{io.EOF, io.ErrUnexpectedEOF} {
|
|
err := providerConnectionError(cause)
|
|
if !errors.Is(err, cause) || !strings.Contains(err.Error(), "扣费记录") {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
}
|
|
if providerConnectionError(context.Canceled) != context.Canceled {
|
|
t.Fatal("cancellation must remain unchanged")
|
|
}
|
|
}
|
|
|
|
func TestImageSubmissionEOFIsNotAutomaticallyReplayed(t *testing.T) {
|
|
t.Setenv("CANVAS_ALLOW_PRIVATE_UPSTREAMS", "true")
|
|
calls := 0
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
calls++
|
|
conn, _, err := w.(http.Hijacker).Hijack()
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
conn.Close()
|
|
}))
|
|
defer server.Close()
|
|
err := postJSON(context.Background(), providerConfig{BaseURL: server.URL}, "/v1/images/generations", map[string]string{"prompt": "cup"}, &map[string]interface{}{})
|
|
if err == nil || !strings.Contains(err.Error(), "连接提前关闭") || calls != 1 {
|
|
t.Fatalf("calls=%d error=%v", calls, err)
|
|
}
|
|
}
|