jest-watch-typeahead/build/test_name_plugin/prompt.js

Jest plugin for filtering by filename or test name

import chalk from 'chalk';
import ansiEscapes from 'ansi-escapes';
import { PatternPrompt, printPatternCaret, printRestoredPatternCaret } from 'jest-watcher';
import { escapeStrForRegex } from 'jest-regex-util';
import scroll from "../lib/scroll.js";
import { formatTestNameByPattern, getTerminalWidth, removeTrimmingDots } from "../lib/utils.js";
import { formatTypeaheadSelection, printMore, printPatternMatches, printStartTyping, printTypeaheadItem } from "../lib/pattern_mode_helpers.js";
export default class TestNamePatternPrompt extends PatternPrompt {
  constructor(pipe, prompt) {
    super(pipe, prompt);
    this._entityName = 'tests';
    this._cachedTestResults = [];
    this._offset = -1;
  }
  _onChange(pattern, options) {
    super._onChange(pattern, options);
    this._offset = options.offset;
    this._printTypeahead(pattern, options);
  }
  _printTypeahead(pattern, options) {
    const matchedTests = this._getMatchedTests(pattern);
    const total = matchedTests.length;
    const pipe = this._pipe;
    const prompt = this._prompt;
    printPatternCaret(pattern, pipe);
    pipe.write(ansiEscapes.cursorLeft);
    if (pattern) {
      printPatternMatches(total, 'test', pipe, ` from ${chalk.yellow('cached')} test suites`);
      const width = getTerminalWidth(pipe);
      const {
        start,
        end,
        index
      } = scroll(total, options);
      prompt.setPromptLength(total);
      matchedTests.slice(start, end).map(name => formatTestNameByPattern(name, pattern, width - 4)).map((item, i) => formatTypeaheadSelection(item, i, index, prompt)).forEach(item => printTypeaheadItem(item, pipe));
      if (total > end) {
        printMore('test', pipe, total - end);
      }
    } else {
      printStartTyping('test name', pipe);
    }
    printRestoredPatternCaret(pattern, this._currentUsageRows, pipe);
  }
  _getMatchedTests(pattern) {
    let regex;
    try {
      regex = new RegExp(pattern, 'i');
    } catch {
      return [];
    }
    return this._cachedTestResults.reduce((matchedTests, {
      testResults
    }) => {
      return matchedTests.concat(testResults.filter(({
        fullName
      }) => regex.test(fullName)).map(({
        fullName
      }) => fullName));
    }, []);
  }
  updateCachedTestResults(testResults = []) {
    this._cachedTestResults = testResults;
  }
  run(onSuccess, onCancel, options) {
    super.run(value => {
      const preparedPattern = escapeStrForRegex(removeTrimmingDots(value));
      const useExactMatch = this._offset !== -1;
      onSuccess(useExactMatch ? `^${preparedPattern}$` : preparedPattern);
    }, onCancel, options);
  }
}