Restoring offsetHeight mock

WordPress · Jan 24, 2024 · e7f4d59 · e7f4d59
1 parent b2fac36
commit e7f4d59
Showing 1 changed file with 18 additions and 8 deletions.
diff --git a/packages/components/src/popover/test/index.tsx b/packages/components/src/popover/test/index.tsx
@@ -37,14 +37,24 @@ type PlacementToInitialTranslationTuple = [
 	CSSProperties[ 'translate' ],
 ];
 
-jest.spyOn( HTMLElement.prototype, 'offsetHeight', 'get' ).mockImplementation(
-	function getOffsetHeight( this: HTMLElement ) {
-		if ( this.tagName === 'BODY' ) {
-			return window.outerHeight;
-		}
-		return 50;
-	}
-);
+let offsetHeightSpy: jest.SpiedGetter<
+	typeof HTMLElement.prototype.offsetHeight
+>;
+
+beforeAll( () => {
+	offsetHeightSpy = jest
+		.spyOn( HTMLElement.prototype, 'offsetHeight', 'get' )
+		.mockImplementation( function getOffsetHeight( this: HTMLElement ) {
+			if ( this.tagName === 'BODY' ) {
+				return window.outerHeight;
+			}
+			return 50;
+		} );
+} );
+
+afterAll( () => {
+	offsetHeightSpy?.mockRestore();
+} );
 
 // There's no matching `placement` for 'middle center' positions,
 // fallback to 'bottom' (same as `floating-ui`'s default.)