diff --git a/tests/components/plex/const.py b/tests/components/plex/const.py index 9e376d19cac..ff9d3c0e9b3 100644 --- a/tests/components/plex/const.py +++ b/tests/components/plex/const.py @@ -1,5 +1,4 @@ """Constants used by Plex tests.""" -from homeassistant.components.media_player import DOMAIN as MP_DOMAIN from homeassistant.components.plex import const from homeassistant.const import ( CONF_CLIENT_ID, @@ -8,6 +7,7 @@ from homeassistant.const import ( CONF_TOKEN, CONF_URL, CONF_VERIFY_SSL, + Platform, ) MOCK_SERVERS = [ @@ -55,7 +55,7 @@ SECONDARY_DATA = { } DEFAULT_OPTIONS = { - MP_DOMAIN: { + Platform.MEDIA_PLAYER: { const.CONF_IGNORE_NEW_SHARED_USERS: False, const.CONF_MONITORED_USERS: MOCK_USERS, const.CONF_USE_EPISODE_ART: False, diff --git a/tests/components/plex/test_config_flow.py b/tests/components/plex/test_config_flow.py index c18d38fdba1..43df5092435 100644 --- a/tests/components/plex/test_config_flow.py +++ b/tests/components/plex/test_config_flow.py @@ -8,7 +8,6 @@ import plexapi.exceptions import pytest import requests.exceptions -from homeassistant.components.media_player import DOMAIN as MP_DOMAIN from homeassistant.components.plex import config_flow from homeassistant.components.plex.const import ( AUTOMATIC_SETUP_STRING, @@ -36,6 +35,7 @@ from homeassistant.const import ( CONF_TOKEN, CONF_URL, CONF_VERIFY_SSL, + Platform, ) from .const import DEFAULT_OPTIONS, MOCK_SERVERS, MOCK_TOKEN, PLEX_DIRECT_URL @@ -414,7 +414,7 @@ async def test_option_flow(hass, entry, mock_plex_server): ) assert result["type"] == "create_entry" assert result["data"] == { - MP_DOMAIN: { + Platform.MEDIA_PLAYER: { CONF_USE_EPISODE_ART: True, CONF_IGNORE_NEW_SHARED_USERS: True, CONF_MONITORED_USERS: { @@ -446,7 +446,7 @@ async def test_missing_option_flow(hass, entry, mock_plex_server): ) assert result["type"] == "create_entry" assert result["data"] == { - MP_DOMAIN: { + Platform.MEDIA_PLAYER: { CONF_USE_EPISODE_ART: True, CONF_IGNORE_NEW_SHARED_USERS: True, CONF_MONITORED_USERS: { @@ -460,7 +460,9 @@ async def test_missing_option_flow(hass, entry, mock_plex_server): async def test_option_flow_new_users_available(hass, entry, setup_plex_server): """Test config options multiselect defaults when new Plex users are seen.""" OPTIONS_OWNER_ONLY = copy.deepcopy(DEFAULT_OPTIONS) - OPTIONS_OWNER_ONLY[MP_DOMAIN][CONF_MONITORED_USERS] = {"User 1": {"enabled": True}} + OPTIONS_OWNER_ONLY[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = { + "User 1": {"enabled": True} + } entry.options = OPTIONS_OWNER_ONLY mock_plex_server = await setup_plex_server(config_entry=entry) diff --git a/tests/components/plex/test_device_handling.py b/tests/components/plex/test_device_handling.py index f36e5dc3641..2c23ee7cb09 100644 --- a/tests/components/plex/test_device_handling.py +++ b/tests/components/plex/test_device_handling.py @@ -1,7 +1,7 @@ """Tests for handling the device registry.""" -from homeassistant.components.media_player.const import DOMAIN as MP_DOMAIN from homeassistant.components.plex.const import DOMAIN +from homeassistant.const import Platform async def test_cleanup_orphaned_devices(hass, entry, setup_plex_server): @@ -18,7 +18,7 @@ async def test_cleanup_orphaned_devices(hass, entry, setup_plex_server): assert test_device is not None test_entity = entity_registry.async_get_or_create( - MP_DOMAIN, DOMAIN, "entity_unique_id_123", device_id=test_device.id + Platform.MEDIA_PLAYER, DOMAIN, "entity_unique_id_123", device_id=test_device.id ) assert test_entity is not None @@ -53,9 +53,9 @@ async def test_migrate_transient_devices( identifiers=plexweb_device_id, model="Plex Web", ) - # plexweb_entity = entity_registry.async_get_or_create(MP_DOMAIN, DOMAIN, "unique_id_123:plexweb_id", suggested_object_id="plex_plex_web_chrome", device_id=plexweb_device.id) + entity_registry.async_get_or_create( - MP_DOMAIN, + Platform.MEDIA_PLAYER, DOMAIN, "unique_id_123:plexweb_id", suggested_object_id="plex_plex_web_chrome", @@ -68,7 +68,7 @@ async def test_migrate_transient_devices( model="Plex for Android (TV)", ) entity_registry.async_get_or_create( - MP_DOMAIN, + Platform.MEDIA_PLAYER, DOMAIN, "unique_id_123:1234567890123456-com-plexapp-android", suggested_object_id="plex_plex_for_android_tv_shield_android_tv", diff --git a/tests/components/plex/test_media_search.py b/tests/components/plex/test_media_search.py index 467ab3555f5..adfdff2d1dc 100644 --- a/tests/components/plex/test_media_search.py +++ b/tests/components/plex/test_media_search.py @@ -4,7 +4,6 @@ from unittest.mock import patch from plexapi.exceptions import BadRequest, NotFound import pytest -from homeassistant.components.media_player import DOMAIN as MP_DOMAIN from homeassistant.components.media_player.const import ( ATTR_MEDIA_CONTENT_ID, ATTR_MEDIA_CONTENT_TYPE, @@ -16,7 +15,7 @@ from homeassistant.components.media_player.const import ( SERVICE_PLAY_MEDIA, ) from homeassistant.components.plex.const import DOMAIN -from homeassistant.const import ATTR_ENTITY_ID +from homeassistant.const import ATTR_ENTITY_ID, Platform from homeassistant.exceptions import HomeAssistantError @@ -30,7 +29,7 @@ async def test_media_lookups( requests_mock.get("/player/playback/playMedia", status_code=200) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -42,7 +41,7 @@ async def test_media_lookups( with pytest.raises(HomeAssistantError) as excinfo: with patch("plexapi.server.PlexServer.fetchItem", side_effect=NotFound): assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -57,7 +56,7 @@ async def test_media_lookups( with pytest.raises(HomeAssistantError) as excinfo: payload = '{"library_name": "Not a Library", "show_name": "TV Show"}' assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -70,7 +69,7 @@ async def test_media_lookups( with patch("plexapi.library.LibrarySection.search") as search: assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -82,7 +81,7 @@ async def test_media_lookups( search.assert_called_with(**{"show.title": "TV Show", "libtype": "show"}) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -96,7 +95,7 @@ async def test_media_lookups( ) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -110,7 +109,7 @@ async def test_media_lookups( ) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -129,7 +128,7 @@ async def test_media_lookups( ) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -141,7 +140,7 @@ async def test_media_lookups( search.assert_called_with(**{"artist.title": "Artist", "libtype": "artist"}) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -153,7 +152,7 @@ async def test_media_lookups( search.assert_called_with(**{"album.title": "Album", "libtype": "album"}) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -167,7 +166,7 @@ async def test_media_lookups( ) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -181,7 +180,7 @@ async def test_media_lookups( ) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -200,7 +199,7 @@ async def test_media_lookups( ) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -220,7 +219,7 @@ async def test_media_lookups( # Movie searches assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -232,7 +231,7 @@ async def test_media_lookups( search.assert_called_with(**{"movie.title": "Movie 1", "libtype": None}) assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -248,7 +247,7 @@ async def test_media_lookups( payload = '{"library_name": "Movies", "title": "Not a Movie"}' with patch("plexapi.library.LibrarySection.search", side_effect=BadRequest): assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -262,7 +261,7 @@ async def test_media_lookups( # Playlist searches assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -275,7 +274,7 @@ async def test_media_lookups( with pytest.raises(HomeAssistantError) as excinfo: payload = '{"playlist_name": "Not a Playlist"}' assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, @@ -290,7 +289,7 @@ async def test_media_lookups( with pytest.raises(HomeAssistantError) as excinfo: payload = "{}" assert await hass.services.async_call( - MP_DOMAIN, + Platform.MEDIA_PLAYER, SERVICE_PLAY_MEDIA, { ATTR_ENTITY_ID: media_player_id, diff --git a/tests/components/plex/test_server.py b/tests/components/plex/test_server.py index 724b34cf729..68032868b1a 100644 --- a/tests/components/plex/test_server.py +++ b/tests/components/plex/test_server.py @@ -4,7 +4,6 @@ from unittest.mock import patch from requests.exceptions import ConnectionError, RequestException -from homeassistant.components.media_player import DOMAIN as MP_DOMAIN from homeassistant.components.plex.const import ( CONF_IGNORE_NEW_SHARED_USERS, CONF_IGNORE_PLEX_WEB_CLIENTS, @@ -13,6 +12,7 @@ from homeassistant.components.plex.const import ( DOMAIN, SERVERS, ) +from homeassistant.const import Platform from .const import DEFAULT_DATA, DEFAULT_OPTIONS from .helpers import trigger_plex_update, wait_for_debouncer @@ -22,7 +22,7 @@ async def test_new_users_available(hass, entry, setup_plex_server): """Test setting up when new users available on Plex server.""" MONITORED_USERS = {"User 1": {"enabled": True}} OPTIONS_WITH_USERS = copy.deepcopy(DEFAULT_OPTIONS) - OPTIONS_WITH_USERS[MP_DOMAIN][CONF_MONITORED_USERS] = MONITORED_USERS + OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = MONITORED_USERS entry.options = OPTIONS_WITH_USERS mock_plex_server = await setup_plex_server(config_entry=entry) @@ -48,8 +48,8 @@ async def test_new_ignored_users_available( """Test setting up when new users available on Plex server but are ignored.""" MONITORED_USERS = {"User 1": {"enabled": True}} OPTIONS_WITH_USERS = copy.deepcopy(DEFAULT_OPTIONS) - OPTIONS_WITH_USERS[MP_DOMAIN][CONF_MONITORED_USERS] = MONITORED_USERS - OPTIONS_WITH_USERS[MP_DOMAIN][CONF_IGNORE_NEW_SHARED_USERS] = True + OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = MONITORED_USERS + OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_IGNORE_NEW_SHARED_USERS] = True entry.options = OPTIONS_WITH_USERS mock_plex_server = await setup_plex_server(config_entry=entry) @@ -151,7 +151,7 @@ async def test_mark_sessions_idle( async def test_ignore_plex_web_client(hass, entry, setup_plex_server): """Test option to ignore Plex Web clients.""" OPTIONS = copy.deepcopy(DEFAULT_OPTIONS) - OPTIONS[MP_DOMAIN][CONF_IGNORE_PLEX_WEB_CLIENTS] = True + OPTIONS[Platform.MEDIA_PLAYER][CONF_IGNORE_PLEX_WEB_CLIENTS] = True entry.options = OPTIONS mock_plex_server = await setup_plex_server(