From 07482de4abd7bf14b6433e9ea8ab8702e5cfa735 Mon Sep 17 00:00:00 2001
From: Shay Levy <levyshay1@gmail.com>
Date: Thu, 9 Jan 2025 19:29:17 +0200
Subject: [PATCH] Fix LG webOS TV init test coverage (#135194)

---
 homeassistant/components/webostv/__init__.py |  3 +-
 tests/components/webostv/test_init.py        | 42 +++++++++++++++++++-
 2 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/webostv/__init__.py b/homeassistant/components/webostv/__init__.py
index 499d0a85518..be0002cc588 100644
--- a/homeassistant/components/webostv/__init__.py
+++ b/homeassistant/components/webostv/__init__.py
@@ -129,8 +129,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
         )
     )
 
-    if not entry.update_listeners:
-        entry.async_on_unload(entry.add_update_listener(async_update_options))
+    entry.async_on_unload(entry.add_update_listener(async_update_options))
 
     async def async_on_stop(_event: Event) -> None:
         """Unregister callbacks and disconnect."""
diff --git a/tests/components/webostv/test_init.py b/tests/components/webostv/test_init.py
index e2638c86f5e..ba755d80b30 100644
--- a/tests/components/webostv/test_init.py
+++ b/tests/components/webostv/test_init.py
@@ -5,12 +5,14 @@ from unittest.mock import Mock
 from aiowebostv import WebOsTvPairError
 import pytest
 
-from homeassistant.components.webostv.const import DOMAIN
+from homeassistant.components.media_player import ATTR_INPUT_SOURCE_LIST
+from homeassistant.components.webostv.const import CONF_SOURCES, DOMAIN
 from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntryState
-from homeassistant.const import CONF_CLIENT_SECRET
+from homeassistant.const import CONF_CLIENT_SECRET, EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant
 
 from . import setup_webostv
+from .const import ENTITY_ID
 
 
 async def test_reauth_setup_entry(
@@ -44,3 +46,39 @@ async def test_key_update_setup_entry(
 
     assert entry.state is ConfigEntryState.LOADED
     assert entry.data[CONF_CLIENT_SECRET] == "new_key"
+
+
+async def test_update_options(hass: HomeAssistant, client) -> None:
+    """Test update options triggers reload."""
+    config_entry = await setup_webostv(hass)
+
+    assert config_entry.state is ConfigEntryState.LOADED
+    assert config_entry.update_listeners is not None
+    sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST]
+    assert sources == ["Input01", "Input02", "Live TV"]
+
+    # remove Input01 and reload
+    new_options = config_entry.options.copy()
+    new_options[CONF_SOURCES] = ["Input02", "Live TV"]
+    hass.config_entries.async_update_entry(config_entry, options=new_options)
+    await hass.async_block_till_done()
+
+    assert config_entry.state is ConfigEntryState.LOADED
+    sources = hass.states.get(ENTITY_ID).attributes[ATTR_INPUT_SOURCE_LIST]
+    assert sources == ["Input02", "Live TV"]
+
+
+async def test_disconnect_on_stop(hass: HomeAssistant, client) -> None:
+    """Test we disconnect the client and clear callbacks when Home Assistants stops."""
+    config_entry = await setup_webostv(hass)
+
+    assert client.disconnect.call_count == 0
+    assert client.clear_state_update_callbacks.call_count == 0
+    assert config_entry.state is ConfigEntryState.LOADED
+
+    hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
+    await hass.async_block_till_done()
+
+    assert client.disconnect.call_count == 1
+    assert client.clear_state_update_callbacks.call_count == 1
+    assert config_entry.state is ConfigEntryState.LOADED