|
|
|
@@ -4,7 +4,6 @@ import ( |
|
|
|
"fmt" |
|
|
|
"net/http/httptest" |
|
|
|
"testing" |
|
|
|
"time" |
|
|
|
|
|
|
|
"github.com/QuantumNous/new-api/dto" |
|
|
|
"github.com/QuantumNous/new-api/types" |
|
|
|
@@ -26,9 +25,9 @@ func buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP string) |
|
|
|
} |
|
|
|
|
|
|
|
func TestObserveChannelAffinityUsageCacheByRelayFormat_ClaudeMode(t *testing.T) { |
|
|
|
ruleName := fmt.Sprintf("rule_%d", time.Now().UnixNano()) |
|
|
|
ruleName := "rule_" + t.Name() |
|
|
|
usingGroup := "default" |
|
|
|
keyFP := fmt.Sprintf("fp_%d", time.Now().UnixNano()) |
|
|
|
keyFP := "fp_" + t.Name() |
|
|
|
ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP) |
|
|
|
|
|
|
|
usage := &dto.Usage{ |
|
|
|
@@ -53,9 +52,9 @@ func TestObserveChannelAffinityUsageCacheByRelayFormat_ClaudeMode(t *testing.T) |
|
|
|
} |
|
|
|
|
|
|
|
func TestObserveChannelAffinityUsageCacheByRelayFormat_MixedMode(t *testing.T) { |
|
|
|
ruleName := fmt.Sprintf("rule_%d", time.Now().UnixNano()) |
|
|
|
ruleName := "rule_" + t.Name() |
|
|
|
usingGroup := "default" |
|
|
|
keyFP := fmt.Sprintf("fp_%d", time.Now().UnixNano()) |
|
|
|
keyFP := "fp_" + t.Name() |
|
|
|
ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP) |
|
|
|
|
|
|
|
openAIUsage := &dto.Usage{ |
|
|
|
@@ -83,9 +82,9 @@ func TestObserveChannelAffinityUsageCacheByRelayFormat_MixedMode(t *testing.T) { |
|
|
|
} |
|
|
|
|
|
|
|
func TestObserveChannelAffinityUsageCacheByRelayFormat_UnsupportedModeKeepsEmpty(t *testing.T) { |
|
|
|
ruleName := fmt.Sprintf("rule_%d", time.Now().UnixNano()) |
|
|
|
ruleName := "rule_" + t.Name() |
|
|
|
usingGroup := "default" |
|
|
|
keyFP := fmt.Sprintf("fp_%d", time.Now().UnixNano()) |
|
|
|
keyFP := "fp_" + t.Name() |
|
|
|
ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP) |
|
|
|
|
|
|
|
usage := &dto.Usage{ |
|
|
|
|