diff --git a/packages/cli/src/ui/commands/ideCommand.test.ts b/packages/cli/src/ui/commands/ideCommand.test.ts index 663e5e07..8576320b 100644 --- a/packages/cli/src/ui/commands/ideCommand.test.ts +++ b/packages/cli/src/ui/commands/ideCommand.test.ts @@ -69,9 +69,6 @@ describe('ideCommand', () => { vi.mocked(mockConfig.getIdeClient).mockReturnValue({ getCurrentIde: () => DetectedIde.VSCode, getDetectedIdeDisplayName: () => 'VS Code', - getConnectionStatus: () => ({ - status: core.IDEConnectionStatus.Connected, - }), } as ReturnType); const command = ideCommand(mockConfig); expect(command).not.toBeNull(); @@ -164,9 +161,7 @@ describe('ideCommand', () => { vi.mocked(mockConfig.getIdeMode).mockReturnValue(true); vi.mocked(mockConfig.getIdeClient).mockReturnValue({ getCurrentIde: () => DetectedIde.VSCode, - getConnectionStatus: () => ({ - status: core.IDEConnectionStatus.Disconnected, - }), + getConnectionStatus: vi.fn(), getDetectedIdeDisplayName: () => 'VS Code', } as unknown as ReturnType); vi.mocked(core.getIdeInstaller).mockReturnValue({ diff --git a/packages/cli/src/ui/commands/ideCommand.ts b/packages/cli/src/ui/commands/ideCommand.ts index ffc9df40..b7cbea3d 100644 --- a/packages/cli/src/ui/commands/ideCommand.ts +++ b/packages/cli/src/ui/commands/ideCommand.ts @@ -237,8 +237,8 @@ export const ideCommand = (config: Config | null): SlashCommand | null => { }, }; - const connectionStatus = ideClient.getConnectionStatus().status; - if (connectionStatus === IDEConnectionStatus.Connected) { + const ideModeEnabled = config.getIdeMode(); + if (ideModeEnabled) { ideSlashCommand.subCommands = [ disableCommand, statusCommand,