96 lines
3.2 KiB
Go
96 lines
3.2 KiB
Go
package creator
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestBailianClientGeneratesAndParsesDecisions(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost || r.URL.Path != "/v1/chat/completions" {
|
|
t.Fatalf("unexpected request: %s %s", r.Method, r.URL.Path)
|
|
}
|
|
if r.Header.Get("Authorization") != "Bearer test-key" {
|
|
t.Fatal("missing authorization")
|
|
}
|
|
var request bailianChatRequest
|
|
if err := json.NewDecoder(r.Body).Decode(&request); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if request.Model != "qwen-test" || len(request.Messages) != 2 {
|
|
t.Fatalf("unexpected request body: %#v", request)
|
|
}
|
|
_ = json.NewEncoder(w).Encode(map[string]any{"choices": []any{map[string]any{
|
|
"message": map[string]string{"role": "assistant", "content": `{"match":true,"reason":"相关"}`},
|
|
}}})
|
|
}))
|
|
defer server.Close()
|
|
|
|
client, err := NewBailianClient(server.URL+"/v1", "test-key", "qwen-test", server.Client())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
text, err := client.Generate(context.Background(), "要求", "输入")
|
|
if err != nil || text == "" {
|
|
t.Fatalf("Generate() = %q, %v", text, err)
|
|
}
|
|
match, reason, err := client.MatchLead(context.Background(), "作品", "评论", "要求")
|
|
if err != nil || !match || reason != "相关" {
|
|
t.Fatalf("MatchLead() = %v, %q, %v", match, reason, err)
|
|
}
|
|
match, reason, err = client.MatchTheme(context.Background(), "标题", "正文", "主题")
|
|
if err != nil || !match || reason != "相关" {
|
|
t.Fatalf("MatchTheme() = %v, %q, %v", match, reason, err)
|
|
}
|
|
}
|
|
|
|
func TestBailianClientRejectsHTTPAndMalformedResponses(t *testing.T) {
|
|
responses := []string{"not-json", `{"choices":[]}`}
|
|
for _, body := range responses {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.WriteHeader(http.StatusBadGateway)
|
|
_, _ = w.Write([]byte(body))
|
|
}))
|
|
client, err := NewBailianClient(server.URL, "key", "model", server.Client())
|
|
if err != nil {
|
|
server.Close()
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := client.Generate(context.Background(), "instruction", "input"); err == nil {
|
|
server.Close()
|
|
t.Fatal("expected HTTP failure")
|
|
}
|
|
server.Close()
|
|
}
|
|
var nilClient *BailianClient
|
|
if _, err := nilClient.Generate(context.Background(), "instruction", "input"); err == nil {
|
|
t.Fatal("nil client must fail closed")
|
|
}
|
|
}
|
|
|
|
func TestBailianClientRejectsIncompleteConfiguration(t *testing.T) {
|
|
if _, err := NewBailianClient("", "", "qwen-test", nil); err == nil {
|
|
t.Fatal("expected missing key to fail")
|
|
}
|
|
if _, err := NewBailianClient("", "test-key", "", nil); err == nil {
|
|
t.Fatal("expected missing model to fail")
|
|
}
|
|
}
|
|
|
|
func TestBailianClientRejectsMissingDecisionField(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
_, _ = w.Write([]byte(`{"choices":[{"message":{"content":"{}"}}]}`))
|
|
}))
|
|
defer server.Close()
|
|
client, err := NewBailianClient(server.URL, "test-key", "qwen-test", server.Client())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, _, err := client.MatchLead(context.Background(), "work", "comment", "requirement"); err == nil {
|
|
t.Fatal("expected missing match decision to fail closed")
|
|
}
|
|
}
|