check for unavailable entities

Author: joelhawksley

app/apis/home_assistant_api.rb

@@ -17,12 +17,13 @@ def prepare_response(response)
     response.filter do
       entity_ids.include?(_1["entity_id"]) ||
         _1.dig("attributes", "device_class") == "battery" ||
-        _1["entity_id"].include?("sensor.timeframe")
+        _1["entity_id"].include?("sensor.timeframe") ||
+        _1["state"] == "unavailable"
     end
   end
 
   def problems
-    data
+    timeframe_sensor_problems = data
       .select { _1[:entity_id].include?("sensor.timeframe") }
       .map do
         if _1[:state] == "on"
@@ -41,6 +42,17 @@ def problems
           }
         end
       end.compact
+
+    unavailable_entity_problems = data
+      .select { _1[:state] == "unavailable" }
+      .map do
+        {
+          icon: "triangle-exclamation",
+          message: "#{_1[:entity_id].split(".").last.humanize} unavailable"
+        }
+      end
+
+    timeframe_sensor_problems + unavailable_entity_problems
   end
 
   def now_playing

test/apis/home_assistant_api_test.rb

@@ -379,4 +379,13 @@ def test_problems_non_binary
       assert_equal(api.problems, [{icon: "print", message: "Ink low"}])
     end
   end
+
+  def test_problems_unavailable
+    data = [{entity_id: "sensor.foo_bar", state: "unavailable"}]
+
+    api = HomeAssistantApi.new({})
+    api.stub :data, data do
+      assert_equal(api.problems, [{icon: "triangle-exclamation", message: "Foo bar unavailable"}])
+    end
+  end
 end