Fix more test

2019-03-15 09:14:20 -07:00 · 2019-03-15 09:14:20 -07:00 · e581d9e249
parent 3ce50b0a6a
commit e581d9e249
1 changed files with 1 additions and 2 deletions
--- a/tests/components/mobile_app/test_entity.py
+++ b/tests/components/mobile_app/test_entity.py
@ -65,8 +65,7 @@ async def test_sensor(hass, create_registrations, webhook_client):  # noqa: F401

    assert update_resp.status == 200

-    updated_entity = hass.states.async_all()[2]
-
+    updated_entity = hass.states.get('sensor.battery_state')
    assert updated_entity.state == '123'