mirror of
https://github.com/home-assistant/core.git
synced 2026-06-30 03:06:10 +01:00
Bernát Gábor
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
"""Test for the switchbot_cloud base entity."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.switchbot_cloud.const import DOMAIN
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
|
|
from . import METER_INFO, configure_integration
|
|
|
|
|
|
async def test_sw_version_cast_to_string(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
mock_list_devices,
|
|
mock_get_status,
|
|
mock_setup_webhook,
|
|
) -> None:
|
|
"""Test the device sw_version is a string when the API returns an int."""
|
|
mock_list_devices.return_value = [METER_INFO]
|
|
mock_get_status.return_value = {"version": 123}
|
|
|
|
with patch("homeassistant.components.switchbot_cloud.PLATFORMS", [Platform.SENSOR]):
|
|
await configure_integration(hass)
|
|
|
|
device = device_registry.async_get_device(
|
|
identifiers={(DOMAIN, METER_INFO.device_id)}
|
|
)
|
|
assert device is not None
|
|
assert device.sw_version == "123"
|
|
|
|
|
|
async def test_sw_version_none_when_missing(
|
|
hass: HomeAssistant,
|
|
device_registry: dr.DeviceRegistry,
|
|
mock_list_devices,
|
|
mock_get_status,
|
|
mock_setup_webhook,
|
|
) -> None:
|
|
"""Test the device sw_version is None when the API omits the version."""
|
|
mock_list_devices.return_value = [METER_INFO]
|
|
mock_get_status.return_value = {}
|
|
|
|
with patch("homeassistant.components.switchbot_cloud.PLATFORMS", [Platform.SENSOR]):
|
|
await configure_integration(hass)
|
|
|
|
device = device_registry.async_get_device(
|
|
identifiers={(DOMAIN, METER_INFO.device_id)}
|
|
)
|
|
assert device is not None
|
|
assert device.sw_version is None
|