From 037da65d8373dc08c96b4bec809407dddfb41390 Mon Sep 17 00:00:00 2001
From: Pavel Feldman <pavel.feldman@gmail.com>
Date: Tue, 6 Jun 2023 14:24:42 -0700
Subject: [PATCH] chore: fix ui mode tests on windows (#23553)

---
 packages/trace-viewer/src/ui/uiModeView.tsx | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/packages/trace-viewer/src/ui/uiModeView.tsx b/packages/trace-viewer/src/ui/uiModeView.tsx
index 9201da88c9..25eda6cfbd 100644
--- a/packages/trace-viewer/src/ui/uiModeView.tsx
+++ b/packages/trace-viewer/src/ui/uiModeView.tsx
@@ -95,6 +95,7 @@ export const UIModeView: React.FC<{}> = ({
 
   React.useEffect(() => {
     inputRef.current?.focus();
+    setIsLoading(true);
     initWebSocket(() => setIsDisconnected(true)).then(() => reloadTests());
   }, [reloadTests]);
 
@@ -403,7 +404,7 @@ const TestList: React.FC<{
 
   // Update watch all.
   React.useEffect(() => {
-    if (!testModel.rootSuite)
+    if (isLoading)
       return;
     if (watchAll) {
       sendMessageNoReply('watch', { fileNames: [...fileNames] });
@@ -417,7 +418,7 @@ const TestList: React.FC<{
       }
       sendMessageNoReply('watch', { fileNames: [...fileNames] });
     }
-  }, [testModel, rootItem, fileNames, watchAll, watchedTreeIds, treeItemMap]);
+  }, [isLoading, rootItem, fileNames, watchAll, watchedTreeIds, treeItemMap]);
 
   const runTreeItem = (treeItem: TreeItem) => {
     setSelectedTreeItemId(treeItem.id);