Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
tests/components/fido/test_sensor.py
View File

@@ -12,7 +12,7 @@ from tests.common import assert_setup_component
CONTRACT = "123456789"
class FidoClientMock():
class FidoClientMock:
"""Fake Fido client."""
def __init__(self, username, password, timeout=None, httpsession=None):
@@ -25,8 +25,7 @@ class FidoClientMock():
def get_data(self):
"""Return fake fido data."""
return {"balance": 160.12,
"1112223344": {"data_remaining": 100.33}}
return {"balance": 160.12, "1112223344": {"data_remaining": 100.33}}
@asyncio.coroutine
def fetch_data(self):
@@ -47,13 +46,13 @@ class PyFidoErrorMock(Exception):
"""Fake PyFido Error."""
class PyFidoClientFakeModule():
class PyFidoClientFakeModule:
"""Fake pyfido.client module."""
PyFidoError = PyFidoErrorMock
class PyFidoFakeModule():
class PyFidoFakeModule:
"""Fake pyfido module."""
FidoClient = FidoClientMockError
@@ -67,31 +66,28 @@ def fake_async_add_entities(component, update_before_add=False):
@asyncio.coroutine
def test_fido_sensor(loop, hass):
"""Test the Fido number sensor."""
sys.modules['pyfido'] = MagicMock()
sys.modules['pyfido.client'] = MagicMock()
sys.modules['pyfido.client.PyFidoError'] = \
PyFidoErrorMock
sys.modules["pyfido"] = MagicMock()
sys.modules["pyfido.client"] = MagicMock()
sys.modules["pyfido.client.PyFidoError"] = PyFidoErrorMock
import pyfido.client
pyfido.FidoClient = FidoClientMock
pyfido.client.PyFidoError = PyFidoErrorMock
config = {
'sensor': {
'platform': 'fido',
'name': 'fido',
'username': 'myusername',
'password': 'password',
'monitored_variables': [
'balance',
'data_remaining',
],
"sensor": {
"platform": "fido",
"name": "fido",
"username": "myusername",
"password": "password",
"monitored_variables": ["balance", "data_remaining"],
}
}
with assert_setup_component(1):
yield from async_setup_component(hass, 'sensor', config)
state = hass.states.get('sensor.fido_1112223344_balance')
yield from async_setup_component(hass, "sensor", config)
state = hass.states.get("sensor.fido_1112223344_balance")
assert state.state == "160.12"
assert state.attributes.get('number') == "1112223344"
state = hass.states.get('sensor.fido_1112223344_data_remaining')
assert state.attributes.get("number") == "1112223344"
state = hass.states.get("sensor.fido_1112223344_data_remaining")
assert state.state == "100.33"
@@ -99,11 +95,10 @@ def test_fido_sensor(loop, hass):
def test_error(hass, caplog):
"""Test the Fido sensor errors."""
caplog.set_level(logging.ERROR)
sys.modules['pyfido'] = PyFidoFakeModule()
sys.modules['pyfido.client'] = PyFidoClientFakeModule()
sys.modules["pyfido"] = PyFidoFakeModule()
sys.modules["pyfido.client"] = PyFidoClientFakeModule()
config = {}
fake_async_add_entities = MagicMock()
yield from fido.async_setup_platform(hass, config,
fake_async_add_entities)
yield from fido.async_setup_platform(hass, config, fake_async_add_entities)
assert fake_async_add_entities.called is False