2021-09-25 02:03:03 +00:00
|
|
|
import { dialog, BrowserWindow, WebContents } from 'electron';
|
2021-06-02 19:18:32 +00:00
|
|
|
jest.mock('loglevel');
|
|
|
|
import { error } from 'loglevel';
|
2021-06-26 13:59:28 +00:00
|
|
|
import { WindowOptions } from '../../../shared/src/options/model';
|
2021-06-02 19:18:32 +00:00
|
|
|
|
|
|
|
jest.mock('./helpers');
|
2021-06-09 13:49:35 +00:00
|
|
|
import { getCSSToInject } from './helpers';
|
2021-06-02 19:18:32 +00:00
|
|
|
jest.mock('./windowEvents');
|
|
|
|
import { clearAppData, createNewTab, injectCSS } from './windowHelpers';
|
|
|
|
|
|
|
|
describe('clearAppData', () => {
|
|
|
|
let window: BrowserWindow;
|
|
|
|
let mockClearCache: jest.SpyInstance;
|
|
|
|
let mockClearStorageData: jest.SpyInstance;
|
|
|
|
const mockShowDialog: jest.SpyInstance = jest.spyOn(dialog, 'showMessageBox');
|
|
|
|
|
|
|
|
beforeEach(() => {
|
|
|
|
window = new BrowserWindow();
|
|
|
|
mockClearCache = jest.spyOn(window.webContents.session, 'clearCache');
|
|
|
|
mockClearStorageData = jest.spyOn(
|
|
|
|
window.webContents.session,
|
|
|
|
'clearStorageData',
|
|
|
|
);
|
|
|
|
mockShowDialog.mockReset().mockResolvedValue(undefined);
|
|
|
|
});
|
|
|
|
|
|
|
|
afterAll(() => {
|
|
|
|
mockClearCache.mockRestore();
|
|
|
|
mockClearStorageData.mockRestore();
|
|
|
|
mockShowDialog.mockRestore();
|
|
|
|
});
|
|
|
|
|
|
|
|
test('will not clear app data if dialog canceled', async () => {
|
|
|
|
mockShowDialog.mockResolvedValue(1);
|
|
|
|
|
|
|
|
await clearAppData(window);
|
|
|
|
|
|
|
|
expect(mockShowDialog).toHaveBeenCalledTimes(1);
|
|
|
|
expect(mockClearCache).not.toHaveBeenCalled();
|
|
|
|
expect(mockClearStorageData).not.toHaveBeenCalled();
|
|
|
|
});
|
|
|
|
|
|
|
|
test('will clear app data if ok is clicked', async () => {
|
|
|
|
mockShowDialog.mockResolvedValue(0);
|
|
|
|
|
|
|
|
await clearAppData(window);
|
|
|
|
|
|
|
|
expect(mockShowDialog).toHaveBeenCalledTimes(1);
|
|
|
|
expect(mockClearCache).not.toHaveBeenCalledTimes(1);
|
|
|
|
expect(mockClearStorageData).not.toHaveBeenCalledTimes(1);
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
describe('createNewTab', () => {
|
|
|
|
const window = new BrowserWindow();
|
2021-06-26 13:59:28 +00:00
|
|
|
const options: WindowOptions = {
|
|
|
|
blockExternalUrls: false,
|
|
|
|
insecure: false,
|
|
|
|
name: 'Test App',
|
|
|
|
targetUrl: 'https://github.com/nativefier/natifefier',
|
|
|
|
zoom: 1.0,
|
|
|
|
};
|
2021-06-02 19:18:32 +00:00
|
|
|
const setupWindow = jest.fn();
|
|
|
|
const url = 'https://github.com/nativefier/nativefier';
|
|
|
|
const mockAddTabbedWindow: jest.SpyInstance = jest.spyOn(
|
|
|
|
BrowserWindow.prototype,
|
|
|
|
'addTabbedWindow',
|
|
|
|
);
|
|
|
|
const mockFocus: jest.SpyInstance = jest.spyOn(
|
|
|
|
BrowserWindow.prototype,
|
|
|
|
'focus',
|
|
|
|
);
|
|
|
|
const mockLoadURL: jest.SpyInstance = jest.spyOn(
|
|
|
|
BrowserWindow.prototype,
|
|
|
|
'loadURL',
|
|
|
|
);
|
|
|
|
|
|
|
|
test('creates new foreground tab', () => {
|
|
|
|
const foreground = true;
|
|
|
|
|
|
|
|
const tab = createNewTab(options, setupWindow, url, foreground, window);
|
|
|
|
|
|
|
|
expect(mockAddTabbedWindow).toHaveBeenCalledWith(tab);
|
|
|
|
expect(setupWindow).toHaveBeenCalledWith(options, tab);
|
|
|
|
expect(mockLoadURL).toHaveBeenCalledWith(url);
|
|
|
|
expect(mockFocus).not.toHaveBeenCalled();
|
|
|
|
});
|
|
|
|
|
|
|
|
test('creates new background tab', () => {
|
|
|
|
const foreground = false;
|
|
|
|
|
|
|
|
const tab = createNewTab(options, setupWindow, url, foreground, window);
|
|
|
|
|
|
|
|
expect(mockAddTabbedWindow).toHaveBeenCalledWith(tab);
|
|
|
|
expect(setupWindow).toHaveBeenCalledWith(options, tab);
|
|
|
|
expect(mockLoadURL).toHaveBeenCalledWith(url);
|
|
|
|
expect(mockFocus).toHaveBeenCalledTimes(1);
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
describe('injectCSS', () => {
|
2021-06-15 13:02:57 +00:00
|
|
|
jest.setTimeout(10000);
|
|
|
|
|
2021-06-02 19:18:32 +00:00
|
|
|
const mockGetCSSToInject: jest.SpyInstance = getCSSToInject as jest.Mock;
|
2021-06-26 13:59:28 +00:00
|
|
|
let mockGetURL: jest.SpyInstance;
|
2021-06-02 19:18:32 +00:00
|
|
|
const mockLogError: jest.SpyInstance = error as jest.Mock;
|
|
|
|
const mockWebContentsInsertCSS: jest.SpyInstance = jest.spyOn(
|
|
|
|
WebContents.prototype,
|
|
|
|
'insertCSS',
|
|
|
|
);
|
|
|
|
|
|
|
|
const css = 'body { color: white; }';
|
2021-06-26 13:59:28 +00:00
|
|
|
let responseHeaders: Record<string, string[]>;
|
2021-06-02 19:18:32 +00:00
|
|
|
|
|
|
|
beforeEach(() => {
|
|
|
|
mockGetCSSToInject.mockReset().mockReturnValue('');
|
2021-06-26 13:59:28 +00:00
|
|
|
mockGetURL = jest
|
|
|
|
.spyOn(WebContents.prototype, 'getURL')
|
|
|
|
.mockReturnValue('https://example.com');
|
2021-06-02 19:18:32 +00:00
|
|
|
mockLogError.mockReset();
|
|
|
|
mockWebContentsInsertCSS.mockReset().mockResolvedValue(undefined);
|
2021-06-26 13:59:28 +00:00
|
|
|
responseHeaders = { 'x-header': ['value'], 'content-type': ['test/other'] };
|
2021-06-02 19:18:32 +00:00
|
|
|
});
|
|
|
|
|
|
|
|
afterAll(() => {
|
|
|
|
mockGetCSSToInject.mockRestore();
|
2021-06-26 13:59:28 +00:00
|
|
|
mockGetURL.mockRestore();
|
2021-06-02 19:18:32 +00:00
|
|
|
mockLogError.mockRestore();
|
|
|
|
mockWebContentsInsertCSS.mockRestore();
|
|
|
|
});
|
|
|
|
|
2021-06-09 13:49:35 +00:00
|
|
|
test('will not inject if getCSSToInject is empty', () => {
|
2021-06-02 19:18:32 +00:00
|
|
|
const window = new BrowserWindow();
|
|
|
|
|
|
|
|
injectCSS(window);
|
|
|
|
|
2021-06-09 13:49:35 +00:00
|
|
|
expect(mockGetCSSToInject).toHaveBeenCalled();
|
2021-06-02 19:18:32 +00:00
|
|
|
expect(mockWebContentsInsertCSS).not.toHaveBeenCalled();
|
|
|
|
});
|
|
|
|
|
2021-09-25 02:03:03 +00:00
|
|
|
test('will inject on did-navigate + onResponseStarted', () => {
|
2021-06-02 19:18:32 +00:00
|
|
|
mockGetCSSToInject.mockReturnValue(css);
|
|
|
|
const window = new BrowserWindow();
|
|
|
|
|
|
|
|
injectCSS(window);
|
|
|
|
|
|
|
|
expect(mockGetCSSToInject).toHaveBeenCalled();
|
|
|
|
|
|
|
|
window.webContents.emit('did-navigate');
|
2021-06-16 02:20:49 +00:00
|
|
|
// @ts-expect-error this function doesn't exist in the actual electron version, but will in our mock
|
2021-06-26 13:59:28 +00:00
|
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-call
|
2021-09-25 02:03:03 +00:00
|
|
|
window.webContents.session.webRequest.send('onResponseStarted', {
|
|
|
|
responseHeaders,
|
|
|
|
webContents: window.webContents,
|
|
|
|
});
|
2021-06-02 19:18:32 +00:00
|
|
|
|
2021-09-25 02:03:03 +00:00
|
|
|
expect(mockWebContentsInsertCSS).toHaveBeenCalledWith(css);
|
2021-06-02 19:18:32 +00:00
|
|
|
});
|
2021-06-09 13:49:35 +00:00
|
|
|
|
2021-09-25 02:03:03 +00:00
|
|
|
test.each<string>(['application/json', 'font/woff2', 'image/png'])(
|
2021-06-15 13:02:57 +00:00
|
|
|
'will not inject for content-type %s',
|
2021-09-25 02:03:03 +00:00
|
|
|
(contentType: string) => {
|
2021-06-09 13:49:35 +00:00
|
|
|
mockGetCSSToInject.mockReturnValue(css);
|
|
|
|
const window = new BrowserWindow();
|
|
|
|
|
2021-06-15 13:02:57 +00:00
|
|
|
responseHeaders['content-type'] = [contentType];
|
|
|
|
|
2021-06-09 13:49:35 +00:00
|
|
|
injectCSS(window);
|
|
|
|
|
|
|
|
expect(mockGetCSSToInject).toHaveBeenCalled();
|
|
|
|
|
2021-06-15 13:02:57 +00:00
|
|
|
expect(window.webContents.emit('did-navigate')).toBe(true);
|
|
|
|
mockWebContentsInsertCSS.mockReset().mockResolvedValue(undefined);
|
2021-06-16 02:20:49 +00:00
|
|
|
// @ts-expect-error this function doesn't exist in the actual electron version, but will in our mock
|
2021-06-26 13:59:28 +00:00
|
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-call
|
2021-09-25 02:03:03 +00:00
|
|
|
window.webContents.session.webRequest.send('onResponseStarted', {
|
|
|
|
responseHeaders,
|
|
|
|
webContents: window.webContents,
|
|
|
|
url: `test-${contentType}`,
|
|
|
|
});
|
|
|
|
// insertCSS will still run once for the did-navigate
|
|
|
|
expect(mockWebContentsInsertCSS).not.toHaveBeenCalled();
|
2021-06-09 13:49:35 +00:00
|
|
|
},
|
|
|
|
);
|
|
|
|
|
2021-09-25 02:03:03 +00:00
|
|
|
test.each<string>(['text/html'])(
|
2021-06-15 13:02:57 +00:00
|
|
|
'will inject for content-type %s',
|
2021-09-25 02:03:03 +00:00
|
|
|
(contentType: string) => {
|
2021-06-09 13:49:35 +00:00
|
|
|
mockGetCSSToInject.mockReturnValue(css);
|
|
|
|
const window = new BrowserWindow();
|
|
|
|
|
2021-06-15 13:02:57 +00:00
|
|
|
responseHeaders['content-type'] = [contentType];
|
|
|
|
|
2021-06-09 13:49:35 +00:00
|
|
|
injectCSS(window);
|
|
|
|
|
|
|
|
expect(mockGetCSSToInject).toHaveBeenCalled();
|
|
|
|
|
|
|
|
window.webContents.emit('did-navigate');
|
2021-06-15 13:02:57 +00:00
|
|
|
mockWebContentsInsertCSS.mockReset().mockResolvedValue(undefined);
|
2021-06-16 02:20:49 +00:00
|
|
|
// @ts-expect-error this function doesn't exist in the actual electron version, but will in our mock
|
2021-06-26 13:59:28 +00:00
|
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-call
|
2021-09-25 02:03:03 +00:00
|
|
|
window.webContents.session.webRequest.send('onResponseStarted', {
|
|
|
|
responseHeaders,
|
|
|
|
webContents: window.webContents,
|
|
|
|
url: `test-${contentType}`,
|
|
|
|
});
|
|
|
|
|
|
|
|
expect(mockWebContentsInsertCSS).toHaveBeenCalledTimes(1);
|
2021-06-09 13:49:35 +00:00
|
|
|
},
|
|
|
|
);
|
|
|
|
|
2021-09-25 02:03:03 +00:00
|
|
|
test.each<string>(['image', 'script', 'stylesheet', 'xhr'])(
|
2021-06-09 13:49:35 +00:00
|
|
|
'will not inject for resource type %s',
|
2021-09-25 02:03:03 +00:00
|
|
|
(resourceType: string) => {
|
2021-06-09 13:49:35 +00:00
|
|
|
mockGetCSSToInject.mockReturnValue(css);
|
|
|
|
const window = new BrowserWindow();
|
|
|
|
|
|
|
|
injectCSS(window);
|
|
|
|
|
|
|
|
expect(mockGetCSSToInject).toHaveBeenCalled();
|
|
|
|
|
|
|
|
window.webContents.emit('did-navigate');
|
2021-06-15 13:02:57 +00:00
|
|
|
mockWebContentsInsertCSS.mockReset().mockResolvedValue(undefined);
|
2021-06-16 02:20:49 +00:00
|
|
|
// @ts-expect-error this function doesn't exist in the actual electron version, but will in our mock
|
2021-06-26 13:59:28 +00:00
|
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-call
|
2021-09-25 02:03:03 +00:00
|
|
|
window.webContents.session.webRequest.send('onResponseStarted', {
|
|
|
|
responseHeaders,
|
|
|
|
webContents: window.webContents,
|
|
|
|
resourceType,
|
|
|
|
url: `test-${resourceType}`,
|
|
|
|
});
|
|
|
|
// insertCSS will still run once for the did-navigate
|
|
|
|
expect(mockWebContentsInsertCSS).not.toHaveBeenCalled();
|
2021-06-09 13:49:35 +00:00
|
|
|
},
|
|
|
|
);
|
|
|
|
|
2021-09-25 02:03:03 +00:00
|
|
|
test.each<string>(['html', 'other'])(
|
2021-06-09 13:49:35 +00:00
|
|
|
'will inject for resource type %s',
|
2021-09-25 02:03:03 +00:00
|
|
|
(resourceType: string) => {
|
2021-06-09 13:49:35 +00:00
|
|
|
mockGetCSSToInject.mockReturnValue(css);
|
|
|
|
const window = new BrowserWindow();
|
|
|
|
|
|
|
|
injectCSS(window);
|
|
|
|
|
|
|
|
expect(mockGetCSSToInject).toHaveBeenCalled();
|
|
|
|
|
|
|
|
window.webContents.emit('did-navigate');
|
2021-06-15 13:02:57 +00:00
|
|
|
mockWebContentsInsertCSS.mockReset().mockResolvedValue(undefined);
|
2021-06-16 02:20:49 +00:00
|
|
|
// @ts-expect-error this function doesn't exist in the actual electron version, but will in our mock
|
2021-06-26 13:59:28 +00:00
|
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-call
|
2021-09-25 02:03:03 +00:00
|
|
|
window.webContents.session.webRequest.send('onResponseStarted', {
|
|
|
|
responseHeaders,
|
|
|
|
webContents: window.webContents,
|
|
|
|
resourceType,
|
|
|
|
url: `test-${resourceType}`,
|
|
|
|
});
|
|
|
|
expect(mockWebContentsInsertCSS).toHaveBeenCalledTimes(1);
|
2021-06-09 13:49:35 +00:00
|
|
|
},
|
|
|
|
);
|
2021-06-02 19:18:32 +00:00
|
|
|
});
|