Add test to check if the navbar displays or not the admin menu

BrunoQuaresma · BrunoQuaresma · commit 810c1214c3a0 · 2022-05-12T15:30:28.000Z
diff --git a/site/src/components/Navbar/Navbar.test.tsx b/site/src/components/Navbar/Navbar.test.tsx
@@ -0,0 +1,40 @@
+import { screen, waitFor } from "@testing-library/react"
+import React from "react"
+import * as API from "../../api"
+import { renderWithAuth } from "../../testHelpers"
+import { checks } from "../../xServices/auth/authXService"
+import { Language as AdminDropdownLanguage } from "../AdminDropdown/AdminDropdown"
+import { Navbar } from "./Navbar"
+
+beforeEach(() => {
+  jest.resetAllMocks()
+})
+
+describe("Navbar", () => {
+  describe("when user has permission to read all users", () => {
+    it("displays the admin menu", async () => {
+      const checkUserPermissionsSpy = jest.spyOn(API, "checkUserPermissions").mockResolvedValueOnce({
+        [checks.readAllUsers]: true,
+      })
+
+      renderWithAuth(<Navbar />)
+
+      // Wait for the request is done
+      await waitFor(() => expect(checkUserPermissionsSpy).toBeCalledTimes(1))
+      await screen.findByRole("button", { name: AdminDropdownLanguage.menuTitle })
+    })
+  })
+
+  describe("when user has NO permission to read all users", () => {
+    it("does not display the admin menu", async () => {
+      const checkUserPermissionsSpy = jest.spyOn(API, "checkUserPermissions").mockResolvedValueOnce({
+        [checks.readAllUsers]: false,
+      })
+      renderWithAuth(<Navbar />)
+
+      // Wait for the request is done
+      await waitFor(() => expect(checkUserPermissionsSpy).toBeCalledTimes(1))
+      expect(screen.queryByRole("button", { name: AdminDropdownLanguage.menuTitle })).not.toBeInTheDocument()
+    })
+  })
+})
diff --git a/site/src/testHelpers/handlers.ts b/site/src/testHelpers/handlers.ts
@@ -1,4 +1,5 @@
 import { rest } from "msw"
+import { permissionsToCheck } from "../xServices/auth/authXService"
 import * as M from "./entities"
 
 export const handlers = [
@@ -54,6 +55,17 @@ export const handlers = [
   rest.get("/api/v2/users/roles", async (req, res, ctx) => {
     return res(ctx.status(200), ctx.json(M.MockSiteRoles))
   }),
+  rest.post("/api/v2/users/:userId/permissions/check", async (req, res, ctx) => {
+    const permissions = Object.keys(permissionsToCheck)
+    const response = permissions.reduce((obj, permission) => {
+      return {
+        ...obj,
+        [permission]: true,
+      }
+    }, {})
+
+    return res(ctx.status(200), ctx.json(response))
+  }),
 
   // workspaces
   rest.get("/api/v2/organizations/:organizationId/workspaces/:userName/:workspaceName", (req, res, ctx) => {
diff --git a/site/src/xServices/auth/authXService.ts b/site/src/xServices/auth/authXService.ts
@@ -8,8 +8,12 @@ export const Language = {
   successProfileUpdate: "Updated preferences.",
 }
 
-const permissionsToCheck = {
-  readAllUsers: {
+export const checks = {
+  readAllUsers: "readAllUsers",
+} as const
+
+export const permissionsToCheck = {
+  [checks.readAllUsers]: {
     object: {
       resource_type: "user",
     },
