fixup! fix: show org summary page if not entitled

code-asher · code-asher · commit 9ef3da706a75 · 2024-08-16T16:27:46.000-08:00
diff --git a/site/src/pages/ManagementSettingsPage/OrganizationSettingsPage.test.tsx b/site/src/pages/ManagementSettingsPage/OrganizationSettingsPage.test.tsx
@@ -2,6 +2,7 @@ import { screen, within } from "@testing-library/react";
 import { http, HttpResponse } from "msw";
 import {
 	MockDefaultOrganization,
+	MockEntitlementsWithMultiOrg,
 	MockOrganization2,
 } from "testHelpers/entities";
 import {
@@ -24,6 +25,9 @@ const renderPage = async () => {
 describe("OrganizationSettingsPage", () => {
 	it("has no editable organizations", async () => {
 		server.use(
+			http.get("/api/v2/entitlements", () => {
+				return HttpResponse.json(MockEntitlementsWithMultiOrg);
+			}),
 			http.get("/api/v2/organizations", () => {
 				return HttpResponse.json([MockDefaultOrganization, MockOrganization2]);
 			}),
@@ -39,6 +43,9 @@ describe("OrganizationSettingsPage", () => {
 
 	it("redirects to default organization", async () => {
 		server.use(
+			http.get("/api/v2/entitlements", () => {
+				return HttpResponse.json(MockEntitlementsWithMultiOrg);
+			}),
 			http.get("/api/v2/organizations", () => {
 				// Default always preferred regardless of order.
 				return HttpResponse.json([MockOrganization2, MockDefaultOrganization]);
@@ -60,6 +67,9 @@ describe("OrganizationSettingsPage", () => {
 
 	it("redirects to non-default organization", async () => {
 		server.use(
+			http.get("/api/v2/entitlements", () => {
+				return HttpResponse.json(MockEntitlementsWithMultiOrg);
+			}),
 			http.get("/api/v2/organizations", () => {
 				return HttpResponse.json([MockDefaultOrganization, MockOrganization2]);
 			}),
