From 9eff21f661aa2f7ddf02fc2cbc16bcbb60bf5c91 Mon Sep 17 00:00:00 2001
From: Miles Malerba <mmalerba@google.com>
Date: Thu, 9 May 2024 15:15:05 -0700
Subject: [PATCH] fix tests

---
 src/material/dialog/dialog.spec.ts | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/material/dialog/dialog.spec.ts b/src/material/dialog/dialog.spec.ts
index 85805e745ddd..efa1d782d0fa 100644
--- a/src/material/dialog/dialog.spec.ts
+++ b/src/material/dialog/dialog.spec.ts
@@ -1274,7 +1274,9 @@ describe('MDC-based MatDialog', () => {
       document.body.appendChild(button);
       button.focus();
 
-      let dialogRef = dialog.open(PizzaMsg, {viewContainerRef: testViewContainerRef});
+      const dialogRef = TestBed.inject(NgZone).run(() =>
+        dialog.open(PizzaMsg, {viewContainerRef: testViewContainerRef}),
+      );
 
       viewContainerFixture.detectChanges();
       flush();
@@ -1531,10 +1533,12 @@ describe('MDC-based MatDialog', () => {
       document.body.appendChild(button);
       button.focus();
 
-      const dialogRef = dialog.open(PizzaMsg, {
-        viewContainerRef: testViewContainerRef,
-        restoreFocus: false,
-      });
+      const dialogRef = TestBed.inject(NgZone).run(() =>
+        dialog.open(PizzaMsg, {
+          viewContainerRef: testViewContainerRef,
+          restoreFocus: false,
+        }),
+      );
 
       viewContainerFixture.detectChanges();
       flush();
@@ -1567,7 +1571,9 @@ describe('MDC-based MatDialog', () => {
       body.appendChild(otherButton);
       button.focus();
 
-      const dialogRef = dialog.open(PizzaMsg, {viewContainerRef: testViewContainerRef});
+      const dialogRef = TestBed.inject(NgZone).run(() =>
+        dialog.open(PizzaMsg, {viewContainerRef: testViewContainerRef}),
+      );
 
       viewContainerFixture.detectChanges();
       flush();