hass-core/tests/components/efergy/test_sensor.py

"""The tests for Efergy sensor platform."""

import asyncio
from datetime import timedelta

from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util

from tests.common import async_fire_time_changed, load_fixture
from tests.test_util.aiohttp import AiohttpClientMocker

token = "9p6QGJ7dpZfO3fqPTBk1fyEmjV1cGoLT"
multi_sensor_token = "9r6QGF7dpZfO3fqPTBl1fyRmjV1cGoLT"

ONE_SENSOR_CONFIG = {
    "platform": "efergy",
    "app_token": token,
    "utc_offset": "300",
    "monitored_variables": [
        {"type": "amount", "period": "day"},
        {"type": "instant_readings"},
        {"type": "budget"},
        {"type": "cost", "period": "day", "currency": "$"},
        {"type": "current_values"},
    ],
}

MULTI_SENSOR_CONFIG = {
    "platform": "efergy",
    "app_token": multi_sensor_token,
    "utc_offset": "300",
    "monitored_variables": [{"type": "current_values"}],
}


def mock_responses(aioclient_mock: AiohttpClientMocker, error: bool = False):
    """Mock responses for Efergy."""
    base_url = "https://engage.efergy.com/mobile_proxy/"
    if error:
        aioclient_mock.get(
            f"{base_url}getCurrentValuesSummary?token={token}", exc=asyncio.TimeoutError
        )
        return
    aioclient_mock.get(
        f"{base_url}getInstant?token={token}",
        text=load_fixture("efergy/efergy_instant.json"),
    )
    aioclient_mock.get(
        f"{base_url}getEnergy?token={token}&offset=300&period=day",
        text=load_fixture("efergy/efergy_energy.json"),
    )
    aioclient_mock.get(
        f"{base_url}getBudget?token={token}",
        text=load_fixture("efergy/efergy_budget.json"),
    )
    aioclient_mock.get(
        f"{base_url}getCost?token={token}&offset=300&period=day",
        text=load_fixture("efergy/efergy_cost.json"),
    )
    aioclient_mock.get(
        f"{base_url}getCurrentValuesSummary?token={token}",
        text=load_fixture("efergy/efergy_current_values_single.json"),
    )
    aioclient_mock.get(
        f"{base_url}getCurrentValuesSummary?token={multi_sensor_token}",
        text=load_fixture("efergy/efergy_current_values_multi.json"),
    )


async def test_single_sensor_readings(
    hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
):
    """Test for successfully setting up the Efergy platform."""
    mock_responses(aioclient_mock)
    assert await async_setup_component(
        hass, SENSOR_DOMAIN, {SENSOR_DOMAIN: ONE_SENSOR_CONFIG}
    )
    await hass.async_block_till_done()

    assert hass.states.get("sensor.energy_consumed").state == "38.21"
    assert hass.states.get("sensor.energy_usage").state == "1580"
    assert hass.states.get("sensor.energy_budget").state == "ok"
    assert hass.states.get("sensor.energy_cost").state == "5.27"
    assert hass.states.get("sensor.efergy_728386").state == "1628"


async def test_multi_sensor_readings(
    hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
):
    """Test for multiple sensors in one household."""
    mock_responses(aioclient_mock)
    assert await async_setup_component(
        hass, SENSOR_DOMAIN, {SENSOR_DOMAIN: MULTI_SENSOR_CONFIG}
    )
    await hass.async_block_till_done()

    assert hass.states.get("sensor.efergy_728386").state == "218"
    assert hass.states.get("sensor.efergy_0").state == "1808"
    assert hass.states.get("sensor.efergy_728387").state == "312"


async def test_failed_getting_sids(
    hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
):
    """Test failed gettings sids."""
    mock_responses(aioclient_mock, error=True)
    assert await async_setup_component(
        hass, SENSOR_DOMAIN, {SENSOR_DOMAIN: ONE_SENSOR_CONFIG}
    )
    assert not hass.states.async_all("sensor")


async def test_failed_update_and_reconnection(
    hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
):
    """Test failed update and reconnection."""
    mock_responses(aioclient_mock)
    assert await async_setup_component(
        hass, SENSOR_DOMAIN, {SENSOR_DOMAIN: ONE_SENSOR_CONFIG}
    )
    aioclient_mock.clear_requests()
    mock_responses(aioclient_mock, error=True)
    next_update = dt_util.utcnow() + timedelta(seconds=3)
    async_fire_time_changed(hass, next_update)
    await hass.async_block_till_done()
    assert hass.states.get("sensor.efergy_728386").state == STATE_UNAVAILABLE
    aioclient_mock.clear_requests()
    mock_responses(aioclient_mock)
    next_update = dt_util.utcnow() + timedelta(seconds=30)
    async_fire_time_changed(hass, next_update)
    await hass.async_block_till_done()
    assert hass.states.get("sensor.efergy_728386").state == "1628"