mirror of
https://github.com/Twigpine/zero.git
synced 2026-11-10 05:01:28 -10:00
PierrunoYTandClaude Opus 5.5
13e2596b2e
Update the Go module path, imports, install scripts, docs, workflows, packaging and action defaults from Gitlawb/zero to Twigpine/zero. The npm scope (already @twigpine/zero on main, with its @gitlawb/zero migration notes), LICENSE, action author, the OpenGateway provider and historical CHANGELOG links are unchanged. The X-AIMLAPI-Integration-Repo header keeps Gitlawb/zero: it is what AIMLAPI sees for attribution, so changing it needs a maintainer's yes. docs/INSTALL.md keeps `go install github.com/Gitlawb/zero/...`, because every tagged release so far declares that module path. The Homebrew formula comment names the Twigpine/homebrew-tap tap. Co-authored-by: Claude Opus 5.5 <[email protected]>
90 lines
2.8 KiB
Go
90 lines
2.8 KiB
Go
package usage
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Twigpine/zero/internal/zeroruntime"
|
|
)
|
|
|
|
func TestTrackerNormalizesUsageAndComputesModelCost(t *testing.T) {
|
|
tracker := NewTracker(TrackerOptions{Now: fixedUsageClock("2026-06-04T13:00:00Z")})
|
|
|
|
record, err := tracker.Record(RecordInput{
|
|
ModelID: "gpt-4.1",
|
|
Source: "exec",
|
|
Usage: zeroruntime.Usage{
|
|
PromptTokens: 1_000,
|
|
CompletionTokens: 250,
|
|
CachedInputTokens: 200,
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Record returned error: %v", err)
|
|
}
|
|
if record.Sequence != 1 || record.ID != "zero_usage_1" || record.CreatedAt != "2026-06-04T13:00:00Z" {
|
|
t.Fatalf("unexpected record identity: %#v", record)
|
|
}
|
|
if record.Usage.InputTokens != 1_000 || record.Usage.OutputTokens != 250 || record.Usage.TotalTokens != 1_250 {
|
|
t.Fatalf("usage not normalized: %#v", record.Usage)
|
|
}
|
|
if record.Cost.TotalCost <= 0 || record.Cost.ModelID != "gpt-4.1" {
|
|
t.Fatalf("cost not computed: %#v", record.Cost)
|
|
}
|
|
|
|
summary := tracker.Summary()
|
|
if summary.RecordCount != 1 || summary.TotalTokens != 1_250 || summary.ByModel[0].ModelID != "gpt-4.1" {
|
|
t.Fatalf("unexpected summary: %#v", summary)
|
|
}
|
|
if FormatSummary(summary) != "1 request, 1,250 tokens, "+summary.FormattedTotalCost {
|
|
t.Fatalf("unexpected formatted summary: %q", FormatSummary(summary))
|
|
}
|
|
}
|
|
|
|
func TestTrackerRejectsInvalidUsageAndUnknownModels(t *testing.T) {
|
|
tracker := NewTracker(TrackerOptions{})
|
|
if _, err := tracker.Record(RecordInput{ModelID: "missing", Usage: zeroruntime.Usage{InputTokens: 1}}); err == nil {
|
|
t.Fatal("expected unknown model error")
|
|
}
|
|
if _, err := tracker.Record(RecordInput{ModelID: "gpt-4.1", Usage: zeroruntime.Usage{InputTokens: -1}}); err == nil {
|
|
t.Fatal("expected invalid usage error")
|
|
}
|
|
}
|
|
|
|
func TestTrackerTreatsReasoningAsOutputBreakdown(t *testing.T) {
|
|
tracker := NewTracker(TrackerOptions{})
|
|
record, err := tracker.Record(RecordInput{
|
|
ModelID: "gpt-4.1",
|
|
Usage: zeroruntime.Usage{
|
|
InputTokens: 100,
|
|
OutputTokens: 40,
|
|
ReasoningTokens: 10,
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Record returned error: %v", err)
|
|
}
|
|
if record.Usage.TotalTokens != 140 {
|
|
t.Fatalf("total tokens = %d, want 140", record.Usage.TotalTokens)
|
|
}
|
|
}
|
|
|
|
func TestTrackerResetClearsRecords(t *testing.T) {
|
|
tracker := NewTracker(TrackerOptions{})
|
|
if _, err := tracker.Record(RecordInput{ModelID: "gpt-4.1", Usage: zeroruntime.Usage{InputTokens: 1, OutputTokens: 1}}); err != nil {
|
|
t.Fatalf("Record returned error: %v", err)
|
|
}
|
|
tracker.Reset()
|
|
if summary := tracker.Summary(); summary.RecordCount != 0 || len(summary.ByModel) != 0 {
|
|
t.Fatalf("Reset did not clear tracker: %#v", summary)
|
|
}
|
|
}
|
|
|
|
func fixedUsageClock(value string) func() time.Time {
|
|
parsed, err := time.Parse(time.RFC3339, value)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return func() time.Time { return parsed }
|
|
}
|