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

68
tests/components/foobot/test_sensor.py
View File

@@ -8,49 +8,55 @@ import pytest
import homeassistant.components.sensor as sensor
from homeassistant.components.foobot import sensor as foobot
from homeassistant.const import (TEMP_CELSIUS)
from homeassistant.const import TEMP_CELSIUS
from homeassistant.exceptions import PlatformNotReady
from homeassistant.setup import async_setup_component
from tests.common import load_fixture
VALID_CONFIG = {
'platform': 'foobot',
'token': 'adfdsfasd',
'username': 'example@example.com',
"platform": "foobot",
"token": "adfdsfasd",
"username": "example@example.com",
}
async def test_default_setup(hass, aioclient_mock):
"""Test the default setup."""
aioclient_mock.get(re.compile('api.foobot.io/v2/owner/.*'),
text=load_fixture('foobot_devices.json'))
aioclient_mock.get(re.compile('api.foobot.io/v2/device/.*'),
text=load_fixture('foobot_data.json'))
assert await async_setup_component(hass, sensor.DOMAIN,
{'sensor': VALID_CONFIG})
aioclient_mock.get(
re.compile("api.foobot.io/v2/owner/.*"),
text=load_fixture("foobot_devices.json"),
)
aioclient_mock.get(
re.compile("api.foobot.io/v2/device/.*"), text=load_fixture("foobot_data.json")
)
assert await async_setup_component(hass, sensor.DOMAIN, {"sensor": VALID_CONFIG})
metrics = {'co2': ['1232.0', 'ppm'],
'temperature': ['21.1', TEMP_CELSIUS],
'humidity': ['49.5', '%'],
'pm2_5': ['144.8', 'µg/m3'],
'voc': ['340.7', 'ppb'],
'index': ['138.9', '%']}
metrics = {
"co2": ["1232.0", "ppm"],
"temperature": ["21.1", TEMP_CELSIUS],
"humidity": ["49.5", "%"],
"pm2_5": ["144.8", "µg/m3"],
"voc": ["340.7", "ppb"],
"index": ["138.9", "%"],
}
for name, value in metrics.items():
state = hass.states.get('sensor.foobot_happybot_%s' % name)
state = hass.states.get("sensor.foobot_happybot_%s" % name)
assert state.state == value[0]
assert state.attributes.get('unit_of_measurement') == value[1]
assert state.attributes.get("unit_of_measurement") == value[1]
async def test_setup_timeout_error(hass, aioclient_mock):
"""Expected failures caused by a timeout in API response."""
fake_async_add_entities = MagicMock()
aioclient_mock.get(re.compile('api.foobot.io/v2/owner/.*'),
exc=asyncio.TimeoutError())
aioclient_mock.get(
re.compile("api.foobot.io/v2/owner/.*"), exc=asyncio.TimeoutError()
)
with pytest.raises(PlatformNotReady):
await foobot.async_setup_platform(hass, {'sensor': VALID_CONFIG},
fake_async_add_entities)
await foobot.async_setup_platform(
hass, {"sensor": VALID_CONFIG}, fake_async_add_entities
)
async def test_setup_permanent_error(hass, aioclient_mock):
@@ -59,11 +65,10 @@ async def test_setup_permanent_error(hass, aioclient_mock):
errors = [400, 401, 403]
for error in errors:
aioclient_mock.get(re.compile('api.foobot.io/v2/owner/.*'),
status=error)
result = await foobot.async_setup_platform(hass,
{'sensor': VALID_CONFIG},
fake_async_add_entities)
aioclient_mock.get(re.compile("api.foobot.io/v2/owner/.*"), status=error)
result = await foobot.async_setup_platform(
hass, {"sensor": VALID_CONFIG}, fake_async_add_entities
)
assert result is None
@@ -73,9 +78,8 @@ async def test_setup_temporary_error(hass, aioclient_mock):
errors = [429, 500]
for error in errors:
aioclient_mock.get(re.compile('api.foobot.io/v2/owner/.*'),
status=error)
aioclient_mock.get(re.compile("api.foobot.io/v2/owner/.*"), status=error)
with pytest.raises(PlatformNotReady):
await foobot.async_setup_platform(hass,
{'sensor': VALID_CONFIG},
fake_async_add_entities)
await foobot.async_setup_platform(
hass, {"sensor": VALID_CONFIG}, fake_async_add_entities
)