Function TestExtractJSONTokenUsage

pkg/workflow/metrics_test.go:121–263

(t *testing.T)

}

func TestExtractJSONTokenUsage(t *testing.T) {
	tests := []struct {
		name     string
		data     map[string]any
		expected int
	}{
		{
			name: "Direct tokens field",
			data: map[string]any{
				"tokens": 500,
			},
			expected: 500,
		},
		{
			name: "Token count field",
			data: map[string]any{
				"token_count": 300,
			},
			expected: 300,
		},
		{
			name: "Usage object with input/output tokens",
			data: map[string]any{
				"usage": map[string]any{
					"input_tokens":  100,
					"output_tokens": 50,
				},
			},
			expected: 150,
		},
		{
			name: "Usage object with cache tokens",
			data: map[string]any{
				"usage": map[string]any{
					"input_tokens":                100,
					"output_tokens":               50,
					"cache_creation_input_tokens": 200,
					"cache_read_input_tokens":     75,
				},
			},
			expected: 425,
		},
		{
			name: "Delta format",
			data: map[string]any{
				"delta": map[string]any{
					"usage": map[string]any{
						"input_tokens":  25,
						"output_tokens": 35,
					},
				},
			},
			expected: 60,
		},
		{
			name: "String token count",
			data: map[string]any{
				"tokens": "750",