Migrating menus test

thamara · Jan 25, 2024 · ace65ac · ace65ac
1 parent b09a153
commit ace65ac
Show file tree

Hide file tree

Showing 2 changed files with 165 additions and 158 deletions.
diff --git a/__mocks__/Mock.mjs b/__mocks__/Mock.mjs
@@ -57,7 +57,7 @@ class MockClass
     }
 
     /**
-     * Test related function to get mocking exported methods
+     * Restore a single mocked method
      * @param {string} methodName
      */
     restoreMock(methodName)
@@ -76,6 +76,20 @@ class MockClass
             this._mocked[methodName] = false;
         }
     }
+
+    /**
+     * Restore all mocked methods
+     */
+    restoreAll()
+    {
+        for (const [methodName, isMocked] of Object.entries(this._mocked))
+        {
+            if (isMocked)
+            {
+                this.restoreMock(methodName);
+            }
+        }
+    }
 }
 
 export {