82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package deepseek
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNew(t *testing.T) {
|
|
client := New("key", "url")
|
|
assert.Equal(t, "key", client.apiKey)
|
|
assert.Equal(t, "url", client.baseURL)
|
|
assert.NotNil(t, client.httpClient)
|
|
}
|
|
|
|
func TestProcess_Success(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "/v1/chat/completions", r.URL.Path)
|
|
assert.Equal(t, "Bearer key", r.Header.Get("Authorization"))
|
|
|
|
resp := response{Choices: []choice{{Message: message{Content: "response"}}}}
|
|
json.NewEncoder(w).Encode(resp)
|
|
}))
|
|
defer server.Close()
|
|
|
|
client := New("key", server.URL)
|
|
result, err := client.Process(context.Background(), "prompt")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "response", result)
|
|
}
|
|
|
|
func TestProcess_HTTPError(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}))
|
|
defer server.Close()
|
|
|
|
client := New("key", server.URL)
|
|
_, err := client.Process(context.Background(), "prompt")
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "status 500")
|
|
}
|
|
|
|
func TestProcess_EmptyChoices(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
json.NewEncoder(w).Encode(response{Choices: []choice{}})
|
|
}))
|
|
defer server.Close()
|
|
|
|
client := New("key", server.URL)
|
|
_, err := client.Process(context.Background(), "prompt")
|
|
assert.Error(t, err)
|
|
assert.Contains(t, err.Error(), "no choices")
|
|
}
|
|
|
|
func TestProcess_InvalidJSON(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("not json"))
|
|
}))
|
|
defer server.Close()
|
|
|
|
client := New("key", server.URL)
|
|
_, err := client.Process(context.Background(), "prompt")
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestProcess_NetworkError(t *testing.T) {
|
|
client := New("key", "http://localhost:9999")
|
|
_, err := client.Process(context.Background(), "prompt")
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestProcess_InvalidURL(t *testing.T) {
|
|
client := New("key", string([]byte{0x7f}))
|
|
_, err := client.Process(context.Background(), "prompt")
|
|
assert.Error(t, err)
|
|
}
|