Function TestGetChatMessagesForPromptByChatID

coderd/database/querier_test.go:11093–11308

(t *testing.T)

}

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

	// This test exercises a complex CTE query for prompt
	// reconstruction after compaction. It requires Postgres.
	db, _ := dbtestutil.NewDB(t)
	ctx := context.Background()

	// Helper: create a chat model config (required FK for chats).
	user := dbgen.User(t, db, database.User{})
	org := dbgen.Organization(t, db, database.Organization{})
	dbgen.OrganizationMember(t, db, database.OrganizationMember{UserID: user.ID, OrganizationID: org.ID})

	// An AI provider row is required as a FK for model configs.
	provider := dbgen.AIProvider(t, db, database.AIProvider{
		Type:        database.AiProviderTypeOpenai,
		Name:        "test-" + uuid.NewString(),
		DisplayName: sql.NullString{String: "OpenAI", Valid: true},
		Enabled:     true,
	})
	dbgen.AIProviderKey(t, db, database.AIProviderKey{
		ProviderID: provider.ID,
		APIKey:     "test-key",
	})

	modelCfg, err := insertChatModelConfigForTest(ctx, t, db, database.InsertChatModelConfigParams{
		Provider:             "openai",
		AIProviderID:         uuid.NullUUID{UUID: provider.ID, Valid: true},
		Model:                "test-model",
		DisplayName:          "Test 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)

	newChat := func(t *testing.T) database.Chat {
		t.Helper()
		chat, err := db.InsertChat(ctx, database.InsertChatParams{
			OrganizationID:    org.ID,
			Status:            database.ChatStatusWaiting,
			ClientType:        database.ChatClientTypeUi,
			OwnerID:           user.ID,
			LastModelConfigID: modelCfg.ID,
			Title:             "test-chat-" + uuid.NewString(),
		})
		require.NoError(t, err)
		return chat
	}

	insertMsg := func(
		t *testing.T,
		chatID uuid.UUID,
		role database.ChatMessageRole,