Function TestRecordStreamRetry

coderd/x/chatd/chatloop/metrics_test.go:281–323

(t *testing.T)

}

func TestRecordStreamRetry(t *testing.T) {
	t.Parallel()

	// One row per ChatErrorKind constant. Production callers always
	// reach RecordStreamRetry through chaterror.Classify, which
	// guarantees Kind is non-empty, so no empty-string case is
	// needed.
	tests := []struct {
		name        string
		kind        codersdk.ChatErrorKind
		chainBroken bool
	}{
		{name: "overloaded", kind: codersdk.ChatErrorKindOverloaded},
		{name: "rate_limit", kind: codersdk.ChatErrorKindRateLimit},
		{name: "timeout", kind: codersdk.ChatErrorKindTimeout},
		{name: "startup_timeout", kind: codersdk.ChatErrorKindStartupTimeout},
		{name: "auth", kind: codersdk.ChatErrorKindAuth},
		{name: "config", kind: codersdk.ChatErrorKindConfig},
		{name: "missing_key", kind: codersdk.ChatErrorKindMissingKey},
		{name: "generic", kind: codersdk.ChatErrorKindGeneric},
		{name: "chain_broken", kind: codersdk.ChatErrorKindGeneric, chainBroken: true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			reg := prometheus.NewRegistry()
			m := chatloop.NewMetrics(reg)
			m.RecordStreamRetry("test-provider", "test-model", chaterror.ClassifiedError{
				Kind:        tt.kind,
				ChainBroken: tt.chainBroken,
			})

			requireCounter(t, reg, "coderd_chatd_stream_retries_total", 1, map[string]string{
				"provider":     "test-provider",
				"model":        "test-model",
				"kind":         string(tt.kind),
				"chain_broken": strconv.FormatBool(tt.chainBroken),
			})
		})
	}
}

func TestRecordStreamBufferDropped(t *testing.T) {
	t.Parallel()