Fix template tests

home-assistant · Nov 24, 2021 · dfbace6 · dfbace6
1 parent 0258eb3
commit dfbace6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/components/template/test_binary_sensor.py b/tests/components/template/test_binary_sensor.py
@@ -13,6 +13,7 @@
     STATE_OFF,
     STATE_ON,
     STATE_UNAVAILABLE,
+    STATE_UNKNOWN,
 )
 from homeassistant.core import Context, CoreState
 from homeassistant.helpers import entity_registry
@@ -498,10 +499,10 @@ async def test_no_update_template_match_all(hass, caplog):
     hass.states.async_set("binary_sensor.test_sensor", "true")
     assert len(hass.states.async_all()) == 5
 
-    assert hass.states.get("binary_sensor.all_state").state == OFF
-    assert hass.states.get("binary_sensor.all_icon").state == OFF
-    assert hass.states.get("binary_sensor.all_entity_picture").state == OFF
-    assert hass.states.get("binary_sensor.all_attribute").state == OFF
+    assert hass.states.get("binary_sensor.all_state").state == STATE_UNKNOWN
+    assert hass.states.get("binary_sensor.all_icon").state == STATE_UNKNOWN
+    assert hass.states.get("binary_sensor.all_entity_picture").state == STATE_UNKNOWN
+    assert hass.states.get("binary_sensor.all_attribute").state == STATE_UNKNOWN
 
     hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
     await hass.async_block_till_done()