Fix test

2019-03-15 07:47:13 -07:00 · 2019-03-15 07:47:13 -07:00 · 3ce50b0a6a
parent 941f9b29dc
commit 3ce50b0a6a
1 changed files with 2 additions and 5 deletions
--- a/tests/components/mobile_app/test_entity.py
+++ b/tests/components/mobile_app/test_entity.py
@ -37,11 +37,8 @@ async def test_sensor(hass, create_registrations, webhook_client):  # noqa: F401
    json = await reg_resp.json()
    assert json == {'status': 'registered'}

-    # 3 because we require device_tracker which adds zone.home and
-    # group.all_devices
-    assert len(hass.states.async_all()) == 3
-
-    entity = hass.states.async_all()[2]
+    entity = hass.states.get('sensor.battery_state')
+    assert entity is not None

    assert entity.attributes['device_class'] == 'battery'
    assert entity.attributes['icon'] == 'mdi:battery'