mirror of
https://github.com/mark3labs/kit.git
synced 2026-06-14 03:30:26 +00:00
Ed Zynda
ce32cea7ee
Migrate from github.com/charmbracelet/* v1 to charm.land/* v2 vanity imports. Key changes: - bubbletea: View() returns tea.View, KeyMsg -> KeyPressMsg, msg.String() matching - lipgloss: AdaptiveColor replaced with cached dark-bg detection helper - bubbles/textarea: Styles()/SetStyles() pattern, KeyMap.InsertNewline override - bubbles/progress: SetWidth(), WithDefaultBlend(), typed Update return - Input: enter always submits, ctrl+j/alt+enter insert newlines - User message newlines preserved through glamour via \n -> \n\n conversion - glamour stays at v1 (no v2 exists)
68 lines
2.2 KiB
Go
68 lines
2.2 KiB
Go
package ui
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mark3labs/mcphost/internal/models"
|
|
)
|
|
|
|
// stripAnsi removes ANSI escape codes from a string for test comparisons.
|
|
var ansiRegex = regexp.MustCompile(`\x1b\[[0-9;]*m`)
|
|
|
|
func stripAnsi(s string) string {
|
|
return ansiRegex.ReplaceAllString(s, "")
|
|
}
|
|
|
|
func TestUsageTracker_RenderUsageInfo_OAuth(t *testing.T) {
|
|
// Create a mock model info with costs and context limit
|
|
modelInfo := &models.ModelInfo{
|
|
ID: "claude-3-5-sonnet-20241022",
|
|
Name: "Claude 3.5 Sonnet v2",
|
|
Cost: models.Cost{
|
|
Input: 3.0,
|
|
Output: 15.0,
|
|
},
|
|
Limit: models.Limit{
|
|
Context: 200000,
|
|
Output: 8192,
|
|
},
|
|
}
|
|
|
|
// Test OAuth rendering (should show $0.00)
|
|
oauthTracker := NewUsageTracker(modelInfo, "anthropic", 80, true)
|
|
oauthTracker.UpdateUsage(1500, 500, 0, 0) // 2000 total tokens
|
|
|
|
rendered := stripAnsi(oauthTracker.RenderUsageInfo())
|
|
|
|
// Should show tokens and percentage, but cost should show "$0.00"
|
|
if !strings.Contains(rendered, "Tokens: 2.0K") {
|
|
t.Errorf("Expected rendered output to contain 'Tokens: 2.0K', got: %s", rendered)
|
|
}
|
|
if !strings.Contains(rendered, "(1%)") { // 2000/200000 = 1%
|
|
t.Errorf("Expected rendered output to contain percentage, got: %s", rendered)
|
|
}
|
|
if !strings.Contains(rendered, "Cost: $0.00") {
|
|
t.Errorf("Expected rendered output to contain 'Cost: $0.00', got: %s", rendered)
|
|
}
|
|
|
|
// Test regular API key rendering (should show actual cost)
|
|
regularTracker := NewUsageTracker(modelInfo, "anthropic", 80, false)
|
|
regularTracker.UpdateUsage(1500, 500, 0, 0) // Same token usage
|
|
|
|
regularRendered := stripAnsi(regularTracker.RenderUsageInfo())
|
|
|
|
// Should show tokens and actual cost
|
|
if !strings.Contains(regularRendered, "Tokens: 2.0K") {
|
|
t.Errorf("Expected regular rendered output to contain 'Tokens: 2.0K', got: %s", regularRendered)
|
|
}
|
|
if strings.Contains(regularRendered, "Cost: $0.00") {
|
|
t.Errorf("Expected regular rendered output to NOT show $0.00, got: %s", regularRendered)
|
|
}
|
|
// Should show actual calculated cost (1500*3 + 500*15)/1000000 = 0.0120
|
|
if !strings.Contains(regularRendered, "Cost: $0.0120") { // Now showing 4 decimal places
|
|
t.Errorf("Expected regular rendered output to show actual cost, got: %s", regularRendered)
|
|
}
|
|
}
|