From 22bc0cafd0fd6b431f94b31b44b8f9c3784bc269 Mon Sep 17 00:00:00 2001
From: Bikram Choudhury <bchoudhury@microsoft.com>
Date: Sun, 28 Dec 2025 14:32:11 +0530
Subject: [PATCH] Make DataExplorerRoot as an optional check

---
 test/fx.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/test/fx.ts b/test/fx.ts
index ac3724eae..0c4cfb36b 100644
--- a/test/fx.ts
+++ b/test/fx.ts
@@ -491,7 +491,7 @@ export class DataExplorer {
   }
 
   /** Waits for the Data Explorer app to load */
-  static async waitForExplorer(page: Page) {
+  static async waitForExplorer(page: Page, options?: TestExplorerUrlOptions): Promise<DataExplorer> {
     const iframeElement = await page.getByTestId("DataExplorerFrame").elementHandle();
     if (iframeElement === null) {
       throw new Error("Explorer iframe not found");
@@ -503,7 +503,9 @@ export class DataExplorer {
       throw new Error("Explorer frame not found");
     }
 
-    await explorerFrame?.getByTestId("DataExplorerRoot").waitFor();
+    if (!options?.enablecontainercopy) {
+      await explorerFrame?.getByTestId("DataExplorerRoot").waitFor();
+    }
 
     return new DataExplorer(explorerFrame);
   }
@@ -523,7 +525,7 @@ export class ContainerCopy {
   ) {}
 
   static async waitForContainerCopy(page: Page): Promise<ContainerCopy> {
-    const explorerFrame = await DataExplorer.waitForExplorer(page);
+    const explorerFrame = await DataExplorer.waitForExplorer(page, { enablecontainercopy: true });
     const containerCopyWrapper = explorerFrame.frame.locator("div#containerCopyWrapper");
     return new ContainerCopy(explorerFrame.frame, containerCopyWrapper);
   }