mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-09-28 11:50:20 +08:00
c8f723e0fb
- Add `BaseURL` field to usage records and host auth file entries. - Extract `base_url` from auth attributes and metadata during usage reporting. - Propagate `base_url` through plugin usage adapters and runtime auth callbacks. Closes: #5693
88 lines
2.3 KiB
Go
88 lines
2.3 KiB
Go
package usage
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestStreamFromContextDefaultsMissingToFalse(t *testing.T) {
|
|
if StreamFromContext(context.Background()) {
|
|
t.Fatalf("StreamFromContext(background) = true, want false")
|
|
}
|
|
}
|
|
|
|
func TestStreamFromContextHonorsExplicitTrue(t *testing.T) {
|
|
ctx := WithStream(context.Background(), true)
|
|
if !StreamFromContext(ctx) {
|
|
t.Fatalf("StreamFromContext(true) = false, want true")
|
|
}
|
|
}
|
|
|
|
func TestRecordStreamField(t *testing.T) {
|
|
record := Record{
|
|
Provider: "openai",
|
|
Model: "gpt-5.4",
|
|
Stream: true,
|
|
}
|
|
if !record.Stream {
|
|
t.Fatalf("Record.Stream = false, want true")
|
|
}
|
|
}
|
|
|
|
func TestRecordBaseURLField(t *testing.T) {
|
|
record := Record{
|
|
Provider: "openai",
|
|
Model: "gpt-5.4",
|
|
BaseURL: "https://custom-gateway.example.com/v1",
|
|
}
|
|
if record.BaseURL != "https://custom-gateway.example.com/v1" {
|
|
t.Fatalf("Record.BaseURL = %q, want %q", record.BaseURL, "https://custom-gateway.example.com/v1")
|
|
}
|
|
}
|
|
|
|
func TestGenerateEnabledDefaultsNilToTrue(t *testing.T) {
|
|
if !GenerateEnabled(nil) {
|
|
t.Fatalf("GenerateEnabled(nil) = false, want true")
|
|
}
|
|
}
|
|
|
|
func TestGenerateEnabledHonorsExplicitFalse(t *testing.T) {
|
|
if GenerateEnabled(GenerateFlag(false)) {
|
|
t.Fatalf("GenerateEnabled(false) = true, want false")
|
|
}
|
|
}
|
|
|
|
func TestGenerateEnabledHonorsExplicitTrue(t *testing.T) {
|
|
if !GenerateEnabled(GenerateFlag(true)) {
|
|
t.Fatalf("GenerateEnabled(true) = false, want true")
|
|
}
|
|
}
|
|
|
|
func TestGenerateFromContextDefaultsMissingToTrue(t *testing.T) {
|
|
if !GenerateFromContext(context.Background()) {
|
|
t.Fatalf("GenerateFromContext(background) = false, want true")
|
|
}
|
|
}
|
|
|
|
func TestGenerateFromContextHonorsExplicitFalse(t *testing.T) {
|
|
ctx := WithGenerate(context.Background(), false)
|
|
if GenerateFromContext(ctx) {
|
|
t.Fatalf("GenerateFromContext(false) = true, want false")
|
|
}
|
|
}
|
|
|
|
func TestRecordOmittedGenerateIsEnabled(t *testing.T) {
|
|
// Existing callers construct Record without setting Generate.
|
|
// Omission must remain distinguishable from explicit false and default to true.
|
|
record := Record{
|
|
Provider: "openai",
|
|
Model: "gpt-5.4",
|
|
}
|
|
if record.Generate != nil {
|
|
t.Fatalf("Record.Generate = %v, want nil for omitted field", record.Generate)
|
|
}
|
|
if !GenerateEnabled(record.Generate) {
|
|
t.Fatalf("GenerateEnabled(omitted) = false, want true")
|
|
}
|
|
}
|