Luis Pater
53d1fd6c5c
- Introduced new xAI `grok-imagine-video` model for video generation with configurable options (e.g., duration, size, resolution). - Implemented video-specific API endpoints (`/v1/videos`, `/v1/videos/generations`, `/v1/videos/edits`, `/v1/videos/extensions`), including request validation and model handling. - Enhanced model registry with `xaiBuiltinVideoModelID` and metadata for video capabilities. - Added unit tests to validate video model support, request structures, and API response handling. - Extended `XAIExecutor` to integrate video generation and retrieval via runtime requests.
76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package logging
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func TestGinLogrusRecoveryRepanicsErrAbortHandler(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
engine := gin.New()
|
|
engine.Use(GinLogrusRecovery())
|
|
engine.GET("/abort", func(c *gin.Context) {
|
|
panic(http.ErrAbortHandler)
|
|
})
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/abort", nil)
|
|
recorder := httptest.NewRecorder()
|
|
|
|
defer func() {
|
|
recovered := recover()
|
|
if recovered == nil {
|
|
t.Fatalf("expected panic, got nil")
|
|
}
|
|
err, ok := recovered.(error)
|
|
if !ok {
|
|
t.Fatalf("expected error panic, got %T", recovered)
|
|
}
|
|
if !errors.Is(err, http.ErrAbortHandler) {
|
|
t.Fatalf("expected ErrAbortHandler, got %v", err)
|
|
}
|
|
if err != http.ErrAbortHandler {
|
|
t.Fatalf("expected exact ErrAbortHandler sentinel, got %v", err)
|
|
}
|
|
}()
|
|
|
|
engine.ServeHTTP(recorder, req)
|
|
}
|
|
|
|
func TestGinLogrusRecoveryHandlesRegularPanic(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
engine := gin.New()
|
|
engine.Use(GinLogrusRecovery())
|
|
engine.GET("/panic", func(c *gin.Context) {
|
|
panic("boom")
|
|
})
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/panic", nil)
|
|
recorder := httptest.NewRecorder()
|
|
|
|
engine.ServeHTTP(recorder, req)
|
|
if recorder.Code != http.StatusInternalServerError {
|
|
t.Fatalf("expected 500, got %d", recorder.Code)
|
|
}
|
|
}
|
|
|
|
func TestIsAIAPIPathIncludesImages(t *testing.T) {
|
|
if !isAIAPIPath("/v1/images/generations") {
|
|
t.Fatalf("expected /v1/images/generations to be treated as AI API path")
|
|
}
|
|
if !isAIAPIPath("/v1/images/edits") {
|
|
t.Fatalf("expected /v1/images/edits to be treated as AI API path")
|
|
}
|
|
if !isAIAPIPath("/v1/videos") {
|
|
t.Fatalf("expected /v1/videos to be treated as AI API path")
|
|
}
|
|
if !isAIAPIPath("/v1/videos/video_123") {
|
|
t.Fatalf("expected /v1/videos/video_123 to be treated as AI API path")
|
|
}
|
|
}
|