mirror of
https://github.com/home-assistant/core.git
synced 2025-04-25 17:57:55 +00:00
aaec464627
* 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
109 lines
3.2 KiB
Python
109 lines
3.2 KiB
Python
"""Test fixtures for the Home Assistant Hardware integration."""
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture(name="addon_running")
|
|
def mock_addon_running(addon_store_info, addon_info):
|
|
"""Mock add-on already running."""
|
|
addon_store_info.return_value = {
|
|
"installed": "1.0.0",
|
|
"state": "started",
|
|
"version": "1.0.0",
|
|
}
|
|
addon_info.return_value["hostname"] = "core-silabs-multiprotocol"
|
|
addon_info.return_value["state"] = "started"
|
|
addon_info.return_value["version"] = "1.0.0"
|
|
return addon_info
|
|
|
|
|
|
@pytest.fixture(name="addon_installed")
|
|
def mock_addon_installed(addon_store_info, addon_info):
|
|
"""Mock add-on already installed but not running."""
|
|
addon_store_info.return_value = {
|
|
"installed": "1.0.0",
|
|
"state": "stopped",
|
|
"version": "1.0.0",
|
|
}
|
|
addon_info.return_value["hostname"] = "core-silabs-multiprotocol"
|
|
addon_info.return_value["state"] = "stopped"
|
|
addon_info.return_value["version"] = "1.0.0"
|
|
return addon_info
|
|
|
|
|
|
@pytest.fixture(name="addon_store_info")
|
|
def addon_store_info_fixture():
|
|
"""Mock Supervisor add-on store info."""
|
|
with patch(
|
|
"homeassistant.components.hassio.addon_manager.async_get_addon_store_info"
|
|
) as addon_store_info:
|
|
addon_store_info.return_value = {
|
|
"installed": None,
|
|
"state": None,
|
|
"version": "1.0.0",
|
|
}
|
|
yield addon_store_info
|
|
|
|
|
|
@pytest.fixture(name="addon_info")
|
|
def addon_info_fixture():
|
|
"""Mock Supervisor add-on info."""
|
|
with patch(
|
|
"homeassistant.components.hassio.addon_manager.async_get_addon_info",
|
|
) as addon_info:
|
|
addon_info.return_value = {
|
|
"hostname": None,
|
|
"options": {},
|
|
"state": None,
|
|
"update_available": False,
|
|
"version": None,
|
|
}
|
|
yield addon_info
|
|
|
|
|
|
@pytest.fixture(name="set_addon_options")
|
|
def set_addon_options_fixture():
|
|
"""Mock set add-on options."""
|
|
with patch(
|
|
"homeassistant.components.hassio.addon_manager.async_set_addon_options"
|
|
) as set_options:
|
|
yield set_options
|
|
|
|
|
|
@pytest.fixture(name="install_addon_side_effect")
|
|
def install_addon_side_effect_fixture(addon_store_info, addon_info):
|
|
"""Return the install add-on side effect."""
|
|
|
|
async def install_addon(hass, slug):
|
|
"""Mock install add-on."""
|
|
addon_store_info.return_value = {
|
|
"installed": "1.0.0",
|
|
"state": "stopped",
|
|
"version": "1.0.0",
|
|
}
|
|
addon_info.return_value["hostname"] = "core-silabs-multiprotocol"
|
|
addon_info.return_value["state"] = "stopped"
|
|
addon_info.return_value["version"] = "1.0.0"
|
|
|
|
return install_addon
|
|
|
|
|
|
@pytest.fixture(name="install_addon")
|
|
def mock_install_addon(install_addon_side_effect):
|
|
"""Mock install add-on."""
|
|
with patch(
|
|
"homeassistant.components.hassio.addon_manager.async_install_addon",
|
|
side_effect=install_addon_side_effect,
|
|
) as install_addon:
|
|
yield install_addon
|
|
|
|
|
|
@pytest.fixture(name="start_addon")
|
|
def start_addon_fixture():
|
|
"""Mock start add-on."""
|
|
with patch(
|
|
"homeassistant.components.hassio.addon_manager.async_start_addon"
|
|
) as start_addon:
|
|
yield start_addon
|