Browse Source

test: stabilize channel affinity usage cache tests

master
fengsilin 1 week ago
parent
commit
fae24fdf37
1 changed files with 6 additions and 7 deletions
  1. +6
    -7
      service/channel_affinity_usage_cache_test.go

+ 6
- 7
service/channel_affinity_usage_cache_test.go View File

@@ -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{


Loading…
Cancel
Save