diff --git a/apps/web-remix/app/components/pages/pipelines/__tests__/pipeline.test.tsx b/apps/web-remix/app/components/pages/pipelines/__tests__/pipeline.test.tsx
index 6b6818951..9f49b454a 100644
--- a/apps/web-remix/app/components/pages/pipelines/__tests__/pipeline.test.tsx
+++ b/apps/web-remix/app/components/pages/pipelines/__tests__/pipeline.test.tsx
@@ -1,60 +1,26 @@
 import React from "react";
 import { test, describe, expect } from "vitest";
+import { render, screen, waitFor, fireEvent, act } from "~/tests/render";
+import { ButtonHandle } from "~/tests/handles/Button.handle";
+import { InputHandle } from "~/tests/handles/Input.handle";
+import userEvent from "@testing-library/user-event";
+import { server } from "~/tests/server.mock";
 import {
   actionWithSession,
   RoutesProps,
   setupRoutes,
 } from "~/tests/setup.tests";
-import { render, screen, waitFor, fireEvent, act } from "~/tests/render";
-import { server } from "~/tests/server.mock";
+import { loader as editBlockLoader } from "../build/editBlock/loader.server";
+import { handlers as blockTypesHandlers } from "./blockTypes.handlers";
+import { action as buildAction } from "../build/action.server";
+import { loader as buildLoader } from "../build/loader.server";
+import { EditBlockPage } from "../build/editBlock/page";
+import { PipelineBuilder } from "../build/page";
 import { PipelinesPage } from "../list/page";
 import {
   handlers as pipelinesHandlers,
   updatedPipelineHandles,
 } from "./pipelines.handlers";
-import { handlers as blockTypesHandlers } from "./blockTypes.handlers";
-import { action as buildAction } from "../build/action.server";
-import { loader as buildLoader } from "../build/loader.server";
-import { PipelineBuilder } from "../build/page";
-import { EditBlockPage } from "../build/editBlock/page";
-import { loader as editBlockLoader } from "../build/editBlock/loader.server";
-import { vi } from "vitest";
-import userEvent from "@testing-library/user-event";
-import { ButtonHandle } from "~/tests/handles/Button.handle";
-import { InputHandle } from "~/tests/handles/Input.handle";
-
-global.ResizeObserver = vi.fn().mockImplementation(() => ({
-  observe: vi.fn(),
-  unobserve: vi.fn(),
-  disconnect: vi.fn(),
-}));
-
-class WebSocketMock {
-  url: string;
-  readyState: any;
-  static readonly CONNECTING = 0;
-  static readonly OPEN = 1;
-  static readonly CLOSING = 2;
-  static readonly CLOSED = 3;
-
-  constructor(url: string) {
-    console.log(`WebSocketMock created for url: ${url}`);
-    this.url = url;
-    this.readyState = WebSocket.OPEN;
-  }
-
-  send() {
-    console.log(`Mock send: `);
-  }
-
-  close() {
-    console.log(`Mock WebSocket closed`);
-    this.readyState = WebSocketMock.CLOSED;
-  }
-}
-
-// @ts-ignore
-global.WebSocket = WebSocketMock;
 
 describe(PipelinesPage.name, () => {
   const setupServer = server([...pipelinesHandlers, ...blockTypesHandlers]);
diff --git a/apps/web-remix/app/tests/WebSocket.mock.ts b/apps/web-remix/app/tests/WebSocket.mock.ts
new file mode 100644
index 000000000..abe2e96ec
--- /dev/null
+++ b/apps/web-remix/app/tests/WebSocket.mock.ts
@@ -0,0 +1,23 @@
+export class WebSocketMock {
+  url: string;
+  readyState: any;
+  static readonly CONNECTING = 0;
+  static readonly OPEN = 1;
+  static readonly CLOSING = 2;
+  static readonly CLOSED = 3;
+
+  constructor(url: string) {
+    console.log(`WebSocketMock created for url: ${url}`);
+    this.url = url;
+    this.readyState = WebSocket.OPEN;
+  }
+
+  send(data: unknown) {
+    console.log(`Mock send: ${JSON.stringify(data)}`);
+  }
+
+  close() {
+    console.log(`Mock WebSocket closed`);
+    this.readyState = WebSocketMock.CLOSED;
+  }
+}
diff --git a/apps/web-remix/app/tests/setupFiles.ts b/apps/web-remix/app/tests/setupFiles.ts
new file mode 100644
index 000000000..bcdd3eeb1
--- /dev/null
+++ b/apps/web-remix/app/tests/setupFiles.ts
@@ -0,0 +1,11 @@
+import { vi } from "vitest";
+import { WebSocketMock } from "./WebSocket.mock";
+
+global.ResizeObserver = vi.fn().mockImplementation(() => ({
+  observe: vi.fn(),
+  unobserve: vi.fn(),
+  disconnect: vi.fn(),
+}));
+
+// @ts-ignore
+global.WebSocket = WebSocketMock;
diff --git a/apps/web-remix/vitest.config.ts b/apps/web-remix/vitest.config.ts
index f8e956f36..3d963659a 100644
--- a/apps/web-remix/vitest.config.ts
+++ b/apps/web-remix/vitest.config.ts
@@ -6,6 +6,7 @@ dotenv.config({ path: ".env.test" });
 
 export default defineConfig({
   test: {
+    setupFiles: ["./app/tests/setupFiles.ts"],
     environment: "jsdom",
     globals: true,
     server: {