new-api/controller/channel_billing_helpers_test.go

package controller

import (
	"net/http"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestGetAuthHeadersUseProviderSpecificNames(t *testing.T) {
	openAI := GetAuthHeader("openai-key")
	claude := GetClaudeAuthHeader("claude-key")

	require.Equal(t, "Bearer openai-key", openAI.Get("Authorization"))
	require.Empty(t, openAI.Get("x-api-key"))
	require.Equal(t, "claude-key", claude.Get("x-api-key"))
	require.Equal(t, "2023-06-01", claude.Get("anthropic-version"))
	require.Empty(t, claude.Get("Authorization"))
}

func TestGetResponseBodyRejectsInvalidMethodURLBeforeNetwork(t *testing.T) {
	_, err := GetResponseBody("GET", "://invalid", nil, http.Header{})

	require.Error(t, err)
}