jest-websocket-mock/src/act-compat.ts

Mock websockets and assert complex websocket interactions with Jest

/**
 * A simple compatibility method for react's "act".
 * If @testing-library/react is already installed, we just use
 * their implementation - it's complete and has useful warnings.
 * If @testing-library/react is *not* installed, then we just assume
 * that the user is not testing a react application, and use a noop instead.
 */

type Callback = () => Promise<void | undefined> | void | undefined;
type AsyncAct = (callback: Callback) => Promise<undefined>;
type SyncAct = (callback: Callback) => void;

let act: AsyncAct | SyncAct;

try {
  act = require("@testing-library/react").act;
} catch (_) {
  act = (callback: Function) => {
    callback();
  };
}

export default act;