core/tests/components/fibaro/test_sensor.py

40 lines
1.3 KiB
Python

"""Test the Fibaro sensor platform."""
from unittest.mock import Mock, patch
from homeassistant.components.sensor import SensorDeviceClass
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from .conftest import init_integration
from tests.common import MockConfigEntry
async def test_power_sensor_detected(
hass: HomeAssistant,
entity_registry: er.EntityRegistry,
mock_fibaro_client: Mock,
mock_config_entry: MockConfigEntry,
mock_power_sensor: Mock,
mock_room: Mock,
) -> None:
"""Test that the strange power entity is detected.
Similar to a Qubino 3-Phase power meter.
"""
# Arrange
mock_fibaro_client.read_rooms.return_value = [mock_room]
mock_fibaro_client.read_devices.return_value = [mock_power_sensor]
with patch("homeassistant.components.fibaro.PLATFORMS", [Platform.SENSOR]):
# Act
await init_integration(hass, mock_config_entry)
# Assert
entry = entity_registry.async_get("sensor.room_1_test_sensor_1_power")
assert entry
assert entry.unique_id == "hc2_111111.1_power"
assert entry.original_name == "Room 1 Test sensor Power"
assert entry.original_device_class == SensorDeviceClass.POWER