From 9a9b6d72ae4852d1b5160f4295bbb427c198d528 Mon Sep 17 00:00:00 2001
From: Christian Fehmer <fehmer@users.noreply.github.com>
Date: Fri, 12 Dec 2025 13:28:49 +0100
Subject: [PATCH] refactor: mock ElementWithUtils (@fehmer) (#7224)

---
 frontend/__tests__/setup-tests.ts | 68 ++++++++++++++++++++-----------
 1 file changed, 44 insertions(+), 24 deletions(-)

diff --git a/frontend/__tests__/setup-tests.ts b/frontend/__tests__/setup-tests.ts
index 43b2255ae..6828d2772 100644
--- a/frontend/__tests__/setup-tests.ts
+++ b/frontend/__tests__/setup-tests.ts
@@ -1,5 +1,7 @@
 import { vi } from "vitest";
 import $ from "jquery";
+import { ElementWithUtils } from "../src/ts/utils/dom";
+
 //@ts-expect-error add to globl
 global["$"] = $;
 //@ts-expect-error add to globl
@@ -19,33 +21,51 @@ vi.mock("../src/ts/firebase", () => ({
 }));
 
 vi.mock("../src/ts/utils/dom", () => {
-  const createMockElement = (): any => {
-    const mock = {
-      qsr: vi.fn(),
+  const createMockElement = (): ElementWithUtils => {
+    return {
+      disable: vi.fn().mockReturnThis(),
+      enable: vi.fn().mockReturnThis(),
+      isDisabled: vi.fn().mockReturnValue(false),
+      getAttribute: vi.fn(),
+      hasAttribute: vi.fn().mockReturnValue(false),
+      setAttribute: vi.fn().mockReturnThis(),
+      removeAttribute: vi.fn().mockReturnThis(),
+      isChecked: vi.fn().mockReturnValue(false),
+      hide: vi.fn().mockReturnThis(),
+      show: vi.fn().mockReturnThis(),
+      addClass: vi.fn().mockReturnThis(),
+      removeClass: vi.fn().mockReturnThis(),
+      hasClass: vi.fn().mockReturnValue(false),
+      toggleClass: vi.fn().mockReturnThis(),
+      on: vi.fn().mockReturnThis(),
+      onChild: vi.fn().mockReturnThis(),
+      setHtml: vi.fn().mockReturnThis(),
+      setText: vi.fn().mockReturnThis(),
+      remove: vi.fn(),
+      setStyle: vi.fn().mockReturnThis(),
+      getStyle: vi.fn().mockReturnValue({}),
+      isFocused: vi.fn().mockReturnValue(false),
       qs: vi.fn().mockReturnValue(null),
-      find: vi.fn(),
-      addClass: vi.fn(),
-      removeClass: vi.fn(),
-      hide: vi.fn(),
-      show: vi.fn(),
-      setText: vi.fn(),
-      prependHtml: vi.fn(),
-      empty: vi.fn(),
-      appendHtml: vi.fn(),
+      qsr: vi.fn().mockImplementation(() => createMockElement()),
+      qsa: vi.fn().mockReturnValue([]),
+      empty: vi.fn().mockReturnThis(),
+      appendHtml: vi.fn().mockReturnThis(),
+      append: vi.fn().mockReturnThis(),
+      prependHtml: vi.fn().mockReturnThis(),
+      dispatch: vi.fn().mockReturnThis(),
+      offset: vi.fn().mockReturnValue({ top: 0, left: 0 }),
+      wrapWith: vi.fn().mockImplementation(() => createMockElement()),
+      setValue: vi.fn().mockReturnThis(),
+      getValue: vi.fn().mockReturnValue(""),
+      getParent: vi.fn().mockImplementation(() => createMockElement()),
+      replaceWith: vi.fn().mockReturnThis(),
+      getOffsetWidth: vi.fn().mockReturnValue(0),
+      getOffsetHeight: vi.fn().mockReturnValue(0),
+      getOffsetTop: vi.fn().mockReturnValue(0),
+      getOffsetLeft: vi.fn().mockReturnValue(0),
+      animate: vi.fn().mockResolvedValue(null),
       native: document.createElement("div"),
     };
-
-    // Make chainable methods return the mock itself
-    mock.qsr.mockImplementation(() => createMockElement());
-    mock.addClass.mockReturnValue(mock);
-    mock.removeClass.mockReturnValue(mock);
-    mock.hide.mockReturnValue(mock);
-    mock.show.mockReturnValue(mock);
-    mock.setText.mockReturnValue(mock);
-    mock.prependHtml.mockReturnValue(mock);
-    mock.empty.mockReturnValue(mock);
-
-    return mock;
   };
 
   return {