Fix test

BrunoQuaresma · BrunoQuaresma · commit 3d7acf16a6b2 · 2022-06-03T12:49:52.000Z
diff --git a/site/src/pages/WorkspaceBuildPage/WorkspaceBuildPage.test.tsx b/site/src/pages/WorkspaceBuildPage/WorkspaceBuildPage.test.tsx
@@ -1,9 +1,21 @@
 import { screen } from "@testing-library/react"
+import * as API from "../../api/api"
 import { MockWorkspaceBuild, MockWorkspaceBuildLogs, renderWithAuth } from "../../testHelpers/renderHelpers"
 import { WorkspaceBuildPage } from "./WorkspaceBuildPage"
 
 describe("WorkspaceBuildPage", () => {
   it("renders the stats and logs", async () => {
+    jest.spyOn(API, "streamWorkspaceBuildLogs").mockResolvedValueOnce({
+      read() {
+        return Promise.resolve({
+          value: undefined,
+          done: true,
+        })
+      },
+      releaseLock: jest.fn(),
+      closed: Promise.resolve(undefined),
+      cancel: jest.fn(),
+    })
     renderWithAuth(<WorkspaceBuildPage />, { route: `/builds/${MockWorkspaceBuild.id}`, path: "/builds/:buildId" })
 
     await screen.findByText(MockWorkspaceBuild.workspace_name)
