You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
session-desktop/ts/test/test-utils/utils/stubbing.ts

81 lines
2.3 KiB
TypeScript

import Sinon from 'sinon';
import * as DataShape from '../../../../ts/data/data';
import * as utilWorker from '../../../webworker/workers/util_worker_interface';
const globalAny: any = global;
// We have to do this in a weird way because Data uses module.exports
// which doesn't play well with sinon or ImportMock
// tslint:disable: no-require-imports no-var-requires
const Data = require('../../../../ts/data/data');
const DataItem = require('../../../../ts/data/channelsItem');
type DataFunction = typeof DataShape;
/**
* Stub a function inside Data.
*
* Note: This uses a custom sandbox.
* Please call `restoreStubs()` or `stub.restore()` to restore original functionality.
*/
export function stubData<K extends keyof DataFunction>(fn: K): sinon.SinonStub {
return Sinon.stub(Data, fn);
}
export function stubDataItem<K extends keyof DataFunction>(fn: K): sinon.SinonStub {
return Sinon.stub(DataItem, fn);
}
export function stubUtilWorker(fnName: string, returnedValue: any): sinon.SinonStub {
return Sinon.stub(utilWorker, 'callUtilsWorker')
.withArgs(fnName)
.resolves(returnedValue);
}
export function stubCreateObjectUrl() {
(global as any).URL = {};
(global as any).URL.createObjectURL = () => {
// tslint:disable-next-line: insecure-random
return `${Date.now()}:${Math.floor(Math.random() * 1000)}`;
};
}
type WindowValue<K extends keyof Window> = Partial<Window[K]> | undefined;
/**
* Stub a window object
*/
export function stubWindow<K extends keyof Window>(fn: K, value: WindowValue<K>) {
// tslint:disable-next-line: no-typeof-undefined
if (typeof globalAny.window === 'undefined') {
globalAny.window = {};
}
const set = (newValue: WindowValue<K>) => {
globalAny.window[fn] = newValue;
};
const get = () => {
return globalAny.window[fn] as WindowValue<K>;
};
globalAny.window[fn] = value;
return {
get,
set,
};
}
const enableLogRedirect = false;
export const stubWindowLog = () => {
stubWindow('log', {
// tslint:disable: no-void-expression
// tslint:disable: no-console
info: (args: any) => (enableLogRedirect ? console.info(args) : {}),
warn: (args: any) => (enableLogRedirect ? console.warn(args) : {}),
error: (args: any) => (enableLogRedirect ? console.error(args) : {}),
debug: (args: any) => (enableLogRedirect ? console.debug(args) : {}),
});
};