Remove analytics from updater (#48518)
This commit is contained in:
parent
76d9c62900
commit
78cff36f96
2 changed files with 36 additions and 101 deletions
|
@ -1,4 +1,4 @@
|
|||
"""The tests for the Updater component."""
|
||||
"""The tests for the Updater integration."""
|
||||
from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
|
@ -12,8 +12,10 @@ from tests.common import mock_component
|
|||
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_RESPONSE = {
|
||||
"current_version": "0.15",
|
||||
"release_notes": "https://home-assistant.io",
|
||||
}
|
||||
MOCK_CONFIG = {updater.DOMAIN: {"reporting": True}}
|
||||
RELEASE_NOTES = "test release notes"
|
||||
|
||||
|
@ -35,19 +37,8 @@ def mock_get_newest_version_fixture():
|
|||
yield mock
|
||||
|
||||
|
||||
@pytest.fixture(name="mock_get_uuid", autouse=True)
|
||||
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
|
||||
):
|
||||
async def test_new_version_shows_entity_true(hass, mock_get_newest_version):
|
||||
"""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: {}})
|
||||
|
||||
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(
|
||||
hass, mock_get_uuid, mock_get_newest_version
|
||||
):
|
||||
async def test_same_version_shows_entity_false(hass, mock_get_newest_version):
|
||||
"""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, "")
|
||||
|
||||
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
|
||||
|
||||
|
||||
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."""
|
||||
mock_get_uuid.return_value = MOCK_HUUID
|
||||
mock_get_newest_version.return_value = (MOCK_VERSION, "")
|
||||
|
||||
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()
|
||||
|
||||
assert hass.states.is_state("binary_sensor.updater", "off")
|
||||
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"])
|
||||
assert "deprecated" in caplog.text
|
||||
|
||||
|
||||
async def test_error_fetching_new_version_bad_json(hass, aioclient_mock):
|
||||
"""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(
|
||||
"homeassistant.helpers.system_info.async_get_system_info",
|
||||
return_value={"fake": "bla"},
|
||||
), 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):
|
||||
"""Test we handle response error while fetching new version."""
|
||||
aioclient_mock.post(
|
||||
aioclient_mock.get(
|
||||
updater.UPDATER_URL,
|
||||
json={
|
||||
"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",
|
||||
return_value={"fake": "bla"},
|
||||
), 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(
|
||||
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."""
|
||||
mock_get_uuid.return_value = MOCK_HUUID
|
||||
mock_component(hass, "hassio")
|
||||
hass.data["hassio_core_info"] = {"version_latest": "999.0"}
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue