add a test for keepAliveAfterFinishCallback

trivago · Dec 18, 2018 · 2a8b6b2 · 2a8b6b2
1 parent f6103d7
commit 2a8b6b2
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/__tests__/index.spec.js b/__tests__/index.spec.js
@@ -128,6 +128,19 @@ describe('index.js', () => {
                 // called with workers
                 expect(workerFarm.end.mock.calls[0][0]).toBe(workerFarm.end);
             });
+            it('should call keepAliveAfterFinishCallback if flag is set', () => {
+                const options = {};
+                Object.defineProperty(options, 'keepAliveAfterFinish', {
+                    value: 500
+                });
+                const keepAliveAfterFinishCallback = jest.fn(() => {
+                    setTimeout(expect.any(Function), options.keepAliveAfterFinish);
+                });
+                keepAliveAfterFinishCallback();
+                debugger;
+                expect(setTimeout).toHaveBeenCalledTimes(1);
+                expect(setTimeout).toHaveBeenLastCalledWith(expect.any(Function), 500);
+            });
         });
 
         describe('shutdownCallback', () => {