fix console & buffered console assert behaviour

Author: ranyitz

packages/jest-util/src/Console.js

@@ -10,6 +10,7 @@
 
 import type {LogType, LogMessage, LogCounters, LogTimers} from 'types/Console';
 
+import assert from 'assert';
 import {format} from 'util';
 import {Console} from 'console';
 import chalk from 'chalk';
@@ -48,8 +49,10 @@ export default class CustomConsole extends Console {
   }
 
   assert(...args: Array<any>) {
-    if (args[0]) {
-      this._log('assert', format(...args.slice(1)));
+    try {
+      assert(!!args[0], ...args.slice(1));
+    } catch (error) {
+      this._log('assert', error.toString());
     }
   }
 

packages/jest-util/src/__tests__/buffered_console.test.js

@@ -23,16 +23,28 @@ describe('CustomConsole', () => {
   });
 
   describe('assert', () => {
-    test('log when the assertion is truthy', () => {
+    test('do not log when the assertion is truthy', () => {
+      _console.assert(true);
+
+      expect(stdout()).toMatch('');
+    });
+
+    test('do not log when the assertion is truthy and there is a message', () => {
       _console.assert(true, 'ok');
 
-      expect(stdout()).toMatch('ok');
+      expect(stdout()).toMatch('');
+    });
+
+    test('log the assertion error when the assertion is falsy', () => {
+      _console.assert(false);
+
+      expect(stdout()).toEqual('AssertionError [ERR_ASSERTION]: false == true');
     });
 
-    test('do not log when the assertion is falsy', () => {
+    test('log the assertion error when the assertion is falsy with another message argument', () => {
       _console.assert(false, 'ok');
 
-      expect(stdout()).toEqual('');
+      expect(stdout()).toEqual('AssertionError [ERR_ASSERTION]: ok');
     });
   });
 

packages/jest-util/src/__tests__/console.test.js

@@ -24,16 +24,30 @@ describe('CustomConsole', () => {
   });
 
   describe('assert', () => {
-    test('log when the assertion is truthy', () => {
+    test('do not log when the assertion is truthy', () => {
+      _console.assert(true);
+
+      expect(_stdout).toMatch('');
+    });
+
+    test('do not log when the assertion is truthy and there is a message', () => {
       _console.assert(true, 'ok');
 
-      expect(_stdout).toMatch('ok');
+      expect(_stdout).toMatch('');
+    });
+
+    test('log the assertion error when the assertion is falsy', () => {
+      _console.assert(false);
+
+      expect(_stdout).toEqual(
+        'AssertionError [ERR_ASSERTION]: false == true\n',
+      );
     });
 
-    test('do not log when the assertion is falsy', () => {
+    test('log the assertion error when the assertion is falsy with another message argument', () => {
       _console.assert(false, 'ok');
 
-      expect(_stdout).toEqual('');
+      expect(_stdout).toEqual('AssertionError [ERR_ASSERTION]: ok\n');
     });
   });
 

packages/jest-util/src/buffered_console.js

@@ -15,6 +15,7 @@ import type {
   LogTimers,
 } from 'types/Console';
 
+import assert from 'assert';
 import {Console} from 'console';
 import {format} from 'util';
 import chalk from 'chalk';
@@ -63,8 +64,10 @@ export default class BufferedConsole extends Console {
   }
 
   assert(...args: Array<any>) {
-    if (args[0]) {
-      this._log('assert', format(...args.slice(1)));
+    try {
+      assert(!!args[0], ...args.slice(1));
+    } catch (error) {
+      this._log('assert', error.toString());
     }
   }