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

41
tests/components/onboarding/test_init.py
View File

@@ -13,14 +13,10 @@ from . import mock_storage
async def test_not_setup_views_if_onboarded(hass, hass_storage):
"""Test if onboarding is done, we don't setup views."""
mock_storage(hass_storage, {
'done': onboarding.STEPS
})
mock_storage(hass_storage, {"done": onboarding.STEPS})
with patch(
'homeassistant.components.onboarding.views.async_setup'
) as mock_setup:
assert await async_setup_component(hass, 'onboarding', {})
with patch("homeassistant.components.onboarding.views.async_setup") as mock_setup:
assert await async_setup_component(hass, "onboarding", {})
assert len(mock_setup.mock_calls) == 0
assert onboarding.DOMAIN not in hass.data
@@ -30,10 +26,10 @@ async def test_not_setup_views_if_onboarded(hass, hass_storage):
async def test_setup_views_if_not_onboarded(hass):
"""Test if onboarding is not done, we setup views."""
with patch(
'homeassistant.components.onboarding.views.async_setup',
return_value=mock_coro()
"homeassistant.components.onboarding.views.async_setup",
return_value=mock_coro(),
) as mock_setup:
assert await async_setup_component(hass, 'onboarding', {})
assert await async_setup_component(hass, "onboarding", {})
assert len(mock_setup.mock_calls) == 1
assert onboarding.DOMAIN in hass.data
@@ -51,9 +47,7 @@ async def test_is_onboarded():
hass.data[onboarding.DOMAIN] = True
assert onboarding.async_is_onboarded(hass)
hass.data[onboarding.DOMAIN] = {
'done': []
}
hass.data[onboarding.DOMAIN] = {"done": []}
assert not onboarding.async_is_onboarded(hass)
@@ -67,9 +61,7 @@ async def test_is_user_onboarded():
hass.data[onboarding.DOMAIN] = True
assert onboarding.async_is_user_onboarded(hass)
hass.data[onboarding.DOMAIN] = {
'done': []
}
hass.data[onboarding.DOMAIN] = {"done": []}
assert not onboarding.async_is_user_onboarded(hass)
@@ -78,25 +70,20 @@ async def test_having_owner_finishes_user_step(hass, hass_storage):
MockUser(is_owner=True).add_to_hass(hass)
with patch(
'homeassistant.components.onboarding.views.async_setup'
) as mock_setup, patch.object(onboarding, 'STEPS', [onboarding.STEP_USER]):
assert await async_setup_component(hass, 'onboarding', {})
"homeassistant.components.onboarding.views.async_setup"
) as mock_setup, patch.object(onboarding, "STEPS", [onboarding.STEP_USER]):
assert await async_setup_component(hass, "onboarding", {})
assert len(mock_setup.mock_calls) == 0
assert onboarding.DOMAIN not in hass.data
assert onboarding.async_is_onboarded(hass)
done = hass_storage[onboarding.STORAGE_KEY]['data']['done']
done = hass_storage[onboarding.STORAGE_KEY]["data"]["done"]
assert onboarding.STEP_USER in done
async def test_migration(hass, hass_storage):
"""Test migrating onboarding to new version."""
hass_storage[onboarding.STORAGE_KEY] = {
'version': 1,
'data': {
'done': ["user"]
}
}
assert await async_setup_component(hass, 'onboarding', {})
hass_storage[onboarding.STORAGE_KEY] = {"version": 1, "data": {"done": ["user"]}}
assert await async_setup_component(hass, "onboarding", {})
assert onboarding.async_is_onboarded(hass)