refactor: clean up eventsource mock

jsjoeio · jsjoeio · commit a4769d8aba3a · 2022-09-29T20:38:27.000Z
diff --git a/site/src/pages/WorkspacePage/WorkspacePage.test.tsx b/site/src/pages/WorkspacePage/WorkspacePage.test.tsx
@@ -1,6 +1,6 @@
 import { fireEvent, screen, waitFor } from "@testing-library/react"
 import userEvent from "@testing-library/user-event"
-import EventSource from "eventsourcemock"
+import EventSourceMock from "eventsourcemock"
 import i18next from "i18next"
 import { rest } from "msw"
 import * as api from "../../api/api"
@@ -70,13 +70,20 @@ const testStatus = async (ws: Workspace, label: string) => {
   expect(status).toHaveTextContent(label)
 }
 
+let originalEventSource: typeof window.EventSource
+
+beforeAll(() => {
+  originalEventSource = window.EventSource
+  // mocking out EventSource for SSE
+  window.EventSource = EventSourceMock
+})
+
 beforeEach(() => {
   jest.resetAllMocks()
+})
 
-  // mocking out EventSource for SSE
-  Object.defineProperty(window, "EventSource", {
-    value: EventSource,
-  })
+afterAll(() => {
+  window.EventSource = originalEventSource
 })
 
 describe("WorkspacePage", () => {
