add only-failures mode (#4886)

* add only-failures mode

Signed-off-by: Łukasz Sentkiewicz <sirmims@gmail.com>

lint fixes

Signed-off-by: Łukasz Sentkiewicz <sirmims@gmail.com>

fix tests

Signed-off-by: Łukasz Sentkiewicz <sirmims@gmail.com>

fix tests

Signed-off-by: Łukasz Sentkiewicz <sirmims@gmail.com>

* replace 'only-failures' message

Signed-off-by: Łukasz Sentkiewicz <sirmims@gmail.com>

Author: lstkz

packages/jest-cli/src/__tests__/__snapshots__/watch.test.js.snap

@@ -12,6 +12,7 @@ Array [
   "
 Watch Usage
  › Press a to run all tests.
+ › Press f to run only failed tests.
  › Press p to filter by a filename regex pattern.
  › Press t to filter by a test name regex pattern.
  › Press q to quit watch mode.
@@ -26,6 +27,7 @@ Array [
     "
 Watch Usage
  › Press a to run all tests.
+ › Press f to run only failed tests.
  › Press p to filter by a filename regex pattern.
  › Press t to filter by a test name regex pattern.
  › Press s to do nothing.

packages/jest-cli/src/__tests__/failed_tests_cache.test.js

@@ -0,0 +1,62 @@
+import FailedTestsCache from '../failed_tests_cache';
+
+describe('FailedTestsCache', () => {
+  test('should filter tests', () => {
+    const failedTestsCache = new FailedTestsCache();
+    failedTestsCache.setTestResults([
+      {
+        numFailingTests: 0,
+        testFilePath: '/path/to/passing.js',
+        testResults: [
+          {fullName: 'test 1', status: 'passed'},
+          {fullName: 'test 2', status: 'passed'},
+        ],
+      },
+      {
+        numFailingTests: 2,
+        testFilePath: '/path/to/failed_1.js',
+        testResults: [
+          {fullName: 'test 3', status: 'failed'},
+          {fullName: 'test 4', status: 'failed'},
+        ],
+      },
+      {
+        numFailingTests: 1,
+        testFilePath: '/path/to/failed_2.js',
+        testResults: [
+          {fullName: 'test 5', status: 'failed'},
+          {fullName: 'test 6', status: 'passed'},
+        ],
+      },
+    ]);
+
+    const result = failedTestsCache.filterTests([
+      {
+        path: '/path/to/passing.js',
+      },
+      {
+        path: '/path/to/failed_1.js',
+      },
+      {
+        path: '/path/to/failed_2.js',
+      },
+      {
+        path: '/path/to/unknown.js',
+      },
+    ]);
+    expect(result).toMatchObject([
+      {
+        path: '/path/to/failed_1.js',
+      },
+      {
+        path: '/path/to/failed_2.js',
+      },
+    ]);
+    expect(failedTestsCache.updateConfig({})).toMatchObject({
+      enabledTestsMap: {
+        '/path/to/failed_1.js': {'test 3': true, 'test 4': true},
+        '/path/to/failed_2.js': {'test 5': true},
+      },
+    });
+  });
+});

packages/jest-cli/src/cli/args.js

@@ -357,6 +357,12 @@ export const options = {
       'running tests in a git repository at the moment.',
     type: 'boolean',
   },
+  onlyFailures: {
+    alias: 'f',
+    default: undefined,
+    description: 'Run tests that failed in the previous execution.',
+    type: 'boolean',
+  },
   outputFile: {
     description:
       'Write test results to a file when the --json option is ' +

packages/jest-cli/src/cli/index.js

@@ -368,6 +368,7 @@ const runWithoutWatch = async (
     return await runJest({
       changedFilesPromise,
       contexts,
+      failedTestsCache: null,
       globalConfig,
       onComplete,
       outputStream,

packages/jest-cli/src/constants.js

@@ -23,6 +23,7 @@ export const KEYS = {
   CONTROL_D: '04',
   ENTER: '0d',
   ESCAPE: '1b',
+  F: '66',
   O: '6f',
   P: '70',
   Q: '71',

packages/jest-cli/src/failed_tests_cache.js

@@ -0,0 +1,49 @@
+/**
+ * Copyright (c) 2014-present, Facebook, Inc. All rights reserved.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @flow
+ */
+
+import type {Test} from 'types/TestRunner';
+import type {TestResult} from 'types/TestResult';
+import type {GlobalConfig} from 'types/Config';
+
+export default class FailedTestsCache {
+  _enabledTestsMap: ?{[key: string]: {[key: string]: boolean}};
+
+  filterTests(tests: Array<Test>): Array<Test> {
+    if (!this._enabledTestsMap) {
+      return tests;
+    }
+    // $FlowFixMe
+    return tests.filter(testResult => this._enabledTestsMap[testResult.path]);
+  }
+
+  setTestResults(testResults: Array<TestResult>) {
+    this._enabledTestsMap = (testResults || [])
+      .filter(testResult => testResult.numFailingTests)
+      .reduce((suiteMap, testResult) => {
+        suiteMap[testResult.testFilePath] = testResult.testResults
+          .filter(test => test.status === 'failed')
+          .reduce((testMap, test) => {
+            testMap[test.fullName] = true;
+            return testMap;
+          }, {});
+        return suiteMap;
+      }, {});
+    this._enabledTestsMap = Object.freeze(this._enabledTestsMap);
+  }
+
+  updateConfig(globalConfig: GlobalConfig): GlobalConfig {
+    if (!this._enabledTestsMap) {
+      return globalConfig;
+    }
+    // $FlowFixMe Object.assign
+    const newConfig: GlobalConfig = Object.assign({}, globalConfig);
+    newConfig.enabledTestsMap = this._enabledTestsMap;
+    return Object.freeze(newConfig);
+  }
+}

packages/jest-cli/src/get_no_test_found_failed.js

@@ -0,0 +1,8 @@
+import chalk from 'chalk';
+
+export default function getNoTestFoundFailed() {
+  return (
+    chalk.bold('No failed test found.\n') +
+    chalk.dim('Press `f` to run all tests.')
+  );
+}

packages/jest-cli/src/get_no_test_found_message.js

@@ -1,11 +1,15 @@
 import getNoTestFound from './get_no_test_found';
 import getNoTestFoundRelatedToChangedFiles from './get_no_test_found_related_to_changed_files';
 import getNoTestFoundVerbose from './get_no_test_found_verbose';
+import getNoTestFoundFailed from './get_no_test_found_failed';
 
 export default function getNoTestsFoundMessage(
   testRunData,
   globalConfig,
 ): string {
+  if (globalConfig.onlyFailures) {
+    return getNoTestFoundFailed();
+  }
   if (globalConfig.onlyChanged) {
     return getNoTestFoundRelatedToChangedFiles(globalConfig);
   }

packages/jest-cli/src/lib/update_global_config.js

@@ -16,6 +16,7 @@ type Options = {
   updateSnapshot?: SnapshotUpdateState,
   mode?: 'watch' | 'watchAll',
   passWithNoTests?: boolean,
+  onlyFailures?: boolean,
 };
 
 export default (globalConfig: GlobalConfig, options: Options): GlobalConfig => {
@@ -60,5 +61,9 @@ export default (globalConfig: GlobalConfig, options: Options): GlobalConfig => {
     newConfig.passWithNoTests = true;
   }
 
+  if ('onlyFailures' in options) {
+    newConfig.onlyFailures = options.onlyFailures || false;
+  }
+
   return Object.freeze(newConfig);
 };

packages/jest-cli/src/run_jest.js

@@ -22,6 +22,7 @@ import SearchSource from './search_source';
 import TestScheduler from './test_scheduler';
 import TestSequencer from './test_sequencer';
 import {makeEmptyAggregatedTestResult} from './test_result_helpers';
+import FailedTestsCache from './failed_tests_cache';
 
 const setConfig = (contexts, newConfig) =>
   contexts.forEach(
@@ -82,6 +83,7 @@ export default (async function runJest({
   startRun,
   changedFilesPromise,
   onComplete,
+  failedTestsCache,
 }: {
   globalConfig: GlobalConfig,
   contexts: Array<Context>,
@@ -90,6 +92,7 @@ export default (async function runJest({
   startRun: (globalConfig: GlobalConfig) => *,
   changedFilesPromise: ?ChangedFilesPromise,
   onComplete: (testResults: AggregatedResult) => any,
+  failedTestsCache: ?FailedTestsCache,
 }) {
   if (globalConfig.globalSetup) {
     // $FlowFixMe
@@ -138,6 +141,12 @@ export default (async function runJest({
     onComplete && onComplete(makeEmptyAggregatedTestResult());
     return null;
   }
+
+  if (globalConfig.onlyFailures && failedTestsCache) {
+    allTests = failedTestsCache.filterTests(allTests);
+    globalConfig = failedTestsCache.updateConfig(globalConfig);
+  }
+
   if (!allTests.length) {
     const noTestsFoundMessage = getNoTestsFoundMessage(
       testRunData,