109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
"""The tests for Efergy sensor platform."""
|
|
import unittest
|
|
|
|
import requests_mock
|
|
|
|
from homeassistant.setup import setup_component
|
|
|
|
from tests.common import load_fixture, get_test_home_assistant
|
|
|
|
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(mock):
|
|
"""Mock responses for Efergy."""
|
|
base_url = 'https://engage.efergy.com/mobile_proxy/'
|
|
mock.get(
|
|
base_url + 'getInstant?token=' + token,
|
|
text=load_fixture('efergy_instant.json'))
|
|
mock.get(
|
|
base_url + 'getEnergy?token=' + token + '&offset=300&period=day',
|
|
text=load_fixture('efergy_energy.json'))
|
|
mock.get(
|
|
base_url + 'getBudget?token=' + token,
|
|
text=load_fixture('efergy_budget.json'))
|
|
mock.get(
|
|
base_url + 'getCost?token=' + token + '&offset=300&period=day',
|
|
text=load_fixture('efergy_cost.json'))
|
|
mock.get(
|
|
base_url + 'getCurrentValuesSummary?token=' + token,
|
|
text=load_fixture('efergy_current_values_single.json'))
|
|
mock.get(
|
|
base_url + 'getCurrentValuesSummary?token=' + multi_sensor_token,
|
|
text=load_fixture('efergy_current_values_multi.json'))
|
|
|
|
|
|
class TestEfergySensor(unittest.TestCase):
|
|
"""Tests the Efergy Sensor platform."""
|
|
|
|
DEVICES = []
|
|
|
|
@requests_mock.Mocker()
|
|
def add_devices(self, devices, mock):
|
|
"""Mock add devices."""
|
|
mock_responses(mock)
|
|
for device in devices:
|
|
device.update()
|
|
self.DEVICES.append(device)
|
|
|
|
def setUp(self):
|
|
"""Initialize values for this testcase class."""
|
|
self.hass = get_test_home_assistant()
|
|
self.config = ONE_SENSOR_CONFIG
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
@requests_mock.Mocker()
|
|
def test_single_sensor_readings(self, mock):
|
|
"""Test for successfully setting up the Efergy platform."""
|
|
mock_responses(mock)
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': ONE_SENSOR_CONFIG})
|
|
self.assertEqual('38.21',
|
|
self.hass.states.get('sensor.energy_consumed').state)
|
|
self.assertEqual('1580',
|
|
self.hass.states.get('sensor.energy_usage').state)
|
|
self.assertEqual('ok',
|
|
self.hass.states.get('sensor.energy_budget').state)
|
|
self.assertEqual('5.27',
|
|
self.hass.states.get('sensor.energy_cost').state)
|
|
self.assertEqual('1628',
|
|
self.hass.states.get('sensor.efergy_728386').state)
|
|
|
|
@requests_mock.Mocker()
|
|
def test_multi_sensor_readings(self, mock):
|
|
"""Test for multiple sensors in one household."""
|
|
mock_responses(mock)
|
|
assert setup_component(self.hass, 'sensor', {
|
|
'sensor': MULTI_SENSOR_CONFIG})
|
|
self.assertEqual('218',
|
|
self.hass.states.get('sensor.efergy_728386').state)
|
|
self.assertEqual('1808',
|
|
self.hass.states.get('sensor.efergy_0').state)
|
|
self.assertEqual('312',
|
|
self.hass.states.get('sensor.efergy_728387').state)
|