test!

aslilac · aslilac · commit 132bc79efda2 · 2024-08-12T18:48:41.000Z
diff --git a/site/src/pages/TemplatePage/TemplateRedirectController.test.tsx b/site/src/pages/TemplatePage/TemplateRedirectController.test.tsx
@@ -0,0 +1,40 @@
+import { API } from "api/api";
+import * as M from "testHelpers/entities";
+import { TemplateRedirectController } from "./TemplateRedirectController";
+import { renderWithAuth } from "testHelpers/renderHelpers";
+import { waitFor } from "@testing-library/react";
+
+const renderTemplateRedirectController = (route: string) => {
+  return renderWithAuth(<TemplateRedirectController />, {
+    route,
+    path: "/templates/:organization?/:template",
+  });
+};
+
+it("redirects from multi-org to single-org", async () => {
+  const page = renderTemplateRedirectController(
+    `/templates/${M.MockTemplate.organization_name}/${M.MockTemplate.name}`,
+  );
+
+  await waitFor(() =>
+    expect(page.router.state.location.pathname).toEqual(
+      `/templates/${M.MockTemplate.name}`,
+    ),
+  );
+});
+
+it("redirects from single-org to multi-org", async () => {
+  jest
+    .spyOn(API, "getOrganizations")
+    .mockResolvedValueOnce([M.MockDefaultOrganization, M.MockOrganization2]);
+
+  const page = renderTemplateRedirectController(
+    `/templates/${M.MockTemplate.name}`,
+  );
+
+  await waitFor(() =>
+    expect(page.router.state.location.pathname).toEqual(
+      `/templates/${M.MockDefaultOrganization.name}/${M.MockTemplate.name}`,
+    ),
+  );
+});
