/* eslint-disable func-names */ import { expect } from 'chai'; import Sinon from 'sinon'; import { Data } from '../../../data/data'; import { OpenGroupData } from '../../../data/opengroups'; import { ConfigDumpData } from '../../../data/configDump/configDump'; import * as utilWorker from '../../../webworker/workers/browser/util_worker_interface'; import * as libsessionWorker from '../../../webworker/workers/browser/libsession_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; /** * Stub a function inside Data. * * Note: This uses a custom sandbox. * Please call `restoreStubs()` or `stub.restore()` to restore original functionality. */ export function stubData(fn: K): sinon.SinonStub { return Sinon.stub(Data, fn); } export function stubOpenGroupData(fn: K): sinon.SinonStub { return Sinon.stub(OpenGroupData, fn); } export function stubConfigDumpData(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 = Partial | undefined; /** * Stub a window object */ export function stubWindow(fn: K, value: WindowValue) { if (typeof globalAny.window === 'undefined') { globalAny.window = {}; } const set = (newValue: WindowValue) => { globalAny.window[fn] = newValue; }; const get = () => { return globalAny.window[fn] as WindowValue; }; 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, 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); } }