test: restore document spies before each test

Move document spies into beforeEach where needed so they are recreated
after restoreMocks resets them between tests.

This keeps the tests compatible with the shared test infrastructure
without changing their behavior.

Author: mahesh-09-12

js/__tests__/toolbar-ui.test.js

@@ -58,8 +58,6 @@ const createMockElement = id => ({
 });
 
 document.getElementById = jest.fn(createMockElement);
-jest.spyOn(document, "addEventListener").mockImplementation(() => {});
-jest.spyOn(document, "removeEventListener").mockImplementation(() => {});
 global.docById = id => document.getElementById(id);
 
 describe("ToolbarUI - Visual Helpers", () => {
@@ -169,6 +167,10 @@ describe("ToolbarUI - Visual Helpers", () => {
 });
 
 describe("FocusCycleManager - dispose", () => {
+    beforeEach(() => {
+        jest.spyOn(document, "addEventListener").mockImplementation(() => {});
+        jest.spyOn(document, "removeEventListener").mockImplementation(() => {});
+    });
     test("dispose removes all document-level event listeners", () => {
         const { FocusCycleManager } = require("../toolbar-ui");
         const manager = new FocusCycleManager();

js/widgets/__tests__/pitchstaircase.test.js

@@ -60,16 +60,6 @@ window.widgetWindows = {
     })
 };
 
-if (typeof document !== "undefined") {
-    jest.spyOn(document, "getElementsByClassName").mockImplementation(() => {
-        return [
-            {
-                style: {}
-            }
-        ];
-    });
-}
-
 describe("PitchStaircase Widget", () => {
     let psc;
 
@@ -285,6 +275,12 @@ describe("PitchStaircase Widget", () => {
                     dict: {}
                 }
             };
+
+            if (typeof document !== "undefined") {
+                jest.spyOn(document, "getElementsByClassName").mockImplementation(() => [
+                    { style: {} }
+                ]);
+            }
         });
 
         test("should set master volume to PREVIEWVOLUME and clear/show widget window on init", () => {