fix: correct tests

SuperViz · Jan 17, 2024 · 6e9151d · 6e9151d
1 parent 9bf2772
commit 6e9151d
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/components/comments/html-pin-adapter/index.test.ts b/src/components/comments/html-pin-adapter/index.test.ts
@@ -96,10 +96,23 @@ describe('HTMLPinAdapter', () => {
       const removeSpy = jest.fn();
       const removeEventListenerSpy = jest.fn();
 
+      const getAttribute = jest
+        .fn()
+        .mockResolvedValue(Math.random() > 0.5 ? '' : 'data-wrapper-type');
+      const parentElement = {
+        remove: removeSpy,
+      };
+
       const wrappers = [...instance['divWrappers']].map(([entry, value]) => {
         return [
           entry,
-          { ...value, remove: removeSpy, removeEventListener: removeEventListenerSpy },
+          {
+            ...value,
+            remove: removeSpy,
+            removeEventListener: removeEventListenerSpy,
+            getAttribute,
+            parentElement,
+          },
         ];
       });
       instance['divWrappers'] = new Map(wrappers as [key: any, value: any][]);