102 lines
3.2 KiB
Go
102 lines
3.2 KiB
Go
package runner
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/easyai/easyai-ai-gateway/apps/api/internal/clients"
|
|
"github.com/easyai/easyai-ai-gateway/apps/api/internal/config"
|
|
"github.com/easyai/easyai-ai-gateway/apps/api/internal/store"
|
|
)
|
|
|
|
func TestHydrateProviderRequestAssetsConvertsStrictBase64Field(t *testing.T) {
|
|
storageDir := t.TempDir()
|
|
fileName := "gateway-request-asset-test.png"
|
|
if err := os.WriteFile(filepath.Join(storageDir, fileName), []byte("image bytes"), 0o644); err != nil {
|
|
t.Fatalf("write request asset: %v", err)
|
|
}
|
|
service := &Service{cfg: config.Config{LocalUploadedStorageDir: storageDir}}
|
|
body := map[string]any{
|
|
"model": "demo",
|
|
"b64_json": map[string]any{
|
|
"assetRef": map[string]any{
|
|
"sha256": "sha-test",
|
|
"contentType": "image/png",
|
|
"url": "/static/uploaded/" + fileName,
|
|
"storageProvider": "local_static",
|
|
},
|
|
"url": "/static/uploaded/" + fileName,
|
|
},
|
|
}
|
|
|
|
hydrated, err := service.hydrateProviderRequestAssets(context.Background(), body)
|
|
if err != nil {
|
|
t.Fatalf("hydrate request assets: %v", err)
|
|
}
|
|
if got, want := stringFromAny(hydrated["b64_json"]), base64.StdEncoding.EncodeToString([]byte("image bytes")); got != want {
|
|
t.Fatalf("unexpected hydrated base64: got %q want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestHydrateProviderRequestAssetsReturnsExpiredError(t *testing.T) {
|
|
expiredAt := time.Now().Add(-time.Minute).UTC().Format(time.RFC3339)
|
|
service := &Service{}
|
|
body := map[string]any{
|
|
"image_url": map[string]any{
|
|
"assetRef": map[string]any{
|
|
"sha256": "sha-expired",
|
|
"contentType": "image/png",
|
|
"url": "/static/uploaded/gateway-request-asset-expired.png",
|
|
"storageProvider": "local_static",
|
|
"expiresAt": expiredAt,
|
|
},
|
|
"url": "/static/uploaded/gateway-request-asset-expired.png",
|
|
},
|
|
}
|
|
|
|
_, err := service.hydrateProviderRequestAssets(context.Background(), body)
|
|
if err == nil {
|
|
t.Fatal("expected expired request asset error")
|
|
}
|
|
var clientErr *clients.ClientError
|
|
if !errors.As(err, &clientErr) || clientErr.Code != "request_asset_expired" {
|
|
t.Fatalf("expected request_asset_expired, got %T %v", err, err)
|
|
}
|
|
}
|
|
|
|
func TestStringFromAnyReadsRequestAssetWrapperURL(t *testing.T) {
|
|
wrapper := map[string]any{
|
|
"assetRef": map[string]any{"url": "https://cdn.example/request.png"},
|
|
}
|
|
if got := stringFromAny(wrapper); got != "https://cdn.example/request.png" {
|
|
t.Fatalf("expected wrapper URL, got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestSlimTaskRequestSnapshotKeepsMessageRefs(t *testing.T) {
|
|
service := &Service{}
|
|
task := store.GatewayTask{Request: map[string]any{
|
|
"conversationId": "conv-1",
|
|
"messageRefs": []any{map[string]any{"messageId": "msg-1", "position": 0}},
|
|
"newMessageCount": 1,
|
|
}}
|
|
body := map[string]any{
|
|
"model": "demo",
|
|
"messages": []any{map[string]any{"role": "user", "content": "hello"}},
|
|
}
|
|
|
|
snapshot := service.slimTaskRequestSnapshot(task, body)
|
|
|
|
if snapshot["messages"] != nil {
|
|
t.Fatalf("snapshot should not persist restored messages: %+v", snapshot)
|
|
}
|
|
if snapshot["messageRefs"] == nil || snapshot["newMessageCount"] != 1 {
|
|
t.Fatalf("snapshot should keep message refs and new count: %+v", snapshot)
|
|
}
|
|
}
|