fix: Fix for checking valid entity id

Fix for checking valid entity id in the poll-state and current-state nodes
zachowj · Mar 8, 2019 · 4ee501e · 4ee501e
1 parent b46a4d5
commit 4ee501e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/nodes/current-state/current-state.js b/nodes/current-state/current-state.js
@@ -59,7 +59,7 @@ module.exports = function(RED) {
                 {},
                 await this.nodeConfig.server.homeAssistant.getStates(entity_id)
             );
-            if (!currentState)
+            if (!currentState.entity_id)
                 return logAndContinueEmpty(
                     `entity could not be found in cache for entity_id: ${entity_id}, sending empty payload`
                 );

diff --git a/nodes/poll-state/poll-state.js b/nodes/poll-state/poll-state.js
@@ -79,11 +79,14 @@ module.exports = function(RED) {
             if (!this.isConnected) return;
 
             try {
-                const pollState = await this.nodeConfig.server.homeAssistant.getStates(
-                    this.entityId
+                const pollState = this.utils.merge(
+                    {},
+                    await this.nodeConfig.server.homeAssistant.getStates(
+                        this.entityId
+                    )
                 );
 
-                if (!pollState) {
+                if (!pollState.entity_id) {
                     this.warn(
                         `could not find state with entity_id "${this.entityId}"`
                     );