SymphonyElectron/spec/downloadHandler.spec.ts

jest.mock('electron-log');

jest.mock('../src/app/window-handler', () => {
    return {
        windowHandler: {
            setIsAutoReload: jest.fn(() => true),
        },
    };
});

jest.mock('../src/app/window-utils', () => {
    return {
        windowExists: jest.fn(() => true),
    };
});

describe('download handler', () => {
    let downloadHandlerInstance;

    beforeEach(() => {
        jest.resetModules();
        // I did it for reset module imported between tests
        const { downloadHandler } = require('../src/app/download-handler');
        downloadHandlerInstance = downloadHandler;
    });

    afterAll((done) => {
        done();
    });

    it('should call `sendDownloadCompleted` when download succeeds', () => {
        const spy: jest.SpyInstance = jest.spyOn(downloadHandlerInstance, 'sendDownloadCompleted')
            .mockImplementation(() => jest.fn());

        const data: any = {
            _id: '121312-123912321-1231231',
            savedPath: '/abc/def/123.txt',
            total: '1234556',
            fileName: 'Test.txt',
        };

        downloadHandlerInstance.onDownloadSuccess(data);
        expect(spy).toBeCalled();
    });

    it('should call `sendDownloadFailed` when download fails', () => {
        const spy: jest.SpyInstance = jest.spyOn(downloadHandlerInstance, 'sendDownloadFailed')
            .mockImplementation(() => jest.fn());

        downloadHandlerInstance.onDownloadFailed();
        expect(spy).toBeCalled();
    });

});