Storybook Helpers (#35158)

app/javascript/testing/rendering.tsx

@@ -0,0 +1,52 @@
+import { IntlProvider } from 'react-intl';
+
+import { MemoryRouter } from 'react-router';
+
+import type { RenderOptions } from '@testing-library/react';
+import { render as rtlRender } from '@testing-library/react';
+
+import { IdentityContext } from '@/mastodon/identity_context';
+
+beforeAll(() => {
+  global.requestIdleCallback = vi.fn((cb: IdleRequestCallback) => {
+    // @ts-expect-error IdleRequestCallback expects an argument of type IdleDeadline,
+    // but that doesn't exist in this environment.
+    cb();
+    return 0;
+  });
+});
+
+function render(
+  ui: React.ReactElement,
+  {
+    locale = 'en',
+    signedIn = true,
+    ...renderOptions
+  }: RenderOptions & { locale?: string; signedIn?: boolean } = {},
+) {
+  const fakeIdentity = {
+    signedIn: signedIn,
+    accountId: '123',
+    disabledAccountId: undefined,
+    permissions: 0,
+  };
+
+  const Wrapper = (props: { children: React.ReactNode }) => {
+    return (
+      <MemoryRouter>
+        <IntlProvider locale={locale}>
+          <IdentityContext.Provider value={fakeIdentity}>
+            {props.children}
+          </IdentityContext.Provider>
+        </IntlProvider>
+      </MemoryRouter>
+    );
+  };
+  return rtlRender(ui, { wrapper: Wrapper, ...renderOptions });
+}
+
+// re-export everything
+export * from '@testing-library/react';
+
+// override render method
+export { render };