From 92a90eb9ae0818dbf4dac7fdbda94fa4e5a4a506 Mon Sep 17 00:00:00 2001
From: Bruno Quaresma <bruno@coder.com>
Date: Wed, 20 Sep 2023 15:19:27 -0300
Subject: [PATCH] test(site): fix terminal page test flakiness (#9798)

---
 site/src/pages/TerminalPage/TerminalPage.test.tsx | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/site/src/pages/TerminalPage/TerminalPage.test.tsx b/site/src/pages/TerminalPage/TerminalPage.test.tsx
index b5659e53b7..dae0cf5e85 100644
--- a/site/src/pages/TerminalPage/TerminalPage.test.tsx
+++ b/site/src/pages/TerminalPage/TerminalPage.test.tsx
@@ -58,7 +58,7 @@ const expectTerminalText = (container: HTMLElement, text: string) => {
       }
       expect(row.textContent).toContain(text);
     },
-    { timeout: 3_000 },
+    { timeout: 5_000 },
   );
 };
 
@@ -67,6 +67,9 @@ describe("TerminalPage", () => {
     const spy = jest
       .spyOn(API, "getWorkspaceByOwnerAndName")
       .mockResolvedValue(MockWorkspace);
+    const ws = new WS(
+      `ws://localhost/api/v2/workspaceagents/${MockWorkspaceAgent.id}/pty`,
+    );
     await renderTerminal(
       `/${MockUser.username}/${MockWorkspace.name}/terminal`,
     );
@@ -77,6 +80,7 @@ describe("TerminalPage", () => {
       );
     });
     spy.mockRestore();
+    ws.close();
   });
 
   it("shows an error if fetching workspace fails", async () => {