evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Rewrite efergy unittest tests to pytest style test functions. (#41769)

Browse source 
Issue: https://github.com/home-assistant/core/issues/40836
This commit is contained in:
thaohtp 2020-10-19 12:09:39 +02:00 • committed by GitHub
parent fe7c8dd80a
commit a57d1c1bf0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 47 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

67
tests/components/efergy/test_sensor.py
View file

@ -1,11 +1,8 @@
"""The tests for Efergy sensor platform.""" """The tests for Efergy sensor platform."""
import unittest
import requests_mock from homeassistant.setup import async_setup_component
from homeassistant.setup import setup_component from tests.common import load_fixture
from tests.common import get_test_home_assistant, load_fixture
token = "9p6QGJ7dpZfO3fqPTBk1fyEmjV1cGoLT" token = "9p6QGJ7dpZfO3fqPTBk1fyEmjV1cGoLT"
multi_sensor_token = "9r6QGF7dpZfO3fqPTBl1fyRmjV1cGoLT" multi_sensor_token = "9r6QGF7dpZfO3fqPTBl1fyRmjV1cGoLT"
@ -60,49 +57,25 @@ def mock_responses(mock):
) )
class TestEfergySensor(unittest.TestCase): async def test_single_sensor_readings(hass, requests_mock):
"""Tests the Efergy Sensor platform.""" """Test for successfully setting up the Efergy platform."""
mock_responses(requests_mock)
assert await async_setup_component(hass, "sensor", {"sensor": ONE_SENSOR_CONFIG})
await hass.async_block_till_done()
DEVICES = [] assert "38.21" == hass.states.get("sensor.energy_consumed").state
assert "1580" == hass.states.get("sensor.energy_usage").state
assert "ok" == hass.states.get("sensor.energy_budget").state
assert "5.27" == hass.states.get("sensor.energy_cost").state
assert "1628" == hass.states.get("sensor.efergy_728386").state
@requests_mock.Mocker()
def add_entities(self, devices, mock):
"""Mock add devices."""
mock_responses(mock)
for device in devices:
device.update()
self.DEVICES.append(device)
def setUp(self): async def test_multi_sensor_readings(hass, requests_mock):
"""Initialize values for this test case class.""" """Test for multiple sensors in one household."""
self.hass = get_test_home_assistant() mock_responses(requests_mock)
self.config = ONE_SENSOR_CONFIG assert await async_setup_component(hass, "sensor", {"sensor": MULTI_SENSOR_CONFIG})
self.addCleanup(self.tear_down_cleanup) await hass.async_block_till_done()
def tear_down_cleanup(self): assert "218" == hass.states.get("sensor.efergy_728386").state
"""Stop everything that was started.""" assert "1808" == hass.states.get("sensor.efergy_0").state
self.hass.stop() assert "312" == hass.states.get("sensor.efergy_728387").state
@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.hass.block_till_done()
assert "38.21" == self.hass.states.get("sensor.energy_consumed").state
assert "1580" == self.hass.states.get("sensor.energy_usage").state
assert "ok" == self.hass.states.get("sensor.energy_budget").state
assert "5.27" == self.hass.states.get("sensor.energy_cost").state
assert "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.hass.block_till_done()
assert "218" == self.hass.states.get("sensor.efergy_728386").state
assert "1808" == self.hass.states.get("sensor.efergy_0").state
assert "312" == self.hass.states.get("sensor.efergy_728387").state