mirror of https://github.com/home-assistant/core
104 lines
3.3 KiB
Python
104 lines
3.3 KiB
Python
"""Test the Home Assistant Yellow hardware platform."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.hassio import DOMAIN as HASSIO_DOMAIN
|
|
from homeassistant.components.homeassistant_yellow.const import DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry, MockModule, mock_integration
|
|
from tests.typing import WebSocketGenerator
|
|
|
|
|
|
@pytest.mark.usefixtures("supervisor_client")
|
|
async def test_hardware_info(
|
|
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, addon_store_info
|
|
) -> None:
|
|
"""Test we can get the board info."""
|
|
mock_integration(hass, MockModule("hassio"))
|
|
await async_setup_component(hass, HASSIO_DOMAIN, {})
|
|
|
|
# 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"},
|
|
):
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
client = await hass_ws_client(hass)
|
|
|
|
with patch(
|
|
"homeassistant.components.homeassistant_yellow.hardware.get_os_info",
|
|
return_value={"board": "yellow"},
|
|
):
|
|
await client.send_json({"id": 1, "type": "hardware/info"})
|
|
msg = await client.receive_json()
|
|
|
|
assert msg["id"] == 1
|
|
assert msg["success"]
|
|
assert msg["result"] == {
|
|
"hardware": [
|
|
{
|
|
"board": {
|
|
"hassio_board_id": "yellow",
|
|
"manufacturer": "homeassistant",
|
|
"model": "yellow",
|
|
"revision": None,
|
|
},
|
|
"config_entries": [config_entry.entry_id],
|
|
"dongle": None,
|
|
"name": "Home Assistant Yellow",
|
|
"url": "https://yellow.home-assistant.io/documentation/",
|
|
}
|
|
]
|
|
}
|
|
|
|
|
|
@pytest.mark.parametrize("os_info", [None, {"board": None}, {"board": "other"}])
|
|
@pytest.mark.usefixtures("supervisor_client")
|
|
async def test_hardware_info_fail(
|
|
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, os_info, addon_store_info
|
|
) -> None:
|
|
"""Test async_info raises if os_info is not as expected."""
|
|
mock_integration(hass, MockModule("hassio"))
|
|
await async_setup_component(hass, HASSIO_DOMAIN, {})
|
|
|
|
# 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"},
|
|
):
|
|
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
client = await hass_ws_client(hass)
|
|
|
|
with patch(
|
|
"homeassistant.components.homeassistant_yellow.hardware.get_os_info",
|
|
return_value=os_info,
|
|
):
|
|
await client.send_json({"id": 1, "type": "hardware/info"})
|
|
msg = await client.receive_json()
|
|
|
|
assert msg["id"] == 1
|
|
assert msg["success"]
|
|
assert msg["result"] == {"hardware": []}
|