How to test a unhandledRejection / uncaughtException handler with jest

前端 未结 2 2095
旧时难觅i
旧时难觅i 2021-02-14 05:36

I have handlers for unhandledRejections and uncaughtExceptions:

bin.js

[\'unhandledRejection\', \'uncaughtExce         


        
2条回答
  •  孤城傲影
    2021-02-14 06:13

    My test strategy is to install spy onto process.on() and logger.error methods using jest.spyOn(object, methodName). After doing this, these methods have no side effects. Then, you can test your code logic in an isolated environment.

    Besides, there are a few things to note:

    • You should spy the functions before require('./bin') statement. Because when you load the bin.js module, the code will be executed.
    • You should use jest.resetModules() in the beforeEach hook to resets the module registry - the cache of all required modules. Why? because require() caches its results. So, the first time a module is required, then its initialization code runs. After that, the cache just returns the value of module.exports without running the initialization code again. But we have two test cases, we want the code in module scope to be executed twice.

    Now, here is the example:

    bin.js:

    const logger = require('./logger');
    
    ['unhandledRejection', 'uncaughtException'].forEach((event) => {
      process.on(event, (err) => logger.error(err));
    });
    

    logger.js:

    const logger = console;
    
    module.exports = logger;
    

    bin.test.js:

    const logger = require('./logger');
    
    describe('52493145', () => {
      beforeEach(() => {
        jest.resetModules();
      });
      afterEach(() => {
        jest.restoreAllMocks();
      });
      test('catches unhandled rejections', () => {
        const error = new Error('mock error');
        jest.spyOn(process, 'on').mockImplementation((event, handler) => {
          if (event === 'unhandledRejection') {
            handler(error);
          }
        });
        jest.spyOn(logger, 'error').mockReturnValueOnce();
        require('./bin');
        expect(process.on).toBeCalledWith('unhandledRejection', expect.any(Function));
        expect(logger.error).toHaveBeenCalledWith(error);
      });
    
      test('catches uncaught exceptions', () => {
        const error = new Error('mock error');
        jest.spyOn(process, 'on').mockImplementation((event, handler) => {
          if (event === 'uncaughtException') {
            handler(error);
          }
        });
        jest.spyOn(logger, 'error').mockReturnValueOnce();
        require('./bin');
        expect(process.on).toBeCalledWith('uncaughtException', expect.any(Function));
        expect(logger.error).toHaveBeenCalledWith(error);
      });
    });
    

    unit test result:

     PASS  examples/52493145/bin.test.js
      52493145
        ✓ catches unhandled rejections (5 ms)
        ✓ catches uncaught exceptions (1 ms)
    
    -----------|---------|----------|---------|---------|-------------------
    File       | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s 
    -----------|---------|----------|---------|---------|-------------------
    All files  |     100 |      100 |     100 |     100 |                   
     bin.js    |     100 |      100 |     100 |     100 |                   
     logger.js |     100 |      100 |     100 |     100 |                   
    -----------|---------|----------|---------|---------|-------------------
    Test Suites: 1 passed, 1 total
    Tests:       2 passed, 2 total
    Snapshots:   0 total
    Time:        2.73 s, estimated 4 s
    

    source code: https://github.com/mrdulin/jest-v26-codelab/tree/main/examples/52493145

提交回复
热议问题