openai_responses_input_tokens_test.go 581 B

123456789101112131415161718192021
  1. package handler
  2. import (
  3. "net/http"
  4. "net/http/httptest"
  5. "testing"
  6. "github.com/gin-gonic/gin"
  7. "github.com/stretchr/testify/require"
  8. )
  9. func TestResponsesInputTokensIsClassifiedAsTokenCountRequest(t *testing.T) {
  10. gin.SetMode(gin.TestMode)
  11. recorder := httptest.NewRecorder()
  12. c, _ := gin.CreateTestContext(recorder)
  13. c.Request = httptest.NewRequest(http.MethodPost, "/v1/responses/input_tokens", nil)
  14. require.True(t, isCountTokensRequest(c))
  15. require.True(t, isTokenCountRequestPath("/responses/input_tokens"))
  16. require.False(t, isTokenCountRequestPath("/v1/responses"))
  17. }