diff --git a/src/fake_data/provide_hass.ts b/src/fake_data/provide_hass.ts
index 7fdec23d33..039ed4085e 100644
--- a/src/fake_data/provide_hass.ts
+++ b/src/fake_data/provide_hass.ts
@@ -114,6 +114,7 @@ export const provideHass = (
   );
 
   const localLanguage = getLocalLanguage();
+  const noop = () => undefined;
 
   const hassObj: MockHomeAssistant = {
     // Home Assistant properties
@@ -123,8 +124,8 @@ export const provideHass = (
       },
     } as any,
     connection: {
-      addEventListener: () => undefined,
-      removeEventListener: () => undefined,
+      addEventListener: noop,
+      removeEventListener: noop,
       sendMessage: (msg) => {
         const callback = wsCommands[msg.type];
 
@@ -168,6 +169,8 @@ export const provideHass = (
           );
         };
       },
+      suspendReconnectUntil: noop,
+      suspend: noop,
       socket: {
         readyState: WebSocket.OPEN,
       },
diff --git a/src/layouts/home-assistant.ts b/src/layouts/home-assistant.ts
index 69e6bd677e..7356291391 100644
--- a/src/layouts/home-assistant.ts
+++ b/src/layouts/home-assistant.ts
@@ -78,7 +78,7 @@ export class HomeAssistantAppEl extends HassElement {
 
     document.addEventListener(
       "visibilitychange",
-      () => this.__handleVisibilityChange(),
+      () => this._handleVisibilityChange(),
       false
     );
   }
@@ -148,7 +148,7 @@ export class HomeAssistantAppEl extends HassElement {
         : route.path.substr(1, dividerPos - 1);
   }
 
-  private __handleVisibilityChange() {
+  protected _handleVisibilityChange() {
     if (document.hidden) {
       // If the document is hidden, we will prevent reconnects until we are visible again
       this.hass!.connection.suspendReconnectUntil(