diff --git a/tests/components/vizio/test_init.py b/tests/components/vizio/test_init.py index ccda9253ec7..288dd2c6ac0 100644 --- a/tests/components/vizio/test_init.py +++ b/tests/components/vizio/test_init.py @@ -3,9 +3,8 @@ from datetime import timedelta import pytest -from homeassistant.components.media_player.const import DOMAIN as MP_DOMAIN from homeassistant.components.vizio.const import DOMAIN -from homeassistant.const import STATE_UNAVAILABLE +from homeassistant.const import STATE_UNAVAILABLE, Platform from homeassistant.core import HomeAssistant from homeassistant.setup import async_setup_component import homeassistant.util.dt as dt_util @@ -25,7 +24,7 @@ async def test_setup_component( hass, DOMAIN, {DOMAIN: MOCK_USER_VALID_TV_CONFIG} ) await hass.async_block_till_done() - assert len(hass.states.async_entity_ids(MP_DOMAIN)) == 1 + assert len(hass.states.async_entity_ids(Platform.MEDIA_PLAYER)) == 1 async def test_tv_load_and_unload( @@ -40,12 +39,12 @@ async def test_tv_load_and_unload( config_entry.add_to_hass(hass) assert await hass.config_entries.async_setup(config_entry.entry_id) await hass.async_block_till_done() - assert len(hass.states.async_entity_ids(MP_DOMAIN)) == 1 + assert len(hass.states.async_entity_ids(Platform.MEDIA_PLAYER)) == 1 assert DOMAIN in hass.data assert await config_entry.async_unload(hass) await hass.async_block_till_done() - entities = hass.states.async_entity_ids(MP_DOMAIN) + entities = hass.states.async_entity_ids(Platform.MEDIA_PLAYER) assert len(entities) == 1 for entity in entities: assert hass.states.get(entity).state == STATE_UNAVAILABLE @@ -64,12 +63,12 @@ async def test_speaker_load_and_unload( config_entry.add_to_hass(hass) assert await hass.config_entries.async_setup(config_entry.entry_id) await hass.async_block_till_done() - assert len(hass.states.async_entity_ids(MP_DOMAIN)) == 1 + assert len(hass.states.async_entity_ids(Platform.MEDIA_PLAYER)) == 1 assert DOMAIN in hass.data assert await config_entry.async_unload(hass) await hass.async_block_till_done() - entities = hass.states.async_entity_ids(MP_DOMAIN) + entities = hass.states.async_entity_ids(Platform.MEDIA_PLAYER) assert len(entities) == 1 for entity in entities: assert hass.states.get(entity).state == STATE_UNAVAILABLE @@ -91,7 +90,7 @@ async def test_coordinator_update_failure( config_entry.add_to_hass(hass) assert await hass.config_entries.async_setup(config_entry.entry_id) await hass.async_block_till_done() - assert len(hass.states.async_entity_ids(MP_DOMAIN)) == 1 + assert len(hass.states.async_entity_ids(Platform.MEDIA_PLAYER)) == 1 assert DOMAIN in hass.data # Failing 25 days in a row should result in a single log message