Enable AP disconnection checks on upgrade page

pi-top · Aug 20, 2024 · 5d8036d · 5d8036d
1 parent 9214fd4
commit 5d8036d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/frontend/src/components/onboarding_app/App.tsx b/frontend/src/components/onboarding_app/App.tsx
@@ -44,7 +44,7 @@ export default () => {
   };
 
   const pathDisplaysApDisconnectDialog = (path: string) => {
-    return ![PageRoute.Upgrade.toString(), PageRoute.Restart.toString()].includes(path);
+    return ![PageRoute.Restart.toString()].includes(path);
   };
 
   return (

diff --git a/frontend/src/components/onboarding_app/__tests__/App.test.tsx b/frontend/src/components/onboarding_app/__tests__/App.test.tsx
@@ -354,6 +354,7 @@ describe("App", () => {
 
     // on upgrade page
     await waitForUpgradePage();
+    await checkForDialog();
 
     // mock upgrade and go to registration page
     await upgrade();