mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 21:06:19 +00:00
Use setup_test_component_platform helper for update entity component tests instead of hass.components (#114203)
* Use `setup_test_component_platform` helper for update entity component tests instead of `hass.components` * Run ruff format
This commit is contained in:
committed by
GitHub
parent
9a1906322b
commit
dce3bde0ab
@@ -1,159 +0,0 @@
|
||||
"""Provide a mock update platform.
|
||||
|
||||
Call init before using it in your tests to ensure clean test data.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from typing import Any
|
||||
|
||||
from homeassistant.components.update import UpdateEntity, UpdateEntityFeature
|
||||
|
||||
from tests.common import MockEntity
|
||||
|
||||
ENTITIES = []
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class MockUpdateEntity(MockEntity, UpdateEntity):
|
||||
"""Mock UpdateEntity class."""
|
||||
|
||||
@property
|
||||
def auto_update(self) -> bool:
|
||||
"""Indicate if the device or service has auto update enabled."""
|
||||
return self._handle("auto_update")
|
||||
|
||||
@property
|
||||
def installed_version(self) -> str | None:
|
||||
"""Version currently installed and in use."""
|
||||
return self._handle("installed_version")
|
||||
|
||||
@property
|
||||
def in_progress(self) -> bool | int | None:
|
||||
"""Update installation progress."""
|
||||
return self._handle("in_progress")
|
||||
|
||||
@property
|
||||
def latest_version(self) -> str | None:
|
||||
"""Latest version available for install."""
|
||||
return self._handle("latest_version")
|
||||
|
||||
@property
|
||||
def release_summary(self) -> str | None:
|
||||
"""Summary of the release notes or changelog."""
|
||||
return self._handle("release_summary")
|
||||
|
||||
@property
|
||||
def release_url(self) -> str | None:
|
||||
"""URL to the full release notes of the latest version available."""
|
||||
return self._handle("release_url")
|
||||
|
||||
@property
|
||||
def title(self) -> str | None:
|
||||
"""Title of the software."""
|
||||
return self._handle("title")
|
||||
|
||||
def install(self, version: str | None, backup: bool, **kwargs: Any) -> None:
|
||||
"""Install an update."""
|
||||
if backup:
|
||||
_LOGGER.info("Creating backup before installing update")
|
||||
|
||||
if version is not None:
|
||||
self._values["installed_version"] = version
|
||||
_LOGGER.info("Installed update with version: %s", version)
|
||||
else:
|
||||
self._values["installed_version"] = self.latest_version
|
||||
_LOGGER.info("Installed latest update")
|
||||
|
||||
def release_notes(self) -> str | None:
|
||||
"""Return the release notes of the latest version."""
|
||||
return "Release notes"
|
||||
|
||||
|
||||
def init(empty=False):
|
||||
"""Initialize the platform with entities."""
|
||||
global ENTITIES
|
||||
|
||||
ENTITIES = (
|
||||
[]
|
||||
if empty
|
||||
else [
|
||||
MockUpdateEntity(
|
||||
name="No Update",
|
||||
unique_id="no_update",
|
||||
installed_version="1.0.0",
|
||||
latest_version="1.0.0",
|
||||
supported_features=UpdateEntityFeature.INSTALL,
|
||||
),
|
||||
MockUpdateEntity(
|
||||
name="Update Available",
|
||||
unique_id="update_available",
|
||||
installed_version="1.0.0",
|
||||
latest_version="1.0.1",
|
||||
supported_features=UpdateEntityFeature.INSTALL,
|
||||
),
|
||||
MockUpdateEntity(
|
||||
name="Update Unknown",
|
||||
unique_id="update_unknown",
|
||||
installed_version="1.0.0",
|
||||
latest_version=None,
|
||||
supported_features=UpdateEntityFeature.INSTALL,
|
||||
),
|
||||
MockUpdateEntity(
|
||||
name="Update Specific Version",
|
||||
unique_id="update_specific_version",
|
||||
installed_version="1.0.0",
|
||||
latest_version="1.0.0",
|
||||
supported_features=UpdateEntityFeature.INSTALL
|
||||
| UpdateEntityFeature.SPECIFIC_VERSION,
|
||||
),
|
||||
MockUpdateEntity(
|
||||
name="Update Backup",
|
||||
unique_id="update_backup",
|
||||
installed_version="1.0.0",
|
||||
latest_version="1.0.1",
|
||||
supported_features=UpdateEntityFeature.INSTALL
|
||||
| UpdateEntityFeature.SPECIFIC_VERSION
|
||||
| UpdateEntityFeature.BACKUP,
|
||||
),
|
||||
MockUpdateEntity(
|
||||
name="Update Already in Progress",
|
||||
unique_id="update_already_in_progres",
|
||||
installed_version="1.0.0",
|
||||
latest_version="1.0.1",
|
||||
in_progress=50,
|
||||
supported_features=UpdateEntityFeature.INSTALL
|
||||
| UpdateEntityFeature.PROGRESS,
|
||||
),
|
||||
MockUpdateEntity(
|
||||
name="Update No Install",
|
||||
unique_id="no_install",
|
||||
installed_version="1.0.0",
|
||||
latest_version="1.0.1",
|
||||
),
|
||||
MockUpdateEntity(
|
||||
name="Update with release notes",
|
||||
unique_id="with_release_notes",
|
||||
installed_version="1.0.0",
|
||||
latest_version="1.0.1",
|
||||
supported_features=UpdateEntityFeature.RELEASE_NOTES,
|
||||
),
|
||||
MockUpdateEntity(
|
||||
name="Update with auto update",
|
||||
unique_id="with_auto_update",
|
||||
installed_version="1.0.0",
|
||||
latest_version="1.0.1",
|
||||
auto_update=True,
|
||||
supported_features=UpdateEntityFeature.INSTALL,
|
||||
),
|
||||
]
|
||||
)
|
||||
|
||||
|
||||
async def async_setup_platform(
|
||||
hass, config, async_add_entities_callback, discovery_info=None
|
||||
):
|
||||
"""Return mock entities."""
|
||||
async_add_entities_callback(ENTITIES)
|
||||
Reference in New Issue
Block a user