mirror of
https://github.com/hanzoai/go-openai-realtime.git
synced 2026-11-03 00:00:22 +00:00
zeekayandHanzo Dev
601155d4d2
Hard-fork of WqyJh/go-openai-realtime (v0.5.x line, commit 3f3cf7a) into the
hanzoai org so the OpenAI/audio-LLM Go stack is fully owned. Renames the module
path across root, contrib/ws-gorilla, and examples, and repoints go-openai from
sashabaranov to github.com/hanzoai/go-openai v1.41.0 (required directly, no
replace directives anywhere).
Co-authored-by: Hanzo Dev <[email protected]>
128 lines
3.5 KiB
Go
128 lines
3.5 KiB
Go
package gorilla_test
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
openairt "github.com/hanzoai/go-openai-realtime"
|
|
gorilla "github.com/hanzoai/go-openai-realtime/contrib/ws-gorilla"
|
|
test "github.com/hanzoai/go-openai-realtime/test"
|
|
"github.com/gorilla/websocket"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWebSocket(t *testing.T) {
|
|
s := test.NewServer(t, time.Millisecond)
|
|
defer s.Server.Close()
|
|
|
|
dialer := gorilla.NewWebSocketDialer(gorilla.WebSocketOptions{})
|
|
|
|
conn, err := dialer.Dial(context.Background(), s.URL, nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, conn)
|
|
defer func() {
|
|
err = conn.Close()
|
|
require.NoError(t, err)
|
|
}()
|
|
|
|
err = conn.WriteMessage(context.Background(), openairt.MessageBinary+1, []byte("hello"))
|
|
require.ErrorIs(t, err, openairt.ErrUnsupportedMessageType)
|
|
|
|
err = conn.WriteMessage(context.Background(), openairt.MessageBinary, []byte("hello"))
|
|
require.NoError(t, err)
|
|
|
|
msgType, data, err := conn.ReadMessage(context.Background())
|
|
require.NoError(t, err)
|
|
require.Equal(t, openairt.MessageBinary, msgType)
|
|
require.Equal(t, []byte("hello"), data)
|
|
|
|
err = conn.WriteMessage(context.Background(), openairt.MessageText, []byte("world"))
|
|
require.NoError(t, err)
|
|
|
|
msgType, data, err = conn.ReadMessage(context.Background())
|
|
require.NoError(t, err)
|
|
require.Equal(t, openairt.MessageText, msgType)
|
|
require.Equal(t, []byte("world"), data)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Microsecond*5)
|
|
defer cancel()
|
|
|
|
_, _, err = conn.ReadMessage(ctx)
|
|
var permanent *openairt.PermanentError
|
|
require.ErrorAs(t, err, &permanent)
|
|
require.ErrorContains(t, permanent.Err, "i/o timeout")
|
|
}
|
|
|
|
func TestWebSocketReadLimitError(t *testing.T) {
|
|
s := test.NewServer(t, time.Millisecond)
|
|
defer s.Server.Close()
|
|
|
|
dialer := gorilla.NewWebSocketDialer(gorilla.WebSocketOptions{
|
|
ReadLimit: 5,
|
|
})
|
|
|
|
conn, err := dialer.Dial(context.Background(), s.URL, nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, conn)
|
|
|
|
err = conn.WriteMessage(context.Background(), openairt.MessageBinary, []byte("hello world"))
|
|
require.NoError(t, err)
|
|
|
|
_, _, err = conn.ReadMessage(context.Background())
|
|
require.Error(t, err)
|
|
var permanent *openairt.PermanentError
|
|
require.ErrorAs(t, err, &permanent)
|
|
t.Logf("error: %v", permanent.Err)
|
|
require.ErrorIs(t, permanent.Err, websocket.ErrReadLimit)
|
|
|
|
err = conn.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestWebSocketReadLimitOK(t *testing.T) {
|
|
s := test.NewServer(t, time.Millisecond)
|
|
defer s.Server.Close()
|
|
|
|
dialer := gorilla.NewWebSocketDialer(gorilla.WebSocketOptions{
|
|
ReadLimit: 11,
|
|
})
|
|
|
|
conn, err := dialer.Dial(context.Background(), s.URL, nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, conn)
|
|
defer func() {
|
|
err = conn.Close()
|
|
require.NoError(t, err)
|
|
}()
|
|
|
|
err = conn.WriteMessage(context.Background(), openairt.MessageBinary, []byte("hello world"))
|
|
require.NoError(t, err)
|
|
|
|
msgType, data, err := conn.ReadMessage(context.Background())
|
|
require.NoError(t, err)
|
|
require.Equal(t, openairt.MessageBinary, msgType)
|
|
require.Equal(t, []byte("hello world"), data)
|
|
}
|
|
|
|
func TestWebSocketDialOptions(t *testing.T) {
|
|
s := test.NewServer(t, time.Millisecond)
|
|
defer s.Server.Close()
|
|
|
|
dialer := gorilla.NewWebSocketDialer(gorilla.WebSocketOptions{
|
|
Dialer: &websocket.Dialer{},
|
|
})
|
|
|
|
conn, err := dialer.Dial(context.Background(), s.URL, http.Header{
|
|
"X-Test": {"test"},
|
|
"X-Test2": {"test2", "test3"},
|
|
})
|
|
require.NoError(t, err)
|
|
require.NotNil(t, conn)
|
|
|
|
header := conn.Response().Header
|
|
require.Equal(t, "test", header.Get("X-Test"))
|
|
require.Equal(t, []string{"test2", "test3"}, header["X-Test2"])
|
|
}
|