43 lines · 1.6 KB · Go
RawDownload
package executor import "testing" func TestParseOpenAIUsageChatCompletions(t *testing.T) {	data := []byte(`{"usage":{"prompt_tokens":1,"completion_tokens":2,"total_tokens":3,"prompt_tokens_details":{"cached_tokens":4},"completion_tokens_details":{"reasoning_tokens":5}}}`)	detail := parseOpenAIUsage(data)	if detail.InputTokens != 1 {		t.Fatalf("input tokens = %d, want %d", detail.InputTokens, 1)	}	if detail.OutputTokens != 2 {		t.Fatalf("output tokens = %d, want %d", detail.OutputTokens, 2)	}	if detail.TotalTokens != 3 {		t.Fatalf("total tokens = %d, want %d", detail.TotalTokens, 3)	}	if detail.CachedTokens != 4 {		t.Fatalf("cached tokens = %d, want %d", detail.CachedTokens, 4)	}	if detail.ReasoningTokens != 5 {		t.Fatalf("reasoning tokens = %d, want %d", detail.ReasoningTokens, 5)	}} func TestParseOpenAIUsageResponses(t *testing.T) {	data := []byte(`{"usage":{"input_tokens":10,"output_tokens":20,"total_tokens":30,"input_tokens_details":{"cached_tokens":7},"output_tokens_details":{"reasoning_tokens":9}}}`)	detail := parseOpenAIUsage(data)	if detail.InputTokens != 10 {		t.Fatalf("input tokens = %d, want %d", detail.InputTokens, 10)	}	if detail.OutputTokens != 20 {		t.Fatalf("output tokens = %d, want %d", detail.OutputTokens, 20)	}	if detail.TotalTokens != 30 {		t.Fatalf("total tokens = %d, want %d", detail.TotalTokens, 30)	}	if detail.CachedTokens != 7 {		t.Fatalf("cached tokens = %d, want %d", detail.CachedTokens, 7)	}	if detail.ReasoningTokens != 9 {		t.Fatalf("reasoning tokens = %d, want %d", detail.ReasoningTokens, 9)	}}