ktx/packages/cli/test/prompt-navigation.test.ts

import { describe, expect, it } from 'vitest';
import {
  FLAT_MULTISELECT_NAVIGATION_HINT,
  MULTISELECT_NAVIGATION_FRAGMENTS,
  SEARCHABLE_MULTISELECT_NAVIGATION_HINT,
  TREE_PICKER_NAVIGATION_HINT,
  withMenuOptionSpacing,
  withMultiselectNavigation,
  withSearchableMultiselectNavigation,
  withTextInputNavigation,
} from '../src/prompt-navigation.js';

describe('prompt navigation helpers', () => {
  it('leaves compact single-line menu prompts unchanged', () => {
    expect(withMenuOptionSpacing('What do you want to do?')).toBe('What do you want to do?');
  });

  it('adds a blank separator between multiline menu copy and the option list', () => {
    expect(withMenuOptionSpacing('Which embedding option should ktx use?\n\nktx uses embeddings for search.')).toBe(
      'Which embedding option should ktx use?\n\nktx uses embeddings for search.\n',
    );
  });

  it('does not duplicate an existing option-list separator', () => {
    expect(withMenuOptionSpacing('Question\n\nContext\n')).toBe('Question\n\nContext\n');
  });

  it('keeps multiselect navigation copy multiline so menu renderers can separate it from options', () => {
    expect(withMultiselectNavigation('Which sources?')).toBe(
      'Which sources?\nUp/Down to move, Tab to select or unselect, Enter to confirm, Escape to go back, Ctrl+C to exit.',
    );
  });

  it('appends the searchable hint for autocomplete multiselect prompts', () => {
    expect(withSearchableMultiselectNavigation('Choose schemas')).toBe(
      'Choose schemas\nUp/Down to move, Tab to select or unselect, Type to search, Enter to confirm, Escape to go back, Ctrl+C to exit.',
    );
  });

  it('does not duplicate the searchable hint when applied twice', () => {
    const once = withSearchableMultiselectNavigation('Choose schemas');
    expect(withSearchableMultiselectNavigation(once)).toBe(once);
  });

  it('matches the approved hint wording for each multi-select surface', () => {
    expect(FLAT_MULTISELECT_NAVIGATION_HINT).toBe(
      'Up/Down to move, Tab to select or unselect, Enter to confirm, Escape to go back, Ctrl+C to exit.',
    );
    expect(SEARCHABLE_MULTISELECT_NAVIGATION_HINT).toBe(
      'Up/Down to move, Tab to select or unselect, Type to search, Enter to confirm, Escape to go back, Ctrl+C to exit.',
    );
    expect(TREE_PICKER_NAVIGATION_HINT).toBe(
      'Up/Down to move, Right/Left to expand or collapse, Tab to select or unselect, Type to search, Enter to confirm, Escape to clear search or go back, Ctrl+C to exit.',
    );
  });

  it('composes every hint from the shared fragment vocabulary so wording cannot drift', () => {
    const hints = [
      FLAT_MULTISELECT_NAVIGATION_HINT,
      SEARCHABLE_MULTISELECT_NAVIGATION_HINT,
      TREE_PICKER_NAVIGATION_HINT,
    ];
    const sharedFragments = [
      MULTISELECT_NAVIGATION_FRAGMENTS.move,
      MULTISELECT_NAVIGATION_FRAGMENTS.select,
      MULTISELECT_NAVIGATION_FRAGMENTS.confirm,
      MULTISELECT_NAVIGATION_FRAGMENTS.exit,
    ];
    for (const fragment of sharedFragments) {
      for (const hint of hints) {
        expect(hint).toContain(fragment);
      }
    }
    expect(MULTISELECT_NAVIGATION_FRAGMENTS.select).toBe('Tab to select or unselect');
    for (const hint of hints) {
      expect(hint).not.toContain('Space');
    }
  });

  it('adds a blank separator between text input helper copy and the editable value', () => {
    expect(
      withTextInputNavigation(
        'Name this PostgreSQL connection\nktx will use this short name in commands and config. You can rename it now.',
      ),
    ).toBe(
      'Name this PostgreSQL connection\n│\n│  ktx will use this short name in commands and config. You can rename it now.\n│  Press Escape to go back.\n│',
    );
  });

  it('adds a blank separator before compact text input values', () => {
    expect(withTextInputNavigation('Project folder path')).toBe('Project folder path\n│  Press Escape to go back.\n│');
  });

  it('normalizes already hinted text input prompts without duplicating the hint', () => {
    expect(
      withTextInputNavigation(
        'Name this PostgreSQL connection\nktx will use this short name in commands and config. You can rename it now.\nPress Escape to go back.',
      ),
    ).toBe(
      'Name this PostgreSQL connection\n│\n│  ktx will use this short name in commands and config. You can rename it now.\n│  Press Escape to go back.\n│',
    );
  });

  it('is idempotent when text input navigation is applied twice', () => {
    const once = withTextInputNavigation('Project folder path');
    expect(withTextInputNavigation(once)).toBe(once);
  });

  it('is idempotent when text input navigation with body is applied twice', () => {
    const once = withTextInputNavigation(
      'Name this PostgreSQL connection\nktx will use this short name in commands and config.',
    );
    expect(withTextInputNavigation(once)).toBe(once);
  });
});