capricorn86#948@patch: Use macro task timing for postmessage api.

DylanPiercey · Jun 8, 2023 · e6d3b1a · e6d3b1a
1 parent f1987a9
commit e6d3b1a
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/packages/happy-dom/src/window/Window.ts b/packages/happy-dom/src/window/Window.ts
@@ -806,14 +806,22 @@ export default class Window extends EventTarget implements IWindow {
 			);
 		}
 
-		this.dispatchEvent(
-			new MessageEvent('message', {
-				data: message,
-				origin: this.parent.location.origin,
-				source: this.parent,
-				lastEventId: ''
-			})
-		);
+		const setImmediate =
+			globalThis.setImmediate || <typeof globalThis.setImmediate>(<unknown>setTimeout);
+		const clearImmediate =
+			globalThis.clearImmediate || <typeof globalThis.clearImmediate>(<unknown>clearTimeout);
+		const taskId = this.happyDOM.asyncTaskManager.startTask(() => clearImmediate(immediateId));
+		const immediateId = setImmediate(() => {
+			this.happyDOM.asyncTaskManager.endTask(taskId);
+			this.dispatchEvent(
+				new MessageEvent('message', {
+					data: message,
+					origin: this.parent.location.origin,
+					source: this.parent,
+					lastEventId: ''
+				})
+			);
+		});
 	}
 
 	/**