Function TestChatDebugSQLGuards

coderd/database/querier_test.go:13595–13722

(t *testing.T)

}

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

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

	org := dbgen.Organization(t, store, database.Organization{})
	user := dbgen.User(t, store, database.User{})

	providerName := "openai"
	modelName := "debug-model-guards-" + uuid.NewString()

	dbgen.ChatProvider(t, store, database.ChatProvider{
		Provider:             providerName,
		DisplayName:          "Debug Provider",
		APIKey:               "test-key",
		Enabled:              true,
		CentralApiKeyEnabled: true,
	})

	modelCfg, err := insertChatModelConfigForTest(ctx, t, store, database.InsertChatModelConfigParams{
		Provider:             providerName,
		Model:                modelName,
		DisplayName:          "Debug Model",
		CreatedBy:            uuid.NullUUID{UUID: user.ID, Valid: true},
		UpdatedBy:            uuid.NullUUID{UUID: user.ID, Valid: true},
		Enabled:              true,
		IsDefault:            true,
		ContextLimit:         128000,
		CompressionThreshold: 80,
		Options:              json.RawMessage(`{}`),
	})
	require.NoError(t, err)

	chatA, err := store.InsertChat(ctx, database.InsertChatParams{
		OrganizationID:    org.ID,
		Status:            database.ChatStatusWaiting,
		ClientType:        database.ChatClientTypeUi,
		OwnerID:           user.ID,
		LastModelConfigID: modelCfg.ID,
		Title:             "chat-guard-A-" + uuid.NewString(),
	})
	require.NoError(t, err)

	chatB, err := store.InsertChat(ctx, database.InsertChatParams{
		OrganizationID:    org.ID,
		Status:            database.ChatStatusWaiting,
		ClientType:        database.ChatClientTypeUi,
		OwnerID:           user.ID,
		LastModelConfigID: modelCfg.ID,
		Title:             "chat-guard-B-" + uuid.NewString(),
	})
	require.NoError(t, err)

	runA, err := store.InsertChatDebugRun(ctx, database.InsertChatDebugRunParams{
		ChatID:              chatA.ID,
		ModelConfigID:       uuid.NullUUID{UUID: modelCfg.ID, Valid: true},
		TriggerMessageID:    sql.NullInt64{Int64: 1, Valid: true},