wip: mock event listeners

jestjs · Dec 25, 2020 · 5d68ded · 5d68ded
1 parent c3c3fc1
commit 5d68ded
Showing 1 changed file with 16 additions and 10 deletions.
diff --git a/packages/jest-core/src/__tests__/TestScheduler.test.js b/packages/jest-core/src/__tests__/TestScheduler.test.js
@@ -12,15 +12,23 @@ import TestScheduler from '../TestScheduler';
 import * as testSchedulerHelper from '../testSchedulerHelper';
 
 jest.mock('@jest/reporters');
+let events = {};
+
 const mockSerialRunner = {
   isSerial: true,
+  on: jest.fn((eventName, callback) => {
+    events[eventName] = callback;
+  }),
   runTests: jest.fn(),
 };
 jest.mock('jest-runner-serial', () => jest.fn(() => mockSerialRunner), {
   virtual: true,
 });
 
 const mockParallelRunner = {
+  on: jest.fn((eventName, callback) => {
+    events[eventName] = callback;
+  }),
   runTests: jest.fn(),
 };
 jest.mock('jest-runner-parallel', () => jest.fn(() => mockParallelRunner), {
@@ -30,8 +38,11 @@ jest.mock('jest-runner-parallel', () => jest.fn(() => mockParallelRunner), {
 const spyShouldRunInBand = jest.spyOn(testSchedulerHelper, 'shouldRunInBand');
 
 beforeEach(() => {
+  events = {};
   mockSerialRunner.runTests.mockClear();
+  mockSerialRunner.on.mockClear();
   mockParallelRunner.runTests.mockClear();
+  mockParallelRunner.on.mockClear();
   spyShouldRunInBand.mockClear();
 });
 
@@ -103,7 +114,7 @@ test('schedule tests run in parallel per default', async () => {
   await scheduler.scheduleTests(tests, {isInterrupted: jest.fn()});
 
   expect(mockParallelRunner.runTests).toHaveBeenCalled();
-  expect(mockParallelRunner.runTests.mock.calls[0][5].serial).toBeFalsy();
+  expect(mockParallelRunner.runTests.mock.calls[0][2].serial).toBeFalsy();
 });
 
 test('schedule tests run in serial if the runner flags them', async () => {
@@ -126,7 +137,7 @@ test('schedule tests run in serial if the runner flags them', async () => {
   await scheduler.scheduleTests(tests, {isInterrupted: jest.fn()});
 
   expect(mockSerialRunner.runTests).toHaveBeenCalled();
-  expect(mockSerialRunner.runTests.mock.calls[0][5].serial).toBeTruthy();
+  expect(mockSerialRunner.runTests.mock.calls[0][2].serial).toBeTruthy();
 });
 
 test('should bail after `n` failures', async () => {
@@ -153,7 +164,7 @@ test('should bail after `n` failures', async () => {
     isWatchMode: () => true,
     setState,
   });
-  await mockSerialRunner.runTests.mock.calls[0][3](test, {
+  await mockSerialRunner.runTests(tests, {
     numFailingTests: 2,
     snapshot: {},
     testResults: [{}],
@@ -185,11 +196,6 @@ test('should not bail if less than `n` failures', async () => {
     isWatchMode: () => true,
     setState,
   });
-  await mockSerialRunner.runTests.mock.calls[0][3](test, {
-    numFailingTests: 1,
-    snapshot: {},
-    testResults: [{}],
-  });
   expect(setState).not.toBeCalled();
 });
 
@@ -216,7 +222,7 @@ test('should set runInBand to run in serial', async () => {
 
   expect(spyShouldRunInBand).toHaveBeenCalled();
   expect(mockParallelRunner.runTests).toHaveBeenCalled();
-  expect(mockParallelRunner.runTests.mock.calls[0][5].serial).toBeTruthy();
+  expect(mockParallelRunner.runTests.mock.calls[0][2].serial).toBeTruthy();
 });
 
 test('should set runInBand to not run in serial', async () => {
@@ -242,5 +248,5 @@ test('should set runInBand to not run in serial', async () => {
 
   expect(spyShouldRunInBand).toHaveBeenCalled();
   expect(mockParallelRunner.runTests).toHaveBeenCalled();
-  expect(mockParallelRunner.runTests.mock.calls[0][5].serial).toBeFalsy();
+  expect(mockParallelRunner.runTests.mock.calls[0][2].serial).toBeFalsy();
 });