Don't display AP disconnect dialog on upgrade (#475)

pi-top · Aug 13, 2024 · cd7b226 · cd7b226
1 parent 7cc50b2
commit cd7b226
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/frontend/src/components/onboarding_app/App.tsx b/frontend/src/components/onboarding_app/App.tsx
@@ -43,6 +43,10 @@ export default () => {
     }
   };
 
+  const pathDisplaysApDisconnectDialog = (path: string) => {
+    return ![PageRoute.Upgrade.toString(), PageRoute.Restart.toString()].includes(path);
+  };
+
   return (
     <>
       <Switch>
@@ -184,7 +188,7 @@ export default () => {
         <Route component={ErrorPage} />
       </Switch>
       <BuildInformation info={buildInfo} className={styles.buildInfo} />
-      <HotspotDisconnectDialog enabled={useLocation().pathname !== PageRoute.Restart && enableDisconnectedFromApDialog} />
+      <HotspotDisconnectDialog enabled={pathDisplaysApDisconnectDialog(useLocation().pathname) && enableDisconnectedFromApDialog} />
     </>
   );
 };
diff --git a/frontend/src/components/onboarding_app/__tests__/App.test.tsx b/frontend/src/components/onboarding_app/__tests__/App.test.tsx
@@ -354,7 +354,6 @@ describe("App", () => {
 
     // on upgrade page
     await waitForUpgradePage();
-    await checkForDialog();
 
     // mock upgrade and go to registration page
     await upgrade();