Utility meter (#19718)

* initial commit * test service calls * lint * float -> Decimal * extra tests * lint * lint * lint * lint * fix self reset * clean * add services * improve service example description * add optional paused initialization * fix * travis fix * fix YEARLY * add tests for previous bug * address comments and suggestions from @ottowinter * lint * remove debug * add discoverability capabilities * no need for _hass * Update homeassistant/components/sensor/utility_meter.py Co-Authored-By: dgomes <diogogomes@gmail.com> * Update homeassistant/components/sensor/utility_meter.py Co-Authored-By: dgomes <diogogomes@gmail.com> * correct comment * improve error handling * address @MartinHjelmare comments * address @MartinHjelmare comments * one patch is enought * follow @ballob suggestion in https://github.com/home-assistant/architecture/issues/131 * fix tests * review fixes * major refactor * lint * lint * address comments by @MartinHjelmare * rename variable
2019-01-26 15:33:11 +00:00 · 2019-01-26 15:33:11 +00:00 · 1d5ffe9ad5
commit 1d5ffe9ad5
parent ed6e349515
7 changed files with 713 additions and 0 deletions
--- a/tests/components/utility_meter/test_init.py
+++ b/tests/components/utility_meter/test_init.py
@ -0,0 +1,102 @@
+"""The tests for the utility_meter component."""
+import logging
+
+from datetime import timedelta
+from unittest.mock import patch
+
+from homeassistant.const import (
+    EVENT_HOMEASSISTANT_START, ATTR_ENTITY_ID)
+from homeassistant.components.utility_meter.const import (
+    SERVICE_RESET, SERVICE_SELECT_TARIFF, SERVICE_SELECT_NEXT_TARIFF,
+    ATTR_TARIFF)
+from homeassistant.setup import async_setup_component
+import homeassistant.util.dt as dt_util
+from homeassistant.components.utility_meter.const import DOMAIN
+from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def test_services(hass):
+    """Test energy sensor reset service."""
+    config = {
+        'utility_meter': {
+            'energy_bill': {
+                'source': 'sensor.energy',
+                'cycle': 'hourly',
+                'tariffs': ['peak', 'offpeak'],
+            }
+        }
+    }
+
+    assert await async_setup_component(hass, DOMAIN, config)
+    assert await async_setup_component(hass, SENSOR_DOMAIN, config)
+    await hass.async_block_till_done()
+
+    hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
+    entity_id = config[DOMAIN]['energy_bill']['source']
+    hass.states.async_set(entity_id, 1, {"unit_of_measurement": "kWh"})
+    await hass.async_block_till_done()
+
+    now = dt_util.utcnow() + timedelta(seconds=10)
+    with patch('homeassistant.util.dt.utcnow',
+               return_value=now):
+        hass.states.async_set(entity_id, 3, {"unit_of_measurement": "kWh"},
+                              force_update=True)
+        await hass.async_block_till_done()
+
+    state = hass.states.get('sensor.energy_bill_peak')
+    assert state.state == '2'
+
+    state = hass.states.get('sensor.energy_bill_offpeak')
+    assert state.state == '0'
+
+    # Next tariff
+    data = {ATTR_ENTITY_ID: 'utility_meter.energy_bill'}
+    await hass.services.async_call(DOMAIN,
+                                   SERVICE_SELECT_NEXT_TARIFF, data)
+    await hass.async_block_till_done()
+
+    now += timedelta(seconds=10)
+    with patch('homeassistant.util.dt.utcnow',
+               return_value=now):
+        hass.states.async_set(entity_id, 4, {"unit_of_measurement": "kWh"},
+                              force_update=True)
+        await hass.async_block_till_done()
+
+    state = hass.states.get('sensor.energy_bill_peak')
+    assert state.state == '2'
+
+    state = hass.states.get('sensor.energy_bill_offpeak')
+    assert state.state == '1'
+
+    # Change tariff
+    data = {ATTR_ENTITY_ID: 'utility_meter.energy_bill',
+            ATTR_TARIFF: 'peak'}
+    await hass.services.async_call(DOMAIN,
+                                   SERVICE_SELECT_TARIFF, data)
+    await hass.async_block_till_done()
+
+    now += timedelta(seconds=10)
+    with patch('homeassistant.util.dt.utcnow',
+               return_value=now):
+        hass.states.async_set(entity_id, 5, {"unit_of_measurement": "kWh"},
+                              force_update=True)
+        await hass.async_block_till_done()
+
+    state = hass.states.get('sensor.energy_bill_peak')
+    assert state.state == '3'
+
+    state = hass.states.get('sensor.energy_bill_offpeak')
+    assert state.state == '1'
+
+    # Reset meters
+    data = {ATTR_ENTITY_ID: 'utility_meter.energy_bill'}
+    await hass.services.async_call(DOMAIN, SERVICE_RESET, data)
+    await hass.async_block_till_done()
+
+    state = hass.states.get('sensor.energy_bill_peak')
+    assert state.state == '0'
+
+    state = hass.states.get('sensor.energy_bill_offpeak')
+    assert state.state == '0'