Remove analytics from updater (#48518)

This commit is contained in:
Joakim Sørensen 2021-03-30 22:22:03 +02:00 committed by GitHub
parent 76d9c62900
commit 78cff36f96
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 36 additions and 101 deletions

View File

@ -5,7 +5,6 @@ import logging
import async_timeout import async_timeout
from awesomeversion import AwesomeVersion from awesomeversion import AwesomeVersion
from distro import linux_distribution
import voluptuous as vol import voluptuous as vol
from homeassistant.const import __version__ as current_version from homeassistant.const import __version__ as current_version
@ -23,20 +22,22 @@ CONF_COMPONENT_REPORTING = "include_used_components"
DOMAIN = "updater" DOMAIN = "updater"
UPDATER_URL = "https://updater.home-assistant.io/" UPDATER_URL = "https://www.home-assistant.io/version.json"
CONFIG_SCHEMA = vol.Schema( CONFIG_SCHEMA = vol.Schema(
{ {
DOMAIN: { DOMAIN: {
vol.Optional(CONF_REPORTING, default=True): cv.boolean, vol.Optional(CONF_REPORTING): cv.boolean,
vol.Optional(CONF_COMPONENT_REPORTING, default=False): cv.boolean, vol.Optional(CONF_COMPONENT_REPORTING): cv.boolean,
} }
}, },
extra=vol.ALLOW_EXTRA, extra=vol.ALLOW_EXTRA,
) )
RESPONSE_SCHEMA = vol.Schema( RESPONSE_SCHEMA = vol.Schema(
{vol.Required("version"): cv.string, vol.Required("release-notes"): cv.url} {vol.Required("current_version"): cv.string, vol.Required("release_notes"): cv.url},
extra=vol.REMOVE_EXTRA,
) )
@ -52,30 +53,27 @@ class Updater:
async def async_setup(hass, config): async def async_setup(hass, config):
"""Set up the updater component.""" """Set up the updater component."""
if "dev" in current_version:
# This component only makes sense in release versions
_LOGGER.info("Running on 'dev', only analytics will be submitted")
conf = config.get(DOMAIN, {}) conf = config.get(DOMAIN, {})
if conf.get(CONF_REPORTING):
huuid = await hass.helpers.instance_id.async_get()
else:
huuid = None
include_components = conf.get(CONF_COMPONENT_REPORTING) for option in (CONF_COMPONENT_REPORTING, CONF_REPORTING):
if option in conf:
_LOGGER.warning(
"Analytics reporting with the option '%s' "
"is deprecated and you should remove that from your configuration. "
"The analytics part of this integration has moved to the new 'analytics' integration",
option,
)
async def check_new_version() -> Updater: async def check_new_version() -> Updater:
"""Check if a new version is available and report if one is.""" """Check if a new version is available and report if one is."""
newest, release_notes = await get_newest_version(
hass, huuid, include_components
)
_LOGGER.debug("Fetched version %s: %s", newest, release_notes)
# Skip on dev # Skip on dev
if "dev" in current_version: if "dev" in current_version:
return Updater(False, "", "") return Updater(False, "", "")
newest, release_notes = await get_newest_version(hass)
_LOGGER.debug("Fetched version %s: %s", newest, release_notes)
# Load data from Supervisor # Load data from Supervisor
if hass.components.hassio.is_hassio(): if hass.components.hassio.is_hassio():
core_info = hass.components.hassio.get_core_info() core_info = hass.components.hassio.get_core_info()
@ -121,34 +119,12 @@ async def async_setup(hass, config):
return True return True
async def get_newest_version(hass, huuid, include_components): async def get_newest_version(hass):
"""Get the newest Home Assistant version.""" """Get the newest Home Assistant version."""
if huuid:
info_object = await hass.helpers.system_info.async_get_system_info()
if include_components:
info_object["components"] = list(hass.config.components)
linux_dist = await hass.async_add_executor_job(linux_distribution, False)
info_object["distribution"] = linux_dist[0]
info_object["os_version"] = linux_dist[1]
info_object["huuid"] = huuid
else:
info_object = {}
session = async_get_clientsession(hass) session = async_get_clientsession(hass)
with async_timeout.timeout(30): with async_timeout.timeout(30):
req = await session.post(UPDATER_URL, json=info_object) req = await session.get(UPDATER_URL)
_LOGGER.info(
(
"Submitted analytics to Home Assistant servers. "
"Information submitted includes %s"
),
info_object,
)
try: try:
res = await req.json() res = await req.json()
@ -159,7 +135,7 @@ async def get_newest_version(hass, huuid, include_components):
try: try:
res = RESPONSE_SCHEMA(res) res = RESPONSE_SCHEMA(res)
return res["version"], res["release-notes"] return res["current_version"], res["release_notes"]
except vol.Invalid as err: except vol.Invalid as err:
raise update_coordinator.UpdateFailed( raise update_coordinator.UpdateFailed(
f"Got unexpected response: {err}" f"Got unexpected response: {err}"

View File

@ -1,4 +1,4 @@
"""The tests for the Updater component.""" """The tests for the Updater integration."""
from unittest.mock import patch from unittest.mock import patch
import pytest import pytest
@ -12,8 +12,10 @@ from tests.common import mock_component
NEW_VERSION = "10000.0" NEW_VERSION = "10000.0"
MOCK_VERSION = "10.0" MOCK_VERSION = "10.0"
MOCK_DEV_VERSION = "10.0.dev0" MOCK_DEV_VERSION = "10.0.dev0"
MOCK_HUUID = "abcdefg" MOCK_RESPONSE = {
MOCK_RESPONSE = {"version": "0.15", "release-notes": "https://home-assistant.io"} "current_version": "0.15",
"release_notes": "https://home-assistant.io",
}
MOCK_CONFIG = {updater.DOMAIN: {"reporting": True}} MOCK_CONFIG = {updater.DOMAIN: {"reporting": True}}
RELEASE_NOTES = "test release notes" RELEASE_NOTES = "test release notes"
@ -35,19 +37,8 @@ def mock_get_newest_version_fixture():
yield mock yield mock
@pytest.fixture(name="mock_get_uuid", autouse=True) async def test_new_version_shows_entity_true(hass, mock_get_newest_version):
def mock_get_uuid_fixture():
"""Fixture to mock get_uuid."""
with patch("homeassistant.helpers.instance_id.async_get") as mock:
yield mock
async def test_new_version_shows_entity_true(
hass, mock_get_uuid, mock_get_newest_version
):
"""Test if sensor is true if new version is available.""" """Test if sensor is true if new version is available."""
mock_get_uuid.return_value = MOCK_HUUID
assert await async_setup_component(hass, updater.DOMAIN, {updater.DOMAIN: {}}) assert await async_setup_component(hass, updater.DOMAIN, {updater.DOMAIN: {}})
await hass.async_block_till_done() await hass.async_block_till_done()
@ -62,11 +53,8 @@ async def test_new_version_shows_entity_true(
) )
async def test_same_version_shows_entity_false( async def test_same_version_shows_entity_false(hass, mock_get_newest_version):
hass, mock_get_uuid, mock_get_newest_version
):
"""Test if sensor is false if no new version is available.""" """Test if sensor is false if no new version is available."""
mock_get_uuid.return_value = MOCK_HUUID
mock_get_newest_version.return_value = (MOCK_VERSION, "") mock_get_newest_version.return_value = (MOCK_VERSION, "")
assert await async_setup_component(hass, updater.DOMAIN, {updater.DOMAIN: {}}) assert await async_setup_component(hass, updater.DOMAIN, {updater.DOMAIN: {}})
@ -81,60 +69,32 @@ async def test_same_version_shows_entity_false(
assert "release_notes" not in hass.states.get("binary_sensor.updater").attributes assert "release_notes" not in hass.states.get("binary_sensor.updater").attributes
async def test_disable_reporting(hass, mock_get_uuid, mock_get_newest_version): async def test_deprecated_reporting(hass, mock_get_newest_version, caplog):
"""Test we do not gather analytics when disable reporting is active.""" """Test we do not gather analytics when disable reporting is active."""
mock_get_uuid.return_value = MOCK_HUUID
mock_get_newest_version.return_value = (MOCK_VERSION, "") mock_get_newest_version.return_value = (MOCK_VERSION, "")
assert await async_setup_component( assert await async_setup_component(
hass, updater.DOMAIN, {updater.DOMAIN: {"reporting": False}} hass, updater.DOMAIN, {updater.DOMAIN: {"reporting": True}}
) )
await hass.async_block_till_done() await hass.async_block_till_done()
assert hass.states.is_state("binary_sensor.updater", "off") assert "deprecated" in caplog.text
await updater.get_newest_version(hass, MOCK_HUUID, MOCK_CONFIG)
call = mock_get_newest_version.mock_calls[0][1]
assert call[0] is hass
assert call[1] is None
async 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
):
res = await updater.get_newest_version(hass, None, False)
assert res == (MOCK_RESPONSE["version"], MOCK_RESPONSE["release-notes"])
async def test_get_newest_version_analytics_when_huuid(hass, aioclient_mock):
"""Test we gather analytics when huuid is passed in."""
aioclient_mock.post(updater.UPDATER_URL, json=MOCK_RESPONSE)
with patch(
"homeassistant.helpers.system_info.async_get_system_info",
return_value={"fake": "bla"},
):
res = await updater.get_newest_version(hass, MOCK_HUUID, False)
assert res == (MOCK_RESPONSE["version"], MOCK_RESPONSE["release-notes"])
async def test_error_fetching_new_version_bad_json(hass, aioclient_mock): async def test_error_fetching_new_version_bad_json(hass, aioclient_mock):
"""Test we handle json error while fetching new version.""" """Test we handle json error while fetching new version."""
aioclient_mock.post(updater.UPDATER_URL, text="not json") aioclient_mock.get(updater.UPDATER_URL, text="not json")
with patch( with patch(
"homeassistant.helpers.system_info.async_get_system_info", "homeassistant.helpers.system_info.async_get_system_info",
return_value={"fake": "bla"}, return_value={"fake": "bla"},
), pytest.raises(UpdateFailed): ), pytest.raises(UpdateFailed):
await updater.get_newest_version(hass, MOCK_HUUID, False) await updater.get_newest_version(hass)
async def test_error_fetching_new_version_invalid_response(hass, aioclient_mock): async def test_error_fetching_new_version_invalid_response(hass, aioclient_mock):
"""Test we handle response error while fetching new version.""" """Test we handle response error while fetching new version."""
aioclient_mock.post( aioclient_mock.get(
updater.UPDATER_URL, updater.UPDATER_URL,
json={ json={
"version": "0.15" "version": "0.15"
@ -146,14 +106,13 @@ async def test_error_fetching_new_version_invalid_response(hass, aioclient_mock)
"homeassistant.helpers.system_info.async_get_system_info", "homeassistant.helpers.system_info.async_get_system_info",
return_value={"fake": "bla"}, return_value={"fake": "bla"},
), pytest.raises(UpdateFailed): ), pytest.raises(UpdateFailed):
await updater.get_newest_version(hass, MOCK_HUUID, False) await updater.get_newest_version(hass)
async def test_new_version_shows_entity_after_hour_hassio( async def test_new_version_shows_entity_after_hour_hassio(
hass, mock_get_uuid, mock_get_newest_version hass, mock_get_newest_version
): ):
"""Test if binary sensor gets updated if new version is available / Hass.io.""" """Test if binary sensor gets updated if new version is available / Hass.io."""
mock_get_uuid.return_value = MOCK_HUUID
mock_component(hass, "hassio") mock_component(hass, "hassio")
hass.data["hassio_core_info"] = {"version_latest": "999.0"} hass.data["hassio_core_info"] = {"version_latest": "999.0"}