26 lines
922 B
Go
26 lines
922 B
Go
|
package contentmodel
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestValidateInputAllowsComposeOllamaEndpoint(t *testing.T) {
|
||
|
|
input := CreateInput{
|
||
|
|
TenantID: "10000000-0000-0000-0000-000000000001", DisplayName: "本地 Qwen",
|
||
|
|
Provider: "openai_compatible", BaseURL: "http://ollama:11434/v1", ModelID: "qwen2.5:7b",
|
||
|
|
APIKey: "ollama-local-key", Temperature: 0.1, MaxTokens: 1200,
|
||
|
|
}
|
||
|
|
if err := validateInput(input); err != nil {
|
||
|
|
t.Fatalf("compose Ollama endpoint was rejected: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestValidateInputRejectsUnapprovedPlainHTTPEndpoint(t *testing.T) {
|
||
|
|
input := CreateInput{
|
||
|
|
TenantID: "10000000-0000-0000-0000-000000000001", DisplayName: "不安全端点",
|
||
|
|
Provider: "openai_compatible", BaseURL: "http://metadata.internal/v1", ModelID: "model",
|
||
|
|
APIKey: "test-api-key", Temperature: 0.1, MaxTokens: 1200,
|
||
|
|
}
|
||
|
|
if err := validateInput(input); err == nil {
|
||
|
|
t.Fatal("unapproved plaintext endpoint was accepted")
|
||
|
|
}
|
||
|
|
}
|