From fdf2212c88fb6b89c8a7025cb4efd5fa05b8bfca Mon Sep 17 00:00:00 2001
From: Lawrence Chen <54008264+lawrencecchen@users.noreply.github.com>
Date: Fri, 6 Mar 2026 04:28:18 -0800
Subject: [PATCH] Add regression test for command preview corpus

---
 Sources/ContentView.swift                     | 19 ++++++++++
 .../CommandPaletteSearchEngineTests.swift     | 36 +++++++++++++++++++
 2 files changed, 55 insertions(+)

diff --git a/Sources/ContentView.swift b/Sources/ContentView.swift
index 5db8359a..9701ca2e 100644
--- a/Sources/ContentView.swift
+++ b/Sources/ContentView.swift
@@ -3434,6 +3434,25 @@ struct ContentView: View {
         return Array(matches.prefix(resultLimit))
     }
 
+    nonisolated static func commandPaletteCommandPreviewMatchCommandIDsForTests(
+        searchCorpus: [CommandPaletteSearchCorpusEntry<String>],
+        candidateCommandIDs: [String],
+        searchCorpusByID: [String: CommandPaletteSearchCorpusEntry<String>],
+        query: String,
+        resultLimit: Int
+    ) -> [String] {
+        let preparedQuery = CommandPaletteFuzzyMatcher.preparedQuery(query)
+        return commandPalettePreviewSearchMatches(
+            candidateCommandIDs: candidateCommandIDs,
+            searchCorpusByID: searchCorpusByID,
+            query: query,
+            usageHistory: [:],
+            queryIsEmpty: preparedQuery.isEmpty,
+            historyTimestamp: 0,
+            resultLimit: resultLimit
+        ).map(\.commandID)
+    }
+
     private func scheduleCommandPaletteResultsRefresh(forceSearchCorpusRefresh: Bool = false) {
         refreshCommandPaletteSearchCorpus(force: forceSearchCorpusRefresh)
 
diff --git a/cmuxTests/CommandPaletteSearchEngineTests.swift b/cmuxTests/CommandPaletteSearchEngineTests.swift
index 32d8ef82..7ce4ef0c 100644
--- a/cmuxTests/CommandPaletteSearchEngineTests.swift
+++ b/cmuxTests/CommandPaletteSearchEngineTests.swift
@@ -219,6 +219,42 @@ final class CommandPaletteSearchEngineTests: XCTestCase {
         XCTAssertGreaterThanOrEqual(cancellationChecks, 4)
     }
 
+    func testCommandPreviewSearchUsesFullCommandCorpus() {
+        let entries = [
+            FixtureEntry(
+                id: "command.find",
+                rank: 0,
+                title: "Find...",
+                searchableTexts: ["Find...", "Search", "find", "search"]
+            ),
+            FixtureEntry(
+                id: "command.finder",
+                rank: 1,
+                title: "Open Current Directory in Finder",
+                searchableTexts: ["Open Current Directory in Finder", "Terminal", "finder", "directory", "open"]
+            ),
+        ]
+        let corpus = entries.map { entry in
+            CommandPaletteSearchCorpusEntry(
+                payload: entry.id,
+                rank: entry.rank,
+                title: entry.title,
+                searchableTexts: entry.searchableTexts
+            )
+        }
+        let corpusByID = Dictionary(uniqueKeysWithValues: corpus.map { ($0.payload, $0) })
+
+        let previewCommandIDs = ContentView.commandPaletteCommandPreviewMatchCommandIDsForTests(
+            searchCorpus: corpus,
+            candidateCommandIDs: ["command.find"],
+            searchCorpusByID: corpusByID,
+            query: "finde",
+            resultLimit: 48
+        )
+
+        XCTAssertEqual(previewCommandIDs.first, "command.finder")
+    }
+
     func testResolvedSelectionIndexPrefersAnchoredCommand() {
         let resultIDs = ["command.0", "command.1", "command.2"]