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

Mock all invocations of coronavirus.get_cases (#32487)

Browse source
This commit is contained in:
Paulus Schoutsen 2020-03-04 21:42:07 -08:00 • committed by GitHub
parent 81810dd920
commit 81f99efda1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 23 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/components/coronavirus/test_init.py
View file

@ -1,6 +1,4 @@
"""Test init of Coronavirus integration."""
from asynctest import Mock, patch
from homeassistant.components.coronavirus.const import DOMAIN, OPTION_WORLDWIDE
from homeassistant.helpers import entity_registry
from homeassistant.setup import async_setup_component
@ -33,15 +31,8 @@ async def test_migration(hass):
),
},
)
with patch(
"coronavirus.get_cases",
return_value=[
Mock(country="Netherlands", confirmed=10, recovered=8, deaths=1, current=1),
Mock(country="Germany", confirmed=1, recovered=0, deaths=0, current=0),
],
):
assert await async_setup_component(hass, DOMAIN, {})
await hass.async_block_till_done()
assert await async_setup_component(hass, DOMAIN, {})
await hass.async_block_till_done()
ent_reg = await entity_registry.async_get_registry(hass)