Fix test

kylecarbs · kylecarbs · commit 5c602b42855f · 2022-05-19T16:56:34.000Z
diff --git a/site/src/pages/CreateWorkspacePage/CreateWorkspacePage.test.tsx b/site/src/pages/CreateWorkspacePage/CreateWorkspacePage.test.tsx
@@ -1,4 +1,4 @@
-import { screen } from "@testing-library/react"
+import { screen, waitFor } from "@testing-library/react"
 import userEvent from "@testing-library/user-event"
 import React from "react"
 import { reach, StringSchema } from "yup"
@@ -42,7 +42,7 @@ describe("CreateWorkspacePage", () => {
     jest.spyOn(API, "createWorkspace").mockResolvedValueOnce(MockWorkspace)
     await fillForm({ name: "test" })
     // Check if the request was made
-    expect(API.createWorkspace).haveBeenCalledTimes(1)
+    await waitFor(() => expect(API.createWorkspace).haveBeenCalledTimes(1))
     // It is good to check if the API was called with the right parameters
     expect(API.createWorkspace).toHaveBeenCalledWith({ name: "test" })
   })
