From 7c5846aed290e405a31d0861a8842d0944280e84 Mon Sep 17 00:00:00 2001
From: Diogo Gomes <diogogomes@gmail.com>
Date: Wed, 3 Apr 2019 07:49:53 +0100
Subject: [PATCH] Fix #22648 - Utility_meter would try to cancel a non existing
 task (#22669)

* don't cancel tariff that are paused

* test tariffs
---
 .../components/utility_meter/sensor.py        |  3 +-
 tests/components/utility_meter/test_sensor.py | 43 ++++++++++++++++---
 2 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/homeassistant/components/utility_meter/sensor.py b/homeassistant/components/utility_meter/sensor.py
index dd1514f5e43..2c151634a95 100644
--- a/homeassistant/components/utility_meter/sensor.py
+++ b/homeassistant/components/utility_meter/sensor.py
@@ -118,7 +118,8 @@ class UtilityMeterSensor(RestoreEntity):
             self._collecting = async_track_state_change(
                 self.hass, self._sensor_source_id, self.async_reading)
         else:
-            self._collecting()
+            if self._collecting:
+                self._collecting()
             self._collecting = None
 
         _LOGGER.debug("%s - %s - source <%s>", self._name,
diff --git a/tests/components/utility_meter/test_sensor.py b/tests/components/utility_meter/test_sensor.py
index ee291439a2c..6b8705bf776 100644
--- a/tests/components/utility_meter/test_sensor.py
+++ b/tests/components/utility_meter/test_sensor.py
@@ -6,10 +6,11 @@ from unittest.mock import patch
 from contextlib import contextmanager
 
 from tests.common import async_fire_time_changed
-from homeassistant.const import EVENT_HOMEASSISTANT_START
+from homeassistant.const import EVENT_HOMEASSISTANT_START, ATTR_ENTITY_ID
 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.utility_meter.const import (
+    DOMAIN, SERVICE_SELECT_TARIFF, ATTR_TARIFF)
 from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
 
 _LOGGER = logging.getLogger(__name__)
@@ -31,7 +32,7 @@ async def test_state(hass):
         'utility_meter': {
             'energy_bill': {
                 'source': 'sensor.energy',
-            }
+                'tariffs': ['onpeak', 'midpeak', 'offpeak']},
         }
     }
 
@@ -51,11 +52,43 @@ async def test_state(hass):
                               force_update=True)
         await hass.async_block_till_done()
 
-    state = hass.states.get('sensor.energy_bill')
+    state = hass.states.get('sensor.energy_bill_onpeak')
     assert state is not None
-
     assert state.state == '1'
 
+    state = hass.states.get('sensor.energy_bill_midpeak')
+    assert state is not None
+    assert state.state == '0'
+
+    state = hass.states.get('sensor.energy_bill_offpeak')
+    assert state is not None
+    assert state.state == '0'
+
+    await hass.services.async_call(DOMAIN, SERVICE_SELECT_TARIFF, {
+        ATTR_ENTITY_ID: 'utility_meter.energy_bill', ATTR_TARIFF: 'offpeak'
+        }, blocking=True)
+
+    await hass.async_block_till_done()
+
+    now = dt_util.utcnow() + timedelta(seconds=20)
+    with patch('homeassistant.util.dt.utcnow',
+               return_value=now):
+        hass.states.async_set(entity_id, 6, {"unit_of_measurement": "kWh"},
+                              force_update=True)
+        await hass.async_block_till_done()
+
+    state = hass.states.get('sensor.energy_bill_onpeak')
+    assert state is not None
+    assert state.state == '1'
+
+    state = hass.states.get('sensor.energy_bill_midpeak')
+    assert state is not None
+    assert state.state == '0'
+
+    state = hass.states.get('sensor.energy_bill_offpeak')
+    assert state is not None
+    assert state.state == '3'
+
 
 async def test_net_consumption(hass):
     """Test utility sensor state."""