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

133
tests/components/updater/test_init.py
View File

@@ -9,55 +9,53 @@ from homeassistant.setup import async_setup_component
from homeassistant.components import updater
import homeassistant.util.dt as dt_util
from tests.common import (
async_fire_time_changed, mock_coro, mock_component, MockDependency)
async_fire_time_changed,
mock_coro,
mock_component,
MockDependency,
)
NEW_VERSION = '10000.0'
MOCK_VERSION = '10.0'
MOCK_DEV_VERSION = '10.0.dev0'
MOCK_HUUID = 'abcdefg'
MOCK_RESPONSE = {
'version': '0.15',
'release-notes': 'https://home-assistant.io'
}
MOCK_CONFIG = {updater.DOMAIN: {
'reporting': True
}}
NEW_VERSION = "10000.0"
MOCK_VERSION = "10.0"
MOCK_DEV_VERSION = "10.0.dev0"
MOCK_HUUID = "abcdefg"
MOCK_RESPONSE = {"version": "0.15", "release-notes": "https://home-assistant.io"}
MOCK_CONFIG = {updater.DOMAIN: {"reporting": True}}
@pytest.fixture(autouse=True)
def mock_distro():
"""Mock distro dep."""
with MockDependency('distro'):
with MockDependency("distro"):
yield
@pytest.fixture
def mock_get_newest_version():
"""Fixture to mock get_newest_version."""
with patch('homeassistant.components.updater.get_newest_version') as mock:
with patch("homeassistant.components.updater.get_newest_version") as mock:
yield mock
@pytest.fixture
def mock_get_uuid():
"""Fixture to mock get_uuid."""
with patch('homeassistant.components.updater._load_uuid') as mock:
with patch("homeassistant.components.updater._load_uuid") as mock:
yield mock
@asyncio.coroutine
def test_new_version_shows_entity_after_hour(
hass, mock_get_uuid, mock_get_newest_version):
hass, mock_get_uuid, mock_get_newest_version
):
"""Test if new entity is created if new version is available."""
mock_get_uuid.return_value = MOCK_HUUID
mock_get_newest_version.return_value = mock_coro((NEW_VERSION, ''))
mock_get_newest_version.return_value = mock_coro((NEW_VERSION, ""))
res = yield from async_setup_component(
hass, updater.DOMAIN, {updater.DOMAIN: {}})
assert res, 'Updater failed to set up'
res = yield from async_setup_component(hass, updater.DOMAIN, {updater.DOMAIN: {}})
assert res, "Updater failed to set up"
with patch('homeassistant.components.updater.current_version',
MOCK_VERSION):
with patch("homeassistant.components.updater.current_version", MOCK_VERSION):
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(hours=1))
yield from hass.async_block_till_done()
@@ -65,18 +63,15 @@ def test_new_version_shows_entity_after_hour(
@asyncio.coroutine
def test_same_version_not_show_entity(
hass, mock_get_uuid, mock_get_newest_version):
def test_same_version_not_show_entity(hass, mock_get_uuid, mock_get_newest_version):
"""Test if new entity is created if new version is available."""
mock_get_uuid.return_value = MOCK_HUUID
mock_get_newest_version.return_value = mock_coro((MOCK_VERSION, ''))
mock_get_newest_version.return_value = mock_coro((MOCK_VERSION, ""))
res = yield from async_setup_component(
hass, updater.DOMAIN, {updater.DOMAIN: {}})
assert res, 'Updater failed to set up'
res = yield from async_setup_component(hass, updater.DOMAIN, {updater.DOMAIN: {}})
assert res, "Updater failed to set up"
with patch('homeassistant.components.updater.current_version',
MOCK_VERSION):
with patch("homeassistant.components.updater.current_version", MOCK_VERSION):
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(hours=1))
yield from hass.async_block_till_done()
@@ -87,16 +82,14 @@ def test_same_version_not_show_entity(
def test_disable_reporting(hass, mock_get_uuid, mock_get_newest_version):
"""Test if new entity is created if new version is available."""
mock_get_uuid.return_value = MOCK_HUUID
mock_get_newest_version.return_value = mock_coro((MOCK_VERSION, ''))
mock_get_newest_version.return_value = mock_coro((MOCK_VERSION, ""))
res = yield from async_setup_component(
hass, updater.DOMAIN, {updater.DOMAIN: {
'reporting': False
}})
assert res, 'Updater failed to set up'
hass, updater.DOMAIN, {updater.DOMAIN: {"reporting": False}}
)
assert res, "Updater failed to set up"
with patch('homeassistant.components.updater.current_version',
MOCK_VERSION):
with patch("homeassistant.components.updater.current_version", MOCK_VERSION):
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(hours=1))
yield from hass.async_block_till_done()
@@ -112,11 +105,11 @@ def test_get_newest_version_no_analytics_when_no_huuid(hass, aioclient_mock):
"""Test we do not gather analytics when no huuid is passed in."""
aioclient_mock.post(updater.UPDATER_URL, json=MOCK_RESPONSE)
with patch('homeassistant.helpers.system_info.async_get_system_info',
side_effect=Exception):
with patch(
"homeassistant.helpers.system_info.async_get_system_info", side_effect=Exception
):
res = yield from updater.get_newest_version(hass, None, False)
assert res == (MOCK_RESPONSE['version'],
MOCK_RESPONSE['release-notes'])
assert res == (MOCK_RESPONSE["version"], MOCK_RESPONSE["release-notes"])
@asyncio.coroutine
@@ -124,19 +117,21 @@ def test_get_newest_version_analytics_when_huuid(hass, aioclient_mock):
"""Test we do not gather analytics when no huuid is passed in."""
aioclient_mock.post(updater.UPDATER_URL, json=MOCK_RESPONSE)
with patch('homeassistant.helpers.system_info.async_get_system_info',
Mock(return_value=mock_coro({'fake': 'bla'}))):
with patch(
"homeassistant.helpers.system_info.async_get_system_info",
Mock(return_value=mock_coro({"fake": "bla"})),
):
res = yield from updater.get_newest_version(hass, MOCK_HUUID, False)
assert res == (MOCK_RESPONSE['version'],
MOCK_RESPONSE['release-notes'])
assert res == (MOCK_RESPONSE["version"], MOCK_RESPONSE["release-notes"])
@asyncio.coroutine
def test_error_fetching_new_version_timeout(hass):
"""Test we do not gather analytics when no huuid is passed in."""
with patch('homeassistant.helpers.system_info.async_get_system_info',
Mock(return_value=mock_coro({'fake': 'bla'}))), \
patch('async_timeout.timeout', side_effect=asyncio.TimeoutError):
with patch(
"homeassistant.helpers.system_info.async_get_system_info",
Mock(return_value=mock_coro({"fake": "bla"})),
), patch("async_timeout.timeout", side_effect=asyncio.TimeoutError):
res = yield from updater.get_newest_version(hass, MOCK_HUUID, False)
assert res is None
@@ -144,10 +139,12 @@ def test_error_fetching_new_version_timeout(hass):
@asyncio.coroutine
def test_error_fetching_new_version_bad_json(hass, aioclient_mock):
"""Test we do not gather analytics when no huuid is passed in."""
aioclient_mock.post(updater.UPDATER_URL, text='not json')
aioclient_mock.post(updater.UPDATER_URL, text="not json")
with patch('homeassistant.helpers.system_info.async_get_system_info',
Mock(return_value=mock_coro({'fake': 'bla'}))):
with patch(
"homeassistant.helpers.system_info.async_get_system_info",
Mock(return_value=mock_coro({"fake": "bla"})),
):
res = yield from updater.get_newest_version(hass, MOCK_HUUID, False)
assert res is None
@@ -155,32 +152,36 @@ def test_error_fetching_new_version_bad_json(hass, aioclient_mock):
@asyncio.coroutine
def test_error_fetching_new_version_invalid_response(hass, aioclient_mock):
"""Test we do not gather analytics when no huuid is passed in."""
aioclient_mock.post(updater.UPDATER_URL, json={
'version': '0.15'
# 'release-notes' is missing
})
aioclient_mock.post(
updater.UPDATER_URL,
json={
"version": "0.15"
# 'release-notes' is missing
},
)
with patch('homeassistant.helpers.system_info.async_get_system_info',
Mock(return_value=mock_coro({'fake': 'bla'}))):
with patch(
"homeassistant.helpers.system_info.async_get_system_info",
Mock(return_value=mock_coro({"fake": "bla"})),
):
res = yield from updater.get_newest_version(hass, MOCK_HUUID, False)
assert res is None
@asyncio.coroutine
def test_new_version_shows_entity_after_hour_hassio(
hass, mock_get_uuid, mock_get_newest_version):
hass, mock_get_uuid, mock_get_newest_version
):
"""Test if new entity is created if new version is available / hass.io."""
mock_get_uuid.return_value = MOCK_HUUID
mock_get_newest_version.return_value = mock_coro((NEW_VERSION, ''))
mock_component(hass, 'hassio')
hass.data['hassio_hass_version'] = "999.0"
mock_get_newest_version.return_value = mock_coro((NEW_VERSION, ""))
mock_component(hass, "hassio")
hass.data["hassio_hass_version"] = "999.0"
res = yield from async_setup_component(
hass, updater.DOMAIN, {updater.DOMAIN: {}})
assert res, 'Updater failed to set up'
res = yield from async_setup_component(hass, updater.DOMAIN, {updater.DOMAIN: {}})
assert res, "Updater failed to set up"
with patch('homeassistant.components.updater.current_version',
MOCK_VERSION):
with patch("homeassistant.components.updater.current_version", MOCK_VERSION):
async_fire_time_changed(hass, dt_util.utcnow() + timedelta(hours=1))
yield from hass.async_block_till_done()