From c275748edab7b69dd88e18d8cfc31fc825742c45 Mon Sep 17 00:00:00 2001
From: Andrey Avtomonov <7889985+andreybavt@users.noreply.github.com>
Date: Thu, 21 May 2026 20:08:12 +0200
Subject: [PATCH] test: update setup prompt stubs for type-check

---
 packages/cli/src/connectors/bigquery/connector.test.ts | 10 ++++++----
 packages/cli/src/setup-databases.test.ts               |  4 +++-
 packages/cli/src/setup-sources-notion.test.ts          |  1 +
 packages/cli/src/setup.test.ts                         |  1 +
 4 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/packages/cli/src/connectors/bigquery/connector.test.ts b/packages/cli/src/connectors/bigquery/connector.test.ts
index 314534dc..c517100a 100644
--- a/packages/cli/src/connectors/bigquery/connector.test.ts
+++ b/packages/cli/src/connectors/bigquery/connector.test.ts
@@ -236,9 +236,11 @@ describe('KtxBigQueryScanConnector', () => {
 
   it('constructs for discovery without dataset scope and lists tables through one region information schema query', async () => {
     const createQueryJob = vi.fn(
-      async (input: { query: string; params?: Record<string, unknown>; location?: string }) => [
+      async (
+        input: { query: string; params?: Record<string, unknown>; location?: string },
+      ): ReturnType<KtxBigQueryClient['createQueryJob']> => [
         {
-          getQueryResults: async () => [
+          getQueryResults: async (): ReturnType<KtxBigQueryQueryJob['getQueryResults']> => [
             [
               { table_schema: 'analytics', table_name: 'orders', table_type: 'BASE TABLE' },
               { table_schema: 'analytics', table_name: 'order_clone', table_type: 'CLONE' },
@@ -260,10 +262,10 @@ describe('KtxBigQueryScanConnector', () => {
     );
     const clientFactory: KtxBigQueryClientFactory = {
       createClient: vi.fn(() => ({
-        getDatasets: vi.fn(async () => [[{ id: 'analytics' }, { id: 'mart' }]]),
+        getDatasets: vi.fn(async () => [[{ id: 'analytics' }, { id: 'mart' }]] as [{ id: string }[]]),
         dataset: vi.fn((datasetId: string) => ({
           get: vi.fn(async () => [{ id: datasetId }]),
-          getTables: vi.fn(async () => [[]]),
+          getTables: vi.fn(async () => [[]] as [never[]]),
         })),
         createQueryJob,
       })),
diff --git a/packages/cli/src/setup-databases.test.ts b/packages/cli/src/setup-databases.test.ts
index d72029d2..904f2c89 100644
--- a/packages/cli/src/setup-databases.test.ts
+++ b/packages/cli/src/setup-databases.test.ts
@@ -101,7 +101,9 @@ function makePromptAdapter(options: {
       if (options.initialValues && options.initialValues.length > 0) {
         return options.initialValues;
       }
-      return options.options.length > 0 ? options.options.map((option) => option.value) : ['back'];
+      return options.options.length > 0
+        ? options.options.map((option: { value: string }) => option.value)
+        : ['back'];
     }),
     select: vi.fn(async ({ message }) => {
       if (message.startsWith('Save ') && message.includes(' or refine tables?')) {
diff --git a/packages/cli/src/setup-sources-notion.test.ts b/packages/cli/src/setup-sources-notion.test.ts
index 84fce4aa..1306b07b 100644
--- a/packages/cli/src/setup-sources-notion.test.ts
+++ b/packages/cli/src/setup-sources-notion.test.ts
@@ -57,6 +57,7 @@ function prompts(values: { multiselect?: string[][]; select?: string[] }): KtxSe
   return {
     multiselect: vi.fn(async () => multiselectValues.shift() ?? []),
     select: vi.fn(async () => selectValues.shift() ?? 'back'),
+    autocomplete: vi.fn(async () => selectValues.shift() ?? 'back'),
     text: vi.fn(async () => ''),
     password: vi.fn(async () => undefined),
     cancel: vi.fn(),
diff --git a/packages/cli/src/setup.test.ts b/packages/cli/src/setup.test.ts
index 9c4a3b58..6ea0f0a4 100644
--- a/packages/cli/src/setup.test.ts
+++ b/packages/cli/src/setup.test.ts
@@ -1410,6 +1410,7 @@ describe('setup status', () => {
     const embeddings = vi.fn(async () => embeddingResults.shift() ?? { status: 'back' as const, projectDir: tempDir });
     const databasePrompts = {
       multiselect: vi.fn(async () => ['back']),
+      autocompleteMultiselect: vi.fn(async () => ['back']),
       select: vi.fn(async () => 'back'),
       text: vi.fn(),
       password: vi.fn(),