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 {