Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Migrate plex to use async_update_entry to alter config entries (#110405)

pull/110414/head^2
J. Nick Koston 2024-02-12 14:38:38 -06:00 committed by GitHub
parent 15e8d66fea
commit 5c60ff19e9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/components/plex/test_config_flow.py
View File

@ -442,7 +442,8 @@ async def test_option_flow_new_users_available(
OPTIONS_OWNER_ONLY[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = { OPTIONS_OWNER_ONLY[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = {
"User 1": {"enabled": True} "User 1": {"enabled": True}
} }
entry.options = OPTIONS_OWNER_ONLY entry.add_to_hass(hass)
hass.config_entries.async_update_entry(entry, options=OPTIONS_OWNER_ONLY)
mock_plex_server = await setup_plex_server(config_entry=entry) mock_plex_server = await setup_plex_server(config_entry=entry)
await hass.async_block_till_done() await hass.async_block_till_done()

8
tests/components/plex/test_init.py
View File

@ -76,7 +76,8 @@ async def test_setup_with_insecure_config_entry(
"""Test setup component with config.""" """Test setup component with config."""
INSECURE_DATA = copy.deepcopy(DEFAULT_DATA) INSECURE_DATA = copy.deepcopy(DEFAULT_DATA)
INSECURE_DATA[const.PLEX_SERVER_CONFIG][CONF_VERIFY_SSL] = False INSECURE_DATA[const.PLEX_SERVER_CONFIG][CONF_VERIFY_SSL] = False
entry.data = INSECURE_DATA entry.add_to_hass(hass)
hass.config_entries.async_update_entry(entry, data=INSECURE_DATA)
await setup_plex_server(config_entry=entry) await setup_plex_server(config_entry=entry)
@ -268,11 +269,12 @@ async def test_setup_when_certificate_changed(
assert old_entry.data[const.PLEX_SERVER_CONFIG][CONF_URL] == new_url assert old_entry.data[const.PLEX_SERVER_CONFIG][CONF_URL] == new_url
async def test_tokenless_server(entry, setup_plex_server) -> None: async def test_tokenless_server(hass, entry, setup_plex_server) -> None:
"""Test setup with a server with token auth disabled.""" """Test setup with a server with token auth disabled."""
TOKENLESS_DATA = copy.deepcopy(DEFAULT_DATA) TOKENLESS_DATA = copy.deepcopy(DEFAULT_DATA)
TOKENLESS_DATA[const.PLEX_SERVER_CONFIG].pop(CONF_TOKEN, None) TOKENLESS_DATA[const.PLEX_SERVER_CONFIG].pop(CONF_TOKEN, None)
entry.data = TOKENLESS_DATA entry.add_to_hass(hass)
hass.config_entries.async_update_entry(entry, data=TOKENLESS_DATA)
await setup_plex_server(config_entry=entry) await setup_plex_server(config_entry=entry)
assert entry.state is ConfigEntryState.LOADED assert entry.state is ConfigEntryState.LOADED

9
tests/components/plex/test_server.py
View File

@ -28,7 +28,8 @@ async def test_new_users_available(
MONITORED_USERS = {"User 1": {"enabled": True}} MONITORED_USERS = {"User 1": {"enabled": True}}
OPTIONS_WITH_USERS = copy.deepcopy(DEFAULT_OPTIONS) OPTIONS_WITH_USERS = copy.deepcopy(DEFAULT_OPTIONS)
OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = MONITORED_USERS OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = MONITORED_USERS
entry.options = OPTIONS_WITH_USERS entry.add_to_hass(hass)
hass.config_entries.async_update_entry(entry, options=OPTIONS_WITH_USERS)
mock_plex_server = await setup_plex_server(config_entry=entry) mock_plex_server = await setup_plex_server(config_entry=entry)
@ -55,7 +56,8 @@ async def test_new_ignored_users_available(
OPTIONS_WITH_USERS = copy.deepcopy(DEFAULT_OPTIONS) OPTIONS_WITH_USERS = copy.deepcopy(DEFAULT_OPTIONS)
OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = MONITORED_USERS OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_MONITORED_USERS] = MONITORED_USERS
OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_IGNORE_NEW_SHARED_USERS] = True OPTIONS_WITH_USERS[Platform.MEDIA_PLAYER][CONF_IGNORE_NEW_SHARED_USERS] = True
entry.options = OPTIONS_WITH_USERS entry.add_to_hass(hass)
hass.config_entries.async_update_entry(entry, options=OPTIONS_WITH_USERS)
mock_plex_server = await setup_plex_server(config_entry=entry) mock_plex_server = await setup_plex_server(config_entry=entry)
@ -167,7 +169,8 @@ async def test_ignore_plex_web_client(
"""Test option to ignore Plex Web clients.""" """Test option to ignore Plex Web clients."""
OPTIONS = copy.deepcopy(DEFAULT_OPTIONS) OPTIONS = copy.deepcopy(DEFAULT_OPTIONS)
OPTIONS[Platform.MEDIA_PLAYER][CONF_IGNORE_PLEX_WEB_CLIENTS] = True OPTIONS[Platform.MEDIA_PLAYER][CONF_IGNORE_PLEX_WEB_CLIENTS] = True
entry.options = OPTIONS entry.add_to_hass(hass)
hass.config_entries.async_update_entry(entry, options=OPTIONS)
mock_plex_server = await setup_plex_server( mock_plex_server = await setup_plex_server(
config_entry=entry, client_type="plexweb", disable_clients=True config_entry=entry, client_type="plexweb", disable_clients=True