Codibre/jest-stub-super-constructor

let cleanups: Function[] = [];

beforeEach(() => {
    cleanups = [];
});

afterEach(() => {
    for (const cleanup of cleanups) {
        cleanup();
    }
    cleanups = [];
});

/**
 * Stub the super call inside a constructor
 * @param cls The class you want to stub the super on the constructor
 * @returns
 */
export function stubSuperConstructor<T>(cls: T) {
    const prototypeBackup = Object.getPrototypeOf(cls);
    if (!prototypeBackup || prototypeBackup.name === '') {
        throw new Error('Object has no super prototype');
    }

    const stub = jest.fn();
    Object.setPrototypeOf(cls, stub);
    cleanups.push(() => Object.setPrototypeOf(cls, prototypeBackup));

    return stub;
}