pass testConsole to TestEnvironment

CHANGELOG.md

@@ -4,6 +4,11 @@
 
 * `[jest-config]` fix unexpected condition to avoid infinite recursion in
   Windows platform. ([#5161](https://github.com/facebook/jest/pull/5161))
+* `[jest-runner]` pass `console` to environments so jsdom can use the same
+  console as the test. ([#5223](https://github.com/facebook/jest/issues/5223))
+* `[jest-environment-jsdom]` pass the `config.console` to a custom
+  instance of `virtualConsole` so jsdom is using the same console as the
+  test. ([#5223](https://github.com/facebook/jest/issues/5223))
 
 ### Features
 

integration_tests/__tests__/__snapshots__/console.test.js.snap

@@ -73,3 +73,21 @@ Snapshots:   0 total
 Time:        <<REPLACED>>
 "
 `;
+
+exports[`the jsdom console is the same as the test console 1`] = `""`;
+
+exports[`the jsdom console is the same as the test console 2`] = `
+"PASS __tests__/console.test.js
+  ✓ can mock console.error calls from jsdom
+
+"
+`;
+
+exports[`the jsdom console is the same as the test console 3`] = `
+"Test Suites: 1 passed, 1 total
+Tests:       1 passed, 1 total
+Snapshots:   0 total
+Time:        <<REPLACED>>
+Ran all test suites.
+"
+`;

integration_tests/__tests__/console.test.js

@@ -54,3 +54,14 @@ test('does not print to console with --silent', () => {
   expect(rest).toMatchSnapshot();
   expect(summary).toMatchSnapshot();
 });
+
+// issue: https://github.com/facebook/jest/issues/5223
+test('the jsdom console is the same as the test console', () => {
+  const {stderr, stdout, status} = runJest('console-jsdom');
+  const {summary, rest} = extractSummary(stderr);
+
+  expect(status).toBe(0);
+  expect(stdout).toMatchSnapshot();
+  expect(rest).toMatchSnapshot();
+  expect(summary).toMatchSnapshot();
+});

integration_tests/console-jsdom/__tests__/console.test.js

@@ -0,0 +1,40 @@
+/**
+ * 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.
+ */
+/* eslint-env browser */
+'use strict';
+
+beforeEach(() => {
+  jest.spyOn(console, 'error');
+  console.error.mockImplementation(() => {});
+});
+
+afterEach(() => {
+  console.error.mockRestore();
+});
+
+test('can mock console.error calls from jsdom', () => {
+  // copied and modified for tests from:
+  // https://github.com/facebook/react/blob/46b3c3e4ae0d52565f7ed2344036a22016781ca0/packages/shared/invokeGuardedCallback.js#L62-L147
+  const fakeNode = document.createElement('react');
+
+  const evt = document.createEvent('Event');
+  const evtType = 'react-invokeguardedcallback';
+  function callCallback() {
+    fakeNode.removeEventListener(evtType, callCallback, false);
+    throw new Error('this is an error in an event callback');
+  }
+
+  function onError(event) {}
+
+  window.addEventListener('error', onError);
+  fakeNode.addEventListener(evtType, callCallback, false);
+  evt.initEvent(evtType, false, false);
+  fakeNode.dispatchEvent(evt);
+  window.removeEventListener('error', onError);
+
+  expect(console.error).toHaveBeenCalledTimes(1);
+});

integration_tests/console-jsdom/package.json

@@ -0,0 +1,5 @@
+{
+  "jest": {
+    "testEnvironment": "jsdom"
+  }
+}

integration_tests/snapshot-escape/__tests__/__snapshots__/snapshot.test.js.snap

@@ -0,0 +1,5 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`escape strings 1`] = `"one: \\\\'"`;
+
+exports[`escape strings two 1`] = `"two: '\\""`;

integration_tests/snapshot-escape/__tests__/snapshot.test.js

@@ -8,3 +8,4 @@
 
 // prettier-ignore
 test('escape strings', () => expect('one: \\\'').toMatchSnapshot());
+test('escape strings two', () => expect('two: \'"').toMatchSnapshot());

packages/jest-environment-jsdom/src/index.js

@@ -13,7 +13,7 @@ import type {ModuleMocker} from 'jest-mock';
 
 import {FakeTimers, installCommonGlobals} from 'jest-util';
 import mock from 'jest-mock';
-import {JSDOM} from 'jsdom';
+import {JSDOM, VirtualConsole} from 'jsdom';
 
 class JSDOMEnvironment {
   dom: ?Object;
@@ -30,6 +30,9 @@ class JSDOMEnvironment {
           pretendToBeVisual: true,
           runScripts: 'dangerously',
           url: config.testURL,
+          virtualConsole: new VirtualConsole().sendTo(
+            config.console || console,
+          ),
         },
         config.testEnvironmentOptions,
       ),

packages/jest-runner/src/run_test.js

@@ -74,11 +74,6 @@ async function runTestInternal(
     RuntimeClass,
   >);
 
-  const environment = new TestEnvironment(config);
-  const leakDetector = config.detectLeaks
-    ? new LeakDetector(environment)
-    : null;
-
   const consoleOut = globalConfig.useStderr ? process.stderr : process.stdout;
   const consoleFormatter = (type, message) =>
     getConsoleOutput(
@@ -98,6 +93,13 @@ async function runTestInternal(
     testConsole = new BufferedConsole();
   }
 
+  const environment = new TestEnvironment(
+    Object.assign({}, config, {console: testConsole}),
+  );
+  const leakDetector = config.detectLeaks
+    ? new LeakDetector(environment)
+    : null;
+
   const cacheFS = {[path]: testSource};
   setGlobal(environment.global, 'console', testConsole);
 

types/Config.js

@@ -210,6 +210,7 @@ export type ProjectConfig = {|
   cache: boolean,
   cacheDirectory: Path,
   clearMocks: boolean,
+  console?: Object,
   coveragePathIgnorePatterns: Array<string>,
   cwd: Path,
   detectLeaks: boolean,