chore: move renderWithAuth to test_hepers/index.tsx

coder · BrunoQuaresma · Apr 15, 2022 · Apr 12, 2022 · Apr 12, 2022 · Apr 12, 2022
commit f3fedd0711de83ecd5747fdc99e3acc3ba2db1cf
@@ -3,7 +3,7 @@ import React from "react"
 import * as API from "../../api"
 import * as AccountForm from "../../components/Preferences/AccountForm"
 import { GlobalSnackbar } from "../../components/Snackbar/GlobalSnackbar"
-import { renderWithAuth } from "../../test_helpers/render"
+import { renderWithAuth } from "../../test_helpers"
 import * as AuthXService from "../../xServices/auth/authXService"
 import { PreferencesAccountPage } from "./account"
 

@@ -2,9 +2,11 @@ import ThemeProvider from "@material-ui/styles/ThemeProvider"
 import { render as wrappedRender, RenderResult } from "@testing-library/react"
 import { createMemoryHistory } from "history"
 import React from "react"
-import { unstable_HistoryRouter as HistoryRouter } from "react-router-dom"
+import { MemoryRouter, Route, Routes, unstable_HistoryRouter as HistoryRouter } from "react-router-dom"
+import { RequireAuth } from "../components"
 import { dark } from "../theme"
 import { XServiceProvider } from "../xServices/StateContext"
+import { MockUser } from "./entities"
 
 export const history = createMemoryHistory()
 
@@ -22,4 +24,23 @@ export const render = (component: React.ReactElement): RenderResult => {
   return wrappedRender(<WrapperComponent>{component}</WrapperComponent>)
 }
 
+type RenderWithAuthResult = RenderResult & { user: typeof MockUser }
+
+export function renderWithAuth(ui: JSX.Element, { route = "/" }: { route?: string } = {}): RenderWithAuthResult {
+  const renderResult = render(
+    <MemoryRouter initialEntries={[route]}>
+      <XServiceProvider>
+        <Routes>
+          <Route path={route} element={<RequireAuth>{ui}</RequireAuth>} />
+        </Routes>
+      </XServiceProvider>
+    </MemoryRouter>,
+  )
+
+  return {
+    user: MockUser,
+    ...renderResult,
+  }
+}
+
 export * from "./entities"
diff --git a/site/src/test_helpers/render.tsx b/site/src/test_helpers/render.tsx