jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 05:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate startca tests from coroutine to async/await (#30354)

Browse Source
This commit is contained in:
Franck Nijhof 2020-01-01 16:17:35 +01:00 committed by Martin Hjelmare
parent e099d57bde
commit 9fbe6d60cb
1 changed files with 8 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/components/startca/test_sensor.py
View File

@ -1,13 +1,10 @@
"""Tests for the Start.ca sensor platform."""
import asyncio
from homeassistant.bootstrap import async_setup_component
from homeassistant.components.startca.sensor import StartcaData
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@asyncio.coroutine
def test_capped_setup(hass, aioclient_mock):
async def test_capped_setup(hass, aioclient_mock):
"""Test the default setup."""
config = {
"platform": "startca",
@ -51,7 +48,7 @@ def test_capped_setup(hass, aioclient_mock):
"https://www.start.ca/support/usage/api?key=" "NOTAKEY", text=result
)
yield from async_setup_component(hass, "sensor", {"sensor": config})
await async_setup_component(hass, "sensor", {"sensor": config})
state = hass.states.get("sensor.start_ca_usage_ratio")
assert state.attributes.get("unit_of_measurement") == "%"
@ -102,8 +99,7 @@ def test_capped_setup(hass, aioclient_mock):
assert state.state == "95.05"
@asyncio.coroutine
def test_unlimited_setup(hass, aioclient_mock):
async def test_unlimited_setup(hass, aioclient_mock):
"""Test the default setup."""
config = {
"platform": "startca",
@ -147,7 +143,7 @@ def test_unlimited_setup(hass, aioclient_mock):
"https://www.start.ca/support/usage/api?key=" "NOTAKEY", text=result
)
yield from async_setup_component(hass, "sensor", {"sensor": config})
await async_setup_component(hass, "sensor", {"sensor": config})
state = hass.states.get("sensor.start_ca_usage_ratio")
assert state.attributes.get("unit_of_measurement") == "%"
@ -198,8 +194,7 @@ def test_unlimited_setup(hass, aioclient_mock):
assert state.state == "inf"
@asyncio.coroutine
def test_bad_return_code(hass, aioclient_mock):
async def test_bad_return_code(hass, aioclient_mock):
"""Test handling a return code that isn't HTTP OK."""
aioclient_mock.get(
"https://www.start.ca/support/usage/api?key=" "NOTAKEY", status=404
@ -207,12 +202,11 @@ def test_bad_return_code(hass, aioclient_mock):
scd = StartcaData(hass.loop, async_get_clientsession(hass), "NOTAKEY", 400)
result = yield from scd.async_update()
result = await scd.async_update()
assert result is False
@asyncio.coroutine
def test_bad_json_decode(hass, aioclient_mock):
async def test_bad_json_decode(hass, aioclient_mock):
"""Test decoding invalid json result."""
aioclient_mock.get(
"https://www.start.ca/support/usage/api?key=" "NOTAKEY", text="this is not xml"
@ -220,5 +214,5 @@ def test_bad_json_decode(hass, aioclient_mock):
scd = StartcaData(hass.loop, async_get_clientsession(hass), "NOTAKEY", 400)
result = yield from scd.async_update()
result = await scd.async_update()
assert result is False