Add support for managing the silabs multiprotocol add-on (#82170)

* Add support for managing the silabs multiprotocol add-on

* Fix passing context when starting option flow

* Allow unloading a ha yellow config entry

* Fix tests

* Log data passed to ZHA option flow

* Improve ZHA migration logic

* Move tests

* Improve test coverage

* Remove dead code

* Drop automatic ZHA migration

tests/components/homeassistant_yellow/test_init.py

@@ -4,6 +4,7 @@ from unittest.mock import patch
 import pytest
 
 from homeassistant.components import zha
+from homeassistant.components.hassio.handler import HassioAPIError
 from homeassistant.components.homeassistant_yellow.const import DOMAIN
 from homeassistant.config_entries import ConfigEntryState
 from homeassistant.core import HomeAssistant
@@ -15,7 +16,7 @@ from tests.common import MockConfigEntry, MockModule, mock_integration
     "onboarded, num_entries, num_flows", ((False, 1, 0), (True, 0, 1))
 )
 async def test_setup_entry(
-    hass: HomeAssistant, onboarded, num_entries, num_flows
+    hass: HomeAssistant, onboarded, num_entries, num_flows, addon_store_info
 ) -> None:
     """Test setup of a config entry, including setup of zha."""
     mock_integration(hass, MockModule("hassio"))
@@ -53,8 +54,11 @@ async def test_setup_entry(
     assert len(hass.config_entries.flow.async_progress_by_handler("zha")) == num_flows
     assert len(hass.config_entries.async_entries("zha")) == num_entries
 
+    # Test unloading the config entry
+    assert await hass.config_entries.async_unload(config_entry.entry_id)
 
-async def test_setup_zha(hass: HomeAssistant) -> None:
+
+async def test_setup_zha(hass: HomeAssistant, addon_store_info) -> None:
     """Test zha gets the right config."""
     mock_integration(hass, MockModule("hassio"))
 
@@ -100,6 +104,54 @@ async def test_setup_zha(hass: HomeAssistant) -> None:
     assert config_entry.title == "Yellow"
 
 
+async def test_setup_zha_multipan(
+    hass: HomeAssistant, addon_info, addon_running
+) -> None:
+    """Test zha gets the right config."""
+    mock_integration(hass, MockModule("hassio"))
+
+    # Setup the config entry
+    config_entry = MockConfigEntry(
+        data={},
+        domain=DOMAIN,
+        options={},
+        title="Home Assistant Yellow",
+    )
+    config_entry.add_to_hass(hass)
+    with patch(
+        "homeassistant.components.homeassistant_yellow.get_os_info",
+        return_value={"board": "yellow"},
+    ) as mock_get_os_info, patch(
+        "homeassistant.components.onboarding.async_is_onboarded", return_value=False
+    ):
+        assert await hass.config_entries.async_setup(config_entry.entry_id)
+        await hass.async_block_till_done()
+        assert len(mock_get_os_info.mock_calls) == 1
+
+    # Finish setting up ZHA
+    zha_flows = hass.config_entries.flow.async_progress_by_handler("zha")
+    assert len(zha_flows) == 1
+    assert zha_flows[0]["step_id"] == "choose_formation_strategy"
+
+    await hass.config_entries.flow.async_configure(
+        zha_flows[0]["flow_id"],
+        user_input={"next_step_id": zha.config_flow.FORMATION_REUSE_SETTINGS},
+    )
+    await hass.async_block_till_done()
+
+    config_entry = hass.config_entries.async_entries("zha")[0]
+    assert config_entry.data == {
+        "device": {
+            "baudrate": 115200,
+            "flow_control": "hardware",
+            "path": "socket://core-silabs-multiprotocol:9999",
+        },
+        "radio_type": "ezsp",
+    }
+    assert config_entry.options == {}
+    assert config_entry.title == "Yellow"
+
+
 async def test_setup_entry_wrong_board(hass: HomeAssistant) -> None:
     """Test setup of a config entry with wrong board type."""
     mock_integration(hass, MockModule("hassio"))
@@ -141,3 +193,55 @@ async def test_setup_entry_wait_hassio(hass: HomeAssistant) -> None:
         await hass.async_block_till_done()
         assert len(mock_get_os_info.mock_calls) == 1
         assert config_entry.state == ConfigEntryState.SETUP_RETRY
+
+
+async def test_setup_entry_addon_info_fails(
+    hass: HomeAssistant, addon_store_info
+) -> None:
+    """Test setup of a config entry when fetching addon info fails."""
+    mock_integration(hass, MockModule("hassio"))
+    addon_store_info.side_effect = HassioAPIError("Boom")
+
+    # Setup the config entry
+    config_entry = MockConfigEntry(
+        data={},
+        domain=DOMAIN,
+        options={},
+        title="Home Assistant Yellow",
+    )
+    config_entry.add_to_hass(hass)
+    with patch(
+        "homeassistant.components.homeassistant_yellow.get_os_info",
+        return_value={"board": "yellow"},
+    ), patch(
+        "homeassistant.components.onboarding.async_is_onboarded", return_value=False
+    ):
+        assert not await hass.config_entries.async_setup(config_entry.entry_id)
+        await hass.async_block_till_done()
+        assert config_entry.state == ConfigEntryState.SETUP_RETRY
+
+
+async def test_setup_entry_addon_not_running(
+    hass: HomeAssistant, addon_installed, start_addon
+) -> None:
+    """Test the addon is started if it is not running."""
+    mock_integration(hass, MockModule("hassio"))
+
+    # Setup the config entry
+    config_entry = MockConfigEntry(
+        data={},
+        domain=DOMAIN,
+        options={},
+        title="Home Assistant Yellow",
+    )
+    config_entry.add_to_hass(hass)
+    with patch(
+        "homeassistant.components.homeassistant_yellow.get_os_info",
+        return_value={"board": "yellow"},
+    ), patch(
+        "homeassistant.components.onboarding.async_is_onboarded", return_value=False
+    ):
+        assert not await hass.config_entries.async_setup(config_entry.entry_id)
+        await hass.async_block_till_done()
+        assert config_entry.state == ConfigEntryState.SETUP_RETRY
+        start_addon.assert_called_once()