TestDefaultCompressionService_Stats 测试统计功能
(t *testing.T)
| 131 | |
| 132 | // TestDefaultCompressionService_Stats 测试统计功能 |
| 133 | func TestDefaultCompressionService_Stats(t *testing.T) { |
| 134 | svc := NewDefaultCompressionService(nil, 100) |
| 135 | prompt := generateTestPrompt() |
| 136 | |
| 137 | // 执行几次压缩 |
| 138 | for range 3 { |
| 139 | _, err := svc.CompressSystemPrompt(context.Background(), prompt, &CompressOptions{ |
| 140 | Mode: ModeSimple, |
| 141 | Level: LevelModerate, |
| 142 | UseCache: false, // 禁用缓存以确保每次都执行 |
| 143 | }) |
| 144 | if err != nil { |
| 145 | t.Fatalf("Compress failed: %v", err) |
| 146 | } |
| 147 | } |
| 148 | |
| 149 | stats := svc.GetStats() |
| 150 | if stats.TotalCompressions != 3 { |
| 151 | t.Errorf("Expected 3 compressions, got %d", stats.TotalCompressions) |
| 152 | } |
| 153 | |
| 154 | // 重置统计 |
| 155 | svc.ResetStats() |
| 156 | stats = svc.GetStats() |
| 157 | if stats.TotalCompressions != 0 { |
| 158 | t.Errorf("Expected 0 compressions after reset, got %d", stats.TotalCompressions) |
| 159 | } |
| 160 | } |
| 161 | |
| 162 | // TestDefaultCompressionService_HybridMode 测试混合模式 |
| 163 | func TestDefaultCompressionService_HybridMode(t *testing.T) { |
nothing calls this directly
no test coverage detected