feat: harden agent tooling compatibility
This commit is contained in:
@@ -1,6 +1,18 @@
|
||||
package httpapi
|
||||
|
||||
import "testing"
|
||||
import (
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"lingma-ipc-proxy/internal/service"
|
||||
)
|
||||
|
||||
func TestNormalizeOpenAIRequestCollectsSystemMessages(t *testing.T) {
|
||||
req := openAIChatRequest{
|
||||
@@ -41,6 +53,34 @@ func TestNormalizeOpenAIRequestCollectsSystemMessages(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestCapabilitiesAdvertiseAgentCompatibility(t *testing.T) {
|
||||
server := NewServer("", service.New(service.Config{
|
||||
Model: "Qwen3-Coder",
|
||||
Timeout: time.Second,
|
||||
}))
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "/capabilities", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
server.http.Handler.ServeHTTP(rec, req)
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("status = %d body = %s", rec.Code, rec.Body.String())
|
||||
}
|
||||
|
||||
var body map[string]any
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
features, ok := body["features"].(map[string]any)
|
||||
if !ok {
|
||||
t.Fatalf("missing features: %#v", body)
|
||||
}
|
||||
for _, key := range []string{"tools", "tool_alias_mapping", "images", "local_image_paths", "image_auto_resize"} {
|
||||
if features[key] != true {
|
||||
t.Fatalf("feature %s = %#v", key, features[key])
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestNormalizeOpenAIRequestRejectsMissingUserAndAssistantMessages(t *testing.T) {
|
||||
req := openAIChatRequest{
|
||||
Model: "test-model",
|
||||
@@ -117,3 +157,75 @@ func TestNormalizeAnthropicRequestRejectsEmptyMessages(t *testing.T) {
|
||||
t.Fatal("expected error for request without usable messages")
|
||||
}
|
||||
}
|
||||
|
||||
func TestDiscoveryCompatibilityEndpoints(t *testing.T) {
|
||||
server := NewServer("", service.New(service.Config{
|
||||
Model: "Qwen3-Coder",
|
||||
Timeout: time.Second,
|
||||
}))
|
||||
|
||||
cases := []string{
|
||||
"/version",
|
||||
"/props",
|
||||
"/v1/props",
|
||||
}
|
||||
for _, path := range cases {
|
||||
req := httptest.NewRequest(http.MethodGet, path, nil)
|
||||
rec := httptest.NewRecorder()
|
||||
server.http.Handler.ServeHTTP(rec, req)
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("%s status = %d body = %s", path, rec.Code, rec.Body.String())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseImageURLReadsLocalFileURL(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
path := filepath.Join(dir, "sample.jpg")
|
||||
data := []byte{0xff, 0xd8, 0xff, 0xd9}
|
||||
if err := os.WriteFile(path, data, 0644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
img := parseImageURL("file://" + path)
|
||||
if img == nil {
|
||||
t.Fatal("expected image")
|
||||
}
|
||||
if img.MediaType != "image/jpeg" {
|
||||
t.Fatalf("media type = %q", img.MediaType)
|
||||
}
|
||||
if img.Data != base64.StdEncoding.EncodeToString(data) {
|
||||
t.Fatalf("data = %q", img.Data)
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseImageURLReadsAbsoluteLocalPath(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
path := filepath.Join(dir, "sample.png")
|
||||
data := []byte{0x89, 0x50, 0x4e, 0x47}
|
||||
if err := os.WriteFile(path, data, 0644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
img := parseImageURL(path)
|
||||
if img == nil {
|
||||
t.Fatal("expected image")
|
||||
}
|
||||
if img.MediaType != "image/png" {
|
||||
t.Fatalf("media type = %q", img.MediaType)
|
||||
}
|
||||
if img.Data != base64.StdEncoding.EncodeToString(data) {
|
||||
t.Fatalf("data = %q", img.Data)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSanitizeRecordedBodyRedactsImagePayloads(t *testing.T) {
|
||||
raw := []byte(`{"messages":[{"content":[{"type":"image_url","image_url":{"url":"data:image/png;base64,` + strings.Repeat("a", 8192) + `"}}]}]}`)
|
||||
got := sanitizeRecordedBody(raw)
|
||||
if strings.Contains(got, "data:image/png;base64") {
|
||||
t.Fatalf("image payload was not redacted: %s", got)
|
||||
}
|
||||
if !strings.Contains(got, "[image payload redacted") {
|
||||
t.Fatalf("missing redaction marker: %s", got)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user