mirror of https://github.com/vector-im/riot-web
97 lines
3.6 KiB
TypeScript
97 lines
3.6 KiB
TypeScript
/*
|
|
Copyright 2021 The Matrix.org Foundation C.I.C.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
import { ReactWrapper } from "enzyme";
|
|
import EventEmitter from "events";
|
|
|
|
import { ActionPayload } from "../../src/dispatcher/payloads";
|
|
import defaultDispatcher from "../../src/dispatcher/dispatcher";
|
|
import { DispatcherAction } from "../../src/dispatcher/actions";
|
|
|
|
export const emitPromise = (e: EventEmitter, k: string | symbol) => new Promise(r => e.once(k, r));
|
|
|
|
export function untilDispatch(waitForAction: DispatcherAction): Promise<ActionPayload> {
|
|
let dispatchHandle: string;
|
|
return new Promise<ActionPayload>(resolve => {
|
|
dispatchHandle = defaultDispatcher.register(payload => {
|
|
if (payload.action === waitForAction) {
|
|
defaultDispatcher.unregister(dispatchHandle);
|
|
resolve(payload);
|
|
}
|
|
});
|
|
});
|
|
}
|
|
|
|
const findByAttr = (attr: string) => (component: ReactWrapper, value: string) => component.find(`[${attr}="${value}"]`);
|
|
export const findByTestId = findByAttr('data-test-id');
|
|
export const findById = findByAttr('id');
|
|
export const findByAriaLabel = findByAttr('aria-label');
|
|
|
|
const findByTagAndAttr = (attr: string) =>
|
|
(component: ReactWrapper, value: string, tag: string) =>
|
|
component.find(`${tag}[${attr}="${value}"]`);
|
|
|
|
export const findByTagAndTestId = findByTagAndAttr('data-test-id');
|
|
|
|
export const flushPromises = async () => await new Promise(resolve => setTimeout(resolve));
|
|
|
|
// with jest's modern fake timers process.nextTick is also mocked,
|
|
// flushing promises in the normal way then waits for some advancement
|
|
// of the fake timers
|
|
// https://gist.github.com/apieceofbart/e6dea8d884d29cf88cdb54ef14ddbcc4?permalink_comment_id=4018174#gistcomment-4018174
|
|
export const flushPromisesWithFakeTimers = async (): Promise<void> => {
|
|
const promise = new Promise(resolve => process.nextTick(resolve));
|
|
jest.advanceTimersByTime(1);
|
|
await promise;
|
|
};
|
|
|
|
/**
|
|
* Call fn before calling componentDidUpdate on a react component instance, inst.
|
|
* @param {React.Component} inst an instance of a React component.
|
|
* @param {number} updates Number of updates to wait for. (Defaults to 1.)
|
|
* @returns {Promise} promise that resolves when componentDidUpdate is called on
|
|
* given component instance.
|
|
*/
|
|
export function waitForUpdate(inst: React.Component, updates = 1): Promise<void> {
|
|
return new Promise<void>((resolve, reject) => {
|
|
const cdu = inst.componentDidUpdate;
|
|
|
|
console.log(`Waiting for ${updates} update(s)`);
|
|
|
|
inst.componentDidUpdate = (prevProps, prevState, snapshot) => {
|
|
updates--;
|
|
console.log(`Got update, ${updates} remaining`);
|
|
|
|
if (updates == 0) {
|
|
inst.componentDidUpdate = cdu;
|
|
resolve();
|
|
}
|
|
|
|
if (cdu) cdu(prevProps, prevState, snapshot);
|
|
};
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Advance jests fake timers and Date.now mock by ms
|
|
* Useful for testing code using timeouts or intervals
|
|
* that also checks timestamps
|
|
*/
|
|
export const advanceDateAndTime = (ms: number) => {
|
|
jest.spyOn(global.Date, 'now').mockReturnValue(Date.now() + ms);
|
|
jest.advanceTimersByTime(ms);
|
|
};
|