From fda6b59bc16717be91b7041ae4c20e5109f78f12 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 8dea7e9eb..967e88396 100644
--- a/test/fx.ts
+++ b/test/fx.ts
@@ -500,7 +500,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");
@@ -512,7 +512,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);
   }
@@ -532,7 +534,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);
   }