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.
111 lines
3.3 KiB
TypeScript
111 lines
3.3 KiB
TypeScript
/* eslint-disable func-names */
|
|
import { expect } from 'chai';
|
|
import Sinon from 'sinon';
|
|
import { ConfigDumpData } from '../../../data/configDump/configDump';
|
|
import { Data } from '../../../data/data';
|
|
import { OpenGroupData } from '../../../data/opengroups';
|
|
|
|
import * as libsessionWorker from '../../../webworker/workers/browser/libsession_worker_interface';
|
|
import * as utilWorker from '../../../webworker/workers/browser/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
|
|
|
|
type DataFunction = typeof Data;
|
|
type OpenGroupDataFunction = typeof OpenGroupData;
|
|
type ConfigDumpDataFunction = typeof ConfigDumpData;
|
|
|
|
export type TypedStub<T extends Record<string, unknown>, K extends keyof T> = T[K] extends (
|
|
...args: any
|
|
) => any
|
|
? Sinon.SinonStub<Parameters<T[K]>, ReturnType<T[K]>>
|
|
: never;
|
|
|
|
/**
|
|
* 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 stubOpenGroupData<K extends keyof OpenGroupDataFunction>(fn: K): sinon.SinonStub {
|
|
return Sinon.stub(OpenGroupData, fn);
|
|
}
|
|
|
|
export function stubConfigDumpData<K extends keyof ConfigDumpDataFunction>(fn: K): sinon.SinonStub {
|
|
return Sinon.stub(ConfigDumpData, fn);
|
|
}
|
|
|
|
export function stubUtilWorker(fnName: string, returnedValue: any): sinon.SinonStub {
|
|
return Sinon.stub(utilWorker, 'callUtilsWorker')
|
|
.withArgs(fnName as any)
|
|
.resolves(returnedValue);
|
|
}
|
|
|
|
export function stubLibSessionWorker(value: any) {
|
|
Sinon.stub(libsessionWorker, 'callLibSessionWorker').resolves(value);
|
|
}
|
|
|
|
export function stubCreateObjectUrl() {
|
|
(global as any).URL = function () {};
|
|
(global as any).URL.createObjectURL = () => {
|
|
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>) {
|
|
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,
|
|
};
|
|
}
|
|
|
|
export const enableLogRedirect = false;
|
|
|
|
export const stubWindowLog = () => {
|
|
stubWindow('log', {
|
|
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) : {}),
|
|
});
|
|
};
|
|
|
|
export const stubWindowFeatureFlags = () => {
|
|
stubWindow('sessionFeatureFlags', { debug: {} } as any);
|
|
};
|
|
|
|
export async function expectAsyncToThrow(toAwait: () => Promise<any>, errorMessageToCatch: string) {
|
|
try {
|
|
await toAwait();
|
|
throw new Error('fake_error');
|
|
} catch (e) {
|
|
expect(e.message).to.not.be.eq('fake_error');
|
|
expect(e.message).to.be.eq(errorMessageToCatch);
|
|
}
|
|
}
|