Browse Source

test: stabilize channel affinity usage cache tests

master
fengsilin 3 months 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" "fmt"
"net/http/httptest" "net/http/httptest"
"testing" "testing"
"time"


"github.com/QuantumNous/new-api/dto" "github.com/QuantumNous/new-api/dto"
"github.com/QuantumNous/new-api/types" "github.com/QuantumNous/new-api/types"
@@ -26,9 +25,9 @@ func buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP string)
} }


func TestObserveChannelAffinityUsageCacheByRelayFormat_ClaudeMode(t *testing.T) { func TestObserveChannelAffinityUsageCacheByRelayFormat_ClaudeMode(t *testing.T) {
ruleName := fmt.Sprintf("rule_%d", time.Now().UnixNano())
ruleName := "rule_" + t.Name()
usingGroup := "default" usingGroup := "default"
keyFP := fmt.Sprintf("fp_%d", time.Now().UnixNano())
keyFP := "fp_" + t.Name()
ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP) ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP)


usage := &dto.Usage{ usage := &dto.Usage{
@@ -53,9 +52,9 @@ func TestObserveChannelAffinityUsageCacheByRelayFormat_ClaudeMode(t *testing.T)
} }


func TestObserveChannelAffinityUsageCacheByRelayFormat_MixedMode(t *testing.T) { func TestObserveChannelAffinityUsageCacheByRelayFormat_MixedMode(t *testing.T) {
ruleName := fmt.Sprintf("rule_%d", time.Now().UnixNano())
ruleName := "rule_" + t.Name()
usingGroup := "default" usingGroup := "default"
keyFP := fmt.Sprintf("fp_%d", time.Now().UnixNano())
keyFP := "fp_" + t.Name()
ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP) ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP)


openAIUsage := &dto.Usage{ openAIUsage := &dto.Usage{
@@ -83,9 +82,9 @@ func TestObserveChannelAffinityUsageCacheByRelayFormat_MixedMode(t *testing.T) {
} }


func TestObserveChannelAffinityUsageCacheByRelayFormat_UnsupportedModeKeepsEmpty(t *testing.T) { func TestObserveChannelAffinityUsageCacheByRelayFormat_UnsupportedModeKeepsEmpty(t *testing.T) {
ruleName := fmt.Sprintf("rule_%d", time.Now().UnixNano())
ruleName := "rule_" + t.Name()
usingGroup := "default" usingGroup := "default"
keyFP := fmt.Sprintf("fp_%d", time.Now().UnixNano())
keyFP := "fp_" + t.Name()
ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP) ctx := buildChannelAffinityStatsContextForTest(ruleName, usingGroup, keyFP)


usage := &dto.Usage{ usage := &dto.Usage{


Loading…
Cancel
Save