DEV Community

Cover image for Mock Class Constructor in Jest Test with Mocking Partials
Kota Ito
Kota Ito

Posted on

2

Mock Class Constructor in Jest Test with Mocking Partials

I've been implementing tests on server-side code using Jest. One thing I initially struggled with was mocking class constructors. Here’s a guide on how to do it :)

Example Code

Here is the code for the function you want to test:

export const abc=(arg)=>{
return new ClassA(arg)
}
Enter fullscreen mode Exit fullscreen mode

In the test, you want to verify that the function abc returns a new instance of ClassA and that the ClassA constructor is called with arg. The first step is to mock the ClassA constructor.

How to mock the Class Constructor?

You can mock the Class Constructor and even its implementation with mock function from Jest .

Here is how you do it.

jest.mock('path for ClassA', () => ({
  ClassA: jest.fn().mockImplementation(() => ({ ClassA: 'dummyClassAResult' })),
}));
Enter fullscreen mode Exit fullscreen mode

In the code above, I am using a partial mock, meaning I’m only mocking part of the module where ClassA is defined.

The module is mocked so that ClassA is a Jest function. Additionally, I completely replace the implementation of the mock function to return an object { ClassA: 'dummyClassAResult' }.

Note that you cannot use mockReturnValue to mock the return value of a class constructor. Instead, you should use mockImplementation to achieve this.

Here is the incorrect approach:

jest.mock('ClassA module path', () => ({
  ClassA: jest.fn().mockReturnValue({ ClassA: 'dummyClassAResult' })
}));
Enter fullscreen mode Exit fullscreen mode

Using mockReturnValue does not work for class constructors because it is intended for mocking the return values of functions. When mocking class constructors, you need to replace the implementation of the constructor itself to ensure the class instance is created correctly.

Official Documentation for Jest mock

How to Keep Some Parts of a Mocked Module

If you mock a module as shown above, only ClassA is mocked. This means that other functions and objects exported from the module will not be mocked and will show as undefined when imported. So, what if you want to keep some of them intact? You can use the jest.requireActual function for such cases.

import { originalExportedObject, ClassA } from 'ClassA module path';

jest.mock('ClassA module path', () => {
  const actualModule = jest.requireActual('ClassA module path');
  return {
    ...actualModule,
    ClassA: jest.fn().mockImplementation(() => ({ ClassA: 'dummyClassAResult' })),
  };
});

Enter fullscreen mode Exit fullscreen mode

In the code above, jest.requireActual is used to extract the original exports from the module. This allows you to keep originalExportedObject and other parts of the module unchanged, while only ClassA is mocked.

Time to write a test

  //import ClassA
  import {ClassA} from 'ClassA module path'

  //mock the module
  jest.mock('ClassA module path', () => ({ 
  ClassA: jest.fn()
  .mockImplementation(() => ({ ClassA:'dummyClassAResult' })),
  }));

  //test
  it('Returning ', () => { 
      const arg='dummyArg'
      const result=abc(arg)

      expect(ClassA).toHaveBeenCalledWith(arg); //// Since ClassA is now a Jest function, you can use assertions on it!
      expect(result).toStrictEqual({ ClassA: 'dummyClassAResult' })
    });
Enter fullscreen mode Exit fullscreen mode

I hope this helps my fellow developers to write test on their code!

Image of AssemblyAI

Automatic Speech Recognition with AssemblyAI

Experience near-human accuracy, low-latency performance, and advanced Speech AI capabilities with AssemblyAI's Speech-to-Text API. Sign up today and get $50 in API credit. No credit card required.

Try the API

Top comments (0)

Image of Timescale

Timescale – the developer's data platform for modern apps, built on PostgreSQL

Timescale Cloud is PostgreSQL optimized for speed, scale, and performance. Over 3 million IoT, AI, crypto, and dev tool apps are powered by Timescale. Try it free today! No credit card required.

Try free

👋 Kindness is contagious

Explore a sea of insights with this enlightening post, highly esteemed within the nurturing DEV Community. Coders of all stripes are invited to participate and contribute to our shared knowledge.

Expressing gratitude with a simple "thank you" can make a big impact. Leave your thanks in the comments!

On DEV, exchanging ideas smooths our way and strengthens our community bonds. Found this useful? A quick note of thanks to the author can mean a lot.

Okay