fix test

f0ssel · f0ssel · commit 259c3966161c · 2022-09-01T23:16:07.000Z
diff --git a/site/src/pages/WorkspacePage/WorkspacePage.test.tsx b/site/src/pages/WorkspacePage/WorkspacePage.test.tsx
@@ -20,6 +20,7 @@ import {
   MockWorkspace,
   MockWorkspaceAgent,
   MockWorkspaceAgentDisconnected,
+  MockWorkspaceAgentConnecting,
   MockWorkspaceBuild,
   renderWithAuth,
 } from "../../testHelpers/renderHelpers"
@@ -205,10 +206,14 @@ describe("WorkspacePage", () => {
         DisplayAgentStatusLanguage[MockWorkspaceAgent.status],
       )
       expect(agent1Status.length).toEqual(4)
-      const agent2Status = await screen.findAllByText(
+      const agentDisconnected = await screen.findAllByText(
         DisplayAgentStatusLanguage[MockWorkspaceAgentDisconnected.status],
       )
-      expect(agent2Status.length).toEqual(2)
+      expect(agentDisconnected.length).toEqual(1)
+      const agentConnecting = await screen.findAllByText(
+        DisplayAgentStatusLanguage[MockWorkspaceAgentConnecting.status],
+      )
+      expect(agentConnecting.length).toEqual(1)
       expect(getTemplateMock).toBeCalled()
     })
   })
