mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-14 12:02:27 +08:00
22 lines
581 B
Go
22 lines
581 B
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestResponsesInputTokensIsClassifiedAsTokenCountRequest(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
recorder := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(recorder)
|
|
c.Request = httptest.NewRequest(http.MethodPost, "/v1/responses/input_tokens", nil)
|
|
|
|
require.True(t, isCountTokensRequest(c))
|
|
require.True(t, isTokenCountRequestPath("/responses/input_tokens"))
|
|
require.False(t, isTokenCountRequestPath("/v1/responses"))
|
|
}
|