mattermostest/src/main/CriticalErrorHandler.test.js

// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.
'use strict';

import {spawn} from 'child_process';

import path from 'path';

import {app, dialog} from 'electron';

import {CriticalErrorHandler} from './CriticalErrorHandler';

jest.mock('path', () => ({
    join: jest.fn().mockImplementation((...args) => args.join('/')),
}));

jest.mock('electron', () => ({
    app: {
        name: 'Mattermost',
        getVersion: () => '5.0.0',
        getPath: (folder) => `/${folder}`,
        relaunch: jest.fn(),
        isReady: jest.fn(),
        exit: jest.fn(),
    },
    dialog: {
        showMessageBox: jest.fn(),
    },
}));

jest.mock('fs', () => ({
    writeFileSync: jest.fn(),
}));

jest.mock('child_process', () => ({
    spawn: jest.fn(),
}));

describe('main/CriticalErrorHandler', () => {
    const criticalErrorHandler = new CriticalErrorHandler();
    beforeEach(() => {
        criticalErrorHandler.setMainWindow({});
    });

    describe('windowUnresponsiveHandler', () => {
        it('should do nothing when mainWindow is null', () => {
            criticalErrorHandler.setMainWindow(null);
            criticalErrorHandler.windowUnresponsiveHandler();
            expect(dialog.showMessageBox).not.toBeCalled();
        });

        it('should call app.relaunch when user elects not to wait', async () => {
            const promise = Promise.resolve({response: 0});
            dialog.showMessageBox.mockImplementation(() => promise);
            criticalErrorHandler.windowUnresponsiveHandler();
            await promise;
            expect(app.relaunch).toBeCalled();
        });
    });

    describe('processUncaughtExceptionHandler', () => {
        beforeEach(() => {
            app.isReady.mockImplementation(() => true);
            criticalErrorHandler.setMainWindow({isVisible: true});
        });

        it('should throw error if app is not ready', () => {
            app.isReady.mockImplementation(() => false);
            expect(() => {
                criticalErrorHandler.processUncaughtExceptionHandler(new Error('test'));
            }).toThrow(Error);
            expect(dialog.showMessageBox).not.toBeCalled();
        });

        it('should do nothing if main window is null or not visible', () => {
            criticalErrorHandler.setMainWindow(null);
            criticalErrorHandler.processUncaughtExceptionHandler(new Error('test'));
            expect(dialog.showMessageBox).not.toBeCalled();

            criticalErrorHandler.setMainWindow({isVisible: false});
            criticalErrorHandler.processUncaughtExceptionHandler(new Error('test'));
            expect(dialog.showMessageBox).not.toBeCalled();
        });

        it('should open external file on Show Details', async () => {
            path.join.mockImplementation(() => 'testfile.txt');
            const promise = Promise.resolve({response: process.platform === 'darwin' ? 2 : 0});
            dialog.showMessageBox.mockImplementation(() => promise);
            criticalErrorHandler.processUncaughtExceptionHandler(new Error('test'));
            await promise;
            expect(spawn).toBeCalledWith(expect.any(String), expect.arrayContaining(['testfile.txt']), expect.any(Object));
        });

        it('should restart app on Reopen', async () => {
            path.join.mockImplementation(() => 'testfile.txt');
            const promise = Promise.resolve({response: process.platform === 'darwin' ? 0 : 2});
            dialog.showMessageBox.mockImplementation(() => promise);
            criticalErrorHandler.processUncaughtExceptionHandler(new Error('test'));
            await promise;
            expect(app.relaunch).toBeCalled();
        });
    });
});