From f6d651304c2e7ab76f4e853217b4bdb40f5315a7 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <balloob@gmail.com>
Date: Fri, 14 Jun 2019 15:09:42 -0700
Subject: [PATCH] Fix demo

---
 src/fake_data/provide_hass.ts | 50 ++++++++++++++++-------------------
 1 file changed, 23 insertions(+), 27 deletions(-)

diff --git a/src/fake_data/provide_hass.ts b/src/fake_data/provide_hass.ts
index 1d065688a0..3512ec9daa 100644
--- a/src/fake_data/provide_hass.ts
+++ b/src/fake_data/provide_hass.ts
@@ -94,10 +94,27 @@ export const provideHass = (
     connection: {
       addEventListener: () => undefined,
       removeEventListener: () => undefined,
-      sendMessagePromise: () =>
-        new Promise(() => {
-          /* we never resolve */
-        }),
+      sendMessage: (msg) => {
+        const callback = wsCommands[msg.type];
+
+        if (callback) {
+          callback(msg);
+        } else {
+          // tslint:disable-next-line
+          console.error(`Unknown WS command: ${msg.type}`);
+        }
+      },
+      sendMessagePromise: (msg) => {
+        const callback = wsCommands[msg.type];
+        return callback
+          ? callback(msg)
+          : Promise.reject({
+              code: "command_not_mocked",
+              message: `WS Command ${
+                msg.type
+              } is not implemented in provide_hass.`,
+            });
+      },
       subscribeEvents: async (
         // @ts-ignore
         callback,
@@ -166,29 +183,8 @@ export const provideHass = (
         : Promise.reject(`API Mock for ${path} is not implemented`);
     },
     fetchWithAuth: () => Promise.reject("Not implemented"),
-    async sendWS(msg) {
-      const callback = wsCommands[msg.type];
-
-      if (callback) {
-        callback(msg);
-      } else {
-        // tslint:disable-next-line
-        console.error(`Unknown WS command: ${msg.type}`);
-      }
-      // tslint:disable-next-line
-      console.log("sendWS", msg);
-    },
-    async callWS(msg) {
-      const callback = wsCommands[msg.type];
-      return callback
-        ? callback(msg)
-        : Promise.reject({
-            code: "command_not_mocked",
-            message: `WS Command ${
-              msg.type
-            } is not implemented in provide_hass.`,
-          });
-    },
+    sendWS: (msg) => hassObj.connection.sendMessage(msg),
+    callWS: (msg) => hassObj.connection.sendMessagePromise(msg),
 
     // Mock stuff
     mockEntities: entities,