From 332cbbd8b1be773593037d293c5dabbf6c100199 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Thu, 2 Jan 2020 21:22:30 +0100 Subject: [PATCH] Migrate freedns tests from coroutine to async/await (#30390) --- tests/components/freedns/test_init.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/tests/components/freedns/test_init.py b/tests/components/freedns/test_init.py index b9e59de9ff1..1a64aa498e2 100644 --- a/tests/components/freedns/test_init.py +++ b/tests/components/freedns/test_init.py @@ -1,6 +1,4 @@ """Test the FreeDNS component.""" -import asyncio - import pytest from homeassistant.components import freedns @@ -37,8 +35,7 @@ def setup_freedns(hass, aioclient_mock): ) -@asyncio.coroutine -def test_setup(hass, aioclient_mock): +async def test_setup(hass, aioclient_mock): """Test setup works if update passes.""" params = {} params[ACCESS_TOKEN] = "" @@ -46,7 +43,7 @@ def test_setup(hass, aioclient_mock): UPDATE_URL, params=params, text="ERROR: Address has not changed." ) - result = yield from async_setup_component( + result = await async_setup_component( hass, freedns.DOMAIN, { @@ -60,18 +57,17 @@ def test_setup(hass, aioclient_mock): assert aioclient_mock.call_count == 1 async_fire_time_changed(hass, utcnow() + UPDATE_INTERVAL) - yield from hass.async_block_till_done() + await hass.async_block_till_done() assert aioclient_mock.call_count == 2 -@asyncio.coroutine -def test_setup_fails_if_wrong_token(hass, aioclient_mock): +async def test_setup_fails_if_wrong_token(hass, aioclient_mock): """Test setup fails if first update fails through wrong token.""" params = {} params[ACCESS_TOKEN] = "" aioclient_mock.get(UPDATE_URL, params=params, text="ERROR: Invalid update URL (2)") - result = yield from async_setup_component( + result = await async_setup_component( hass, freedns.DOMAIN, {