37 lines
920 B
Go
37 lines
920 B
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNormalizeKnownOpenAICodexModel_BareGPT56RoutesToSol(t *testing.T) {
|
||
|
|
tests := map[string]string{
|
||
|
|
"gpt-5.6": "gpt-5.6-sol",
|
||
|
|
"openai/gpt-5.6": "gpt-5.6-sol",
|
||
|
|
"gpt5.6": "gpt-5.6-sol",
|
||
|
|
"gpt-5.6-high": "gpt-5.6-sol",
|
||
|
|
"gpt-5.6-max": "gpt-5.6-sol",
|
||
|
|
"gpt-5.6-2026-07-09": "gpt-5.6-sol",
|
||
|
|
"openai/gpt-5.6-max": "gpt-5.6-sol",
|
||
|
|
}
|
||
|
|
|
||
|
|
for input, expected := range tests {
|
||
|
|
t.Run(input, func(t *testing.T) {
|
||
|
|
require.Equal(t, expected, normalizeKnownOpenAICodexModel(input))
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUsageBillingModelCandidates_BareGPT56IncludesSol(t *testing.T) {
|
||
|
|
require.Equal(t,
|
||
|
|
[]string{"gpt-5.6", "gpt-5.6-sol"},
|
||
|
|
usageBillingModelCandidates("gpt-5.6"),
|
||
|
|
)
|
||
|
|
require.Equal(t,
|
||
|
|
[]string{"openai/gpt-5.6", "gpt-5.6", "gpt-5.6-sol"},
|
||
|
|
usageBillingModelCandidates("openai/gpt-5.6"),
|
||
|
|
)
|
||
|
|
}
|