Function TestChatsTelemetry

coderd/telemetry/telemetry_test.go:1554–1890

(t *testing.T)

}

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

	ctx := testutil.Context(t, testutil.WaitMedium)
	db, _ := dbtestutil.NewDB(t)

	user := dbgen.User(t, db, database.User{})

	// Create chat providers (required FK for model configs).
	_ = dbgen.ChatProvider(t, db, database.ChatProvider{
		Provider:    "anthropic",
		DisplayName: "Anthropic",
	})
	_ = dbgen.ChatProvider(t, db, database.ChatProvider{
		Provider:    "openai",
		DisplayName: "OpenAI",
	})

	// Create a model config.
	modelCfg := dbgen.ChatModelConfig(t, db, database.ChatModelConfig{
		Provider:     "anthropic",
		Model:        "claude-sonnet-4-20250514",
		DisplayName:  "Claude Sonnet",
		IsDefault:    true,
		ContextLimit: 200000,
	})

	// Create a second model config to test full dump.
	modelCfg2 := dbgen.ChatModelConfig(t, db, database.ChatModelConfig{
		Provider:    "openai",
		Model:       "gpt-4o",
		DisplayName: "GPT-4o",
	})

	// Create a soft-deleted model config — should NOT appear in telemetry.
	deletedCfg := dbgen.ChatModelConfig(t, db, database.ChatModelConfig{
		Provider:     "anthropic",
		Model:        "claude-deleted",
		DisplayName:  "Deleted Model",
		ContextLimit: 100000,
	})
	err := db.DeleteChatModelConfigByID(ctx, deletedCfg.ID)
	require.NoError(t, err)

	// Create a root chat with a workspace.
	org, err := db.GetDefaultOrganization(ctx)
	require.NoError(t, err)
	job := dbgen.ProvisionerJob(t, db, nil, database.ProvisionerJob{
		OrganizationID: org.ID,
		Type:           database.ProvisionerJobTypeTemplateVersionDryRun,
	})
	tpl := dbgen.Template(t, db, database.Template{
		OrganizationID: org.ID,
		CreatedBy:      user.ID,
	})
	tv := dbgen.TemplateVersion(t, db, database.TemplateVersion{
		OrganizationID: org.ID,
		TemplateID:     uuid.NullUUID{UUID: tpl.ID, Valid: true},