"""Tests for the Plugwise Sensor integration."""

from homeassistant.config_entries import ConfigEntryState

from tests.common import Mock
from tests.components.plugwise.common import async_init_integration


async def test_adam_climate_sensor_entities(hass, mock_smile_adam):
    """Test creation of climate related sensor entities."""
    entry = await async_init_integration(hass, mock_smile_adam)
    assert entry.state is ConfigEntryState.LOADED

    state = hass.states.get("sensor.adam_outdoor_temperature")
    assert float(state.state) == 7.81

    state = hass.states.get("sensor.cv_pomp_electricity_consumed")
    assert float(state.state) == 35.6

    state = hass.states.get("sensor.auxiliary_water_temperature")
    assert float(state.state) == 70.0

    state = hass.states.get("sensor.cv_pomp_electricity_consumed_interval")
    assert float(state.state) == 7.37

    await hass.helpers.entity_component.async_update_entity(
        "sensor.zone_lisa_wk_battery"
    )

    state = hass.states.get("sensor.zone_lisa_wk_battery")
    assert int(state.state) == 34


async def test_anna_as_smt_climate_sensor_entities(hass, mock_smile_anna):
    """Test creation of climate related sensor entities."""
    entry = await async_init_integration(hass, mock_smile_anna)
    assert entry.state is ConfigEntryState.LOADED

    state = hass.states.get("sensor.auxiliary_outdoor_temperature")
    assert float(state.state) == 18.0

    state = hass.states.get("sensor.auxiliary_water_temperature")
    assert float(state.state) == 29.1

    state = hass.states.get("sensor.anna_illuminance")
    assert float(state.state) == 86.0


async def test_anna_climate_sensor_entities(hass, mock_smile_anna):
    """Test creation of climate related sensor entities as single master thermostat."""
    mock_smile_anna.single_master_thermostat.side_effect = Mock(return_value=False)
    entry = await async_init_integration(hass, mock_smile_anna)
    assert entry.state is ConfigEntryState.LOADED

    state = hass.states.get("sensor.auxiliary_outdoor_temperature")
    assert float(state.state) == 18.0


async def test_p1_dsmr_sensor_entities(hass, mock_smile_p1):
    """Test creation of power related sensor entities."""
    entry = await async_init_integration(hass, mock_smile_p1)
    assert entry.state is ConfigEntryState.LOADED

    state = hass.states.get("sensor.p1_net_electricity_point")
    assert float(state.state) == -2761.0

    state = hass.states.get("sensor.p1_electricity_consumed_off_peak_cumulative")
    assert float(state.state) == 551.09

    state = hass.states.get("sensor.p1_electricity_produced_peak_point")
    assert float(state.state) == 2761.0

    state = hass.states.get("sensor.p1_electricity_consumed_peak_cumulative")
    assert float(state.state) == 442.932

    state = hass.states.get("sensor.p1_gas_consumed_cumulative")
    assert float(state.state) == 584.85


async def test_stretch_sensor_entities(hass, mock_stretch):
    """Test creation of power related sensor entities."""
    entry = await async_init_integration(hass, mock_stretch)
    assert entry.state is ConfigEntryState.LOADED

    state = hass.states.get("sensor.koelkast_92c4a_electricity_consumed")
    assert float(state.state) == 50.5

    state = hass.states.get("sensor.droger_52559_electricity_consumed_interval")
    assert float(state.state) == 0.0