Electric-Special/ha-core
1
0
Fork 0
mirror of https://github.com/Electric-Special/ha-core.git synced 2026-03-21 05:06:13 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use constants for update_entity calls in tests (#162920)

Browse Source
This commit is contained in:
epenet
2026-02-13 13:54:40 +01:00
committed by GitHub
parent d1aadb5842
commit fd3258a6d3
23 changed files with 183 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/components/accuweather/test_sensor.py
View File

@@ -12,6 +12,10 @@ from homeassistant.components.accuweather.const import (
UPDATE_INTERVAL_DAILY_FORECAST,
UPDATE_INTERVAL_OBSERVATION,
)
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.const import (
ATTR_ENTITY_ID,
ATTR_UNIT_OF_MEASUREMENT,
@@ -133,13 +137,13 @@ async def test_manual_update_entity(
"""Test manual update entity via service homeassistant/update_entity."""
await init_integration(hass)
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
assert mock_accuweather_client.async_get_current_conditions.call_count == 1
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["sensor.home_cloud_ceiling"]},
blocking=True,
)

10
tests/components/accuweather/test_weather.py
View File

@@ -8,6 +8,10 @@ import pytest
from syrupy.assertion import SnapshotAssertion
from homeassistant.components.accuweather.const import UPDATE_INTERVAL_DAILY_FORECAST
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.weather import (
ATTR_FORECAST_CONDITION,
DOMAIN as WEATHER_DOMAIN,
@@ -78,13 +82,13 @@ async def test_manual_update_entity(
"""Test manual update entity via service homeassistant/update_entity."""
await init_integration(hass)
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
assert mock_accuweather_client.async_get_current_conditions.call_count == 1
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["weather.home"]},
blocking=True,
)

10
tests/components/airly/test_sensor.py
View File

@@ -8,6 +8,10 @@ from airly.exceptions import AirlyError
from syrupy.assertion import SnapshotAssertion
from homeassistant.components.airly.const import DOMAIN
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.const import ATTR_ENTITY_ID, STATE_UNAVAILABLE, Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
@@ -83,10 +87,10 @@ async def test_manual_update_entity(
await init_integration(hass, aioclient_mock)
call_count = aioclient_mock.call_count
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["sensor.home_humidity"]},
blocking=True,
)

10
tests/components/apcupsd/test_sensor.py
View File

@@ -9,6 +9,10 @@ from syrupy.assertion import SnapshotAssertion
from homeassistant.components import automation, script
from homeassistant.components.apcupsd.const import DEPRECATED_SENSORS, DOMAIN
from homeassistant.components.apcupsd.coordinator import REQUEST_REFRESH_COOLDOWN
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.const import (
ATTR_ENTITY_ID,
STATE_UNAVAILABLE,
@@ -101,7 +105,7 @@ async def test_manual_update_entity(
assert state.state == "14.0"
# Setup HASS for calling the update_entity service.
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
mock_request_status.return_value = MOCK_STATUS | {
"LOADPCT": "15.0 Percent",
@@ -113,8 +117,8 @@ async def test_manual_update_entity(
future = utcnow() + timedelta(seconds=REQUEST_REFRESH_COOLDOWN)
async_fire_time_changed(hass, future)
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
ATTR_ENTITY_ID: [
f"sensor.{device_slug}_load",

9
tests/components/devialet/test_media_player.py
View File

@@ -7,7 +7,10 @@ from devialet.const import UrlSuffix
from yarl import URL
from homeassistant.components.devialet.media_player import SUPPORT_DEVIALET
from homeassistant.components.homeassistant import SERVICE_UPDATE_ENTITY
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.media_player import (
ATTR_INPUT_SOURCE,
ATTR_INPUT_SOURCE_LIST,
@@ -104,13 +107,13 @@ async def test_media_player_playing(
hass: HomeAssistant, aioclient_mock: AiohttpClientMocker
) -> None:
"""Test the Devialet configuration entry loading and unloading."""
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
entry = await setup_integration(hass, aioclient_mock)
assert entry.state is ConfigEntryState.LOADED
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: [f"{MP_DOMAIN}.{NAME.lower()}"]},
blocking=True,

9
tests/components/easyenergy/test_diagnostics.py
View File

@@ -6,7 +6,10 @@ from easyenergy import EasyEnergyNoDataError
import pytest
from syrupy.assertion import SnapshotAssertion
from homeassistant.components.homeassistant import SERVICE_UPDATE_ENTITY
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
@@ -39,11 +42,11 @@ async def test_diagnostics_no_gas_today(
snapshot: SnapshotAssertion,
) -> None:
"""Test diagnostics, no gas sensors available."""
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
mock_easyenergy.gas_prices.side_effect = EasyEnergyNoDataError
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["sensor.easyenergy_today_gas_current_hour_price"]},
blocking=True,

9
tests/components/easyenergy/test_sensor.py
View File

@@ -6,7 +6,10 @@ from easyenergy import EasyEnergyNoDataError
import pytest
from homeassistant.components.easyenergy.const import DOMAIN
from homeassistant.components.homeassistant import SERVICE_UPDATE_ENTITY
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.sensor import (
ATTR_STATE_CLASS,
SensorDeviceClass,
@@ -303,12 +306,12 @@ async def test_no_gas_today(
hass: HomeAssistant, mock_easyenergy: MagicMock, init_integration: MockConfigEntry
) -> None:
"""Test the easyEnergy - No gas data available."""
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
mock_easyenergy.gas_prices.side_effect = EasyEnergyNoDataError
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: "sensor.easyenergy_today_gas_current_hour_price"},
blocking=True,

10
tests/components/flo/test_sensor.py
View File

@@ -2,6 +2,10 @@
import pytest
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorStateClass
from homeassistant.const import ATTR_ENTITY_ID
from homeassistant.core import HomeAssistant
@@ -91,12 +95,12 @@ async def test_manual_update_entity(
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
call_count = aioclient_mock.call_count
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["sensor.smart_water_shutoff_current_system_mode"]},
blocking=True,
)

10
tests/components/google_weather/test_sensor.py
View File

@@ -8,6 +8,10 @@ from google_weather_api import GoogleWeatherApiError
import pytest
from syrupy.assertion import SnapshotAssertion
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.const import (
ATTR_ENTITY_ID,
ATTR_UNIT_OF_MEASUREMENT,
@@ -92,15 +96,15 @@ async def test_manual_update_entity(
"""Test manual update entity via service homeassistant/update_entity."""
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
mock_google_weather_api.async_get_current_conditions.assert_called_once_with(
latitude=10.1, longitude=20.1
)
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["sensor.home_temperature"]},
blocking=True,
)

10
tests/components/google_weather/test_weather.py
View File

@@ -9,6 +9,10 @@ import pytest
from syrupy.assertion import SnapshotAssertion
from homeassistant.components.google_weather.weather import _CONDITION_MAP
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.weather import (
ATTR_CONDITION_CLEAR_NIGHT,
ATTR_CONDITION_PARTLYCLOUDY,
@@ -91,7 +95,7 @@ async def test_manual_update_entity(
"""Test manual update entity via service homeassistant/update_entity."""
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
assert mock_google_weather_api.async_get_current_conditions.call_count == 1
mock_google_weather_api.async_get_current_conditions.assert_called_with(
@@ -99,8 +103,8 @@ async def test_manual_update_entity(
)
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["weather.home"]},
blocking=True,
)

54
tests/components/hassio/test_init.py
View File

@@ -25,6 +25,10 @@ from homeassistant.components.hassio.const import (
HASSIO_UPDATE_INTERVAL,
REQUEST_REFRESH_DELAY,
)
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, issue_registry as ir
@@ -991,7 +995,7 @@ async def test_coordinator_updates(
hass: HomeAssistant, caplog: pytest.LogCaptureFixture, supervisor_client: AsyncMock
) -> None:
"""Test coordinator updates."""
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
with patch.dict(os.environ, MOCK_ENVIRON):
config_entry = MockConfigEntry(domain=DOMAIN, data={}, unique_id=DOMAIN)
config_entry.add_to_hass(hass)
@@ -1008,8 +1012,8 @@ async def test_coordinator_updates(
supervisor_client.refresh_updates.assert_not_called()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",
@@ -1030,8 +1034,8 @@ async def test_coordinator_updates(
supervisor_client.refresh_updates.reset_mock()
supervisor_client.refresh_updates.side_effect = SupervisorError("Unknown")
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",
@@ -1056,7 +1060,7 @@ async def test_coordinator_updates_stats_entities_enabled(
supervisor_client: AsyncMock,
) -> None:
"""Test coordinator updates with stats entities enabled."""
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
with patch.dict(os.environ, MOCK_ENVIRON):
config_entry = MockConfigEntry(domain=DOMAIN, data={}, unique_id=DOMAIN)
config_entry.add_to_hass(hass)
@@ -1079,8 +1083,8 @@ async def test_coordinator_updates_stats_entities_enabled(
supervisor_client.refresh_updates.assert_not_called()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",
@@ -1100,8 +1104,8 @@ async def test_coordinator_updates_stats_entities_enabled(
supervisor_client.refresh_updates.reset_mock()
supervisor_client.refresh_updates.side_effect = SupervisorError("Unknown")
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",
@@ -1204,7 +1208,7 @@ async def test_deprecated_installation_issue_os_armv7(
),
patch("homeassistant.components.hardware.async_setup", return_value=True),
):
assert await async_setup_component(hass, "homeassistant", {})
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
config_entry = MockConfigEntry(domain=DOMAIN, data={}, unique_id=DOMAIN)
config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(config_entry.entry_id)
@@ -1213,8 +1217,8 @@ async def test_deprecated_installation_issue_os_armv7(
async_fire_time_changed(hass)
await hass.async_block_till_done()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",
@@ -1267,7 +1271,7 @@ async def test_deprecated_installation_issue_32bit_os(
),
patch("homeassistant.components.hardware.async_setup", return_value=True),
):
assert await async_setup_component(hass, "homeassistant", {})
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
config_entry = MockConfigEntry(domain=DOMAIN, data={}, unique_id=DOMAIN)
config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(config_entry.entry_id)
@@ -1276,8 +1280,8 @@ async def test_deprecated_installation_issue_32bit_os(
async_fire_time_changed(hass)
await hass.async_block_till_done()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",
@@ -1328,7 +1332,7 @@ async def test_deprecated_installation_issue_32bit_supervised(
),
patch("homeassistant.components.hardware.async_setup", return_value=True),
):
assert await async_setup_component(hass, "homeassistant", {})
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
config_entry = MockConfigEntry(domain=DOMAIN, data={}, unique_id=DOMAIN)
config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(config_entry.entry_id)
@@ -1337,8 +1341,8 @@ async def test_deprecated_installation_issue_32bit_supervised(
async_fire_time_changed(hass)
await hass.async_block_till_done()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",
@@ -1393,7 +1397,7 @@ async def test_deprecated_installation_issue_64bit_supervised(
),
patch("homeassistant.components.hardware.async_setup", return_value=True),
):
assert await async_setup_component(hass, "homeassistant", {})
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
config_entry = MockConfigEntry(domain=DOMAIN, data={}, unique_id=DOMAIN)
config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(config_entry.entry_id)
@@ -1402,8 +1406,8 @@ async def test_deprecated_installation_issue_64bit_supervised(
async_fire_time_changed(hass)
await hass.async_block_till_done()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",
@@ -1454,7 +1458,7 @@ async def test_deprecated_installation_issue_supported_board(
return_value={"hassos": True, "arch": "aarch64"},
),
):
assert await async_setup_component(hass, "homeassistant", {})
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
config_entry = MockConfigEntry(domain=DOMAIN, data={}, unique_id=DOMAIN)
config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(config_entry.entry_id)
@@ -1463,8 +1467,8 @@ async def test_deprecated_installation_issue_supported_board(
async_fire_time_changed(hass)
await hass.async_block_till_done()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{
"entity_id": [
"update.home_assistant_core_update",

14
tests/components/homeassistant_hardware/test_update.py
View File

@@ -11,6 +11,10 @@ from unittest.mock import Mock, patch
import aiohttp
import pytest
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.homeassistant_hardware.coordinator import (
FirmwareUpdateCoordinator,
)
@@ -227,7 +231,7 @@ async def mock_update_config_entry(
hass: HomeAssistant,
) -> AsyncGenerator[ConfigEntry]:
"""Set up a mock Home Assistant Hardware firmware update entity."""
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
await async_setup_component(hass, "homeassistant_hardware", {})
mock_integration(
@@ -398,8 +402,8 @@ async def test_update_entity_installation_failure(
await hass.async_block_till_done()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{"entity_id": TEST_UPDATE_ENTITY_ID},
blocking=True,
)
@@ -442,8 +446,8 @@ async def test_update_entity_installation_probe_failure(
await hass.async_block_till_done()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{"entity_id": TEST_UPDATE_ENTITY_ID},
blocking=True,
)

2
tests/components/modbus/test_cover.py
View File

@@ -187,7 +187,7 @@ async def test_service_cover_update(hass: HomeAssistant, mock_modbus_ha) -> None
"""Run test for service homeassistant.update_entity."""
await hass.services.async_call(
HOMEASSISTANT_DOMAIN,
"update_entity",
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ENTITY_ID},
blocking=True,
)

10
tests/components/nam/test_sensor.py
View File

@@ -9,6 +9,10 @@ import pytest
from syrupy.assertion import SnapshotAssertion
from tenacity import RetryError
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.nam.const import DEFAULT_UPDATE_INTERVAL, DOMAIN
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN, SensorDeviceClass
from homeassistant.const import (
@@ -151,7 +155,7 @@ async def test_manual_update_entity(hass: HomeAssistant) -> None:
await init_integration(hass)
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
update_response = Mock(json=AsyncMock(return_value=nam_data))
with (
@@ -162,8 +166,8 @@ async def test_manual_update_entity(hass: HomeAssistant) -> None:
) as mock_get_data,
):
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["sensor.nettigo_air_monitor_bme280_temperature"]},
blocking=True,
)

10
tests/components/nws/test_weather.py
View File

@@ -9,6 +9,10 @@ import pytest
from syrupy.assertion import SnapshotAssertion
from homeassistant.components import nws
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.nws.const import (
DEFAULT_SCAN_INTERVAL,
OBSERVATION_VALID_TIME,
@@ -267,7 +271,7 @@ async def test_entity_refresh(hass: HomeAssistant, mock_simple_nws, no_sensor) -
"""Test manual refresh."""
instance = mock_simple_nws.return_value
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
entry = MockConfigEntry(
domain=nws.DOMAIN,
@@ -281,8 +285,8 @@ async def test_entity_refresh(hass: HomeAssistant, mock_simple_nws, no_sensor) -
instance.update_forecast_hourly.assert_called_once()
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{"entity_id": "weather.abc"},
blocking=True,
)

19
tests/components/rituals_perfume_genie/test_number.py
View File

@@ -4,7 +4,10 @@ from __future__ import annotations
import pytest
from homeassistant.components.homeassistant import SERVICE_UPDATE_ENTITY
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.number import (
ATTR_MAX,
ATTR_MIN,
@@ -50,7 +53,7 @@ async def test_set_number_value(hass: HomeAssistant) -> None:
config_entry = mock_config_entry(unique_id="number_set_value_test")
diffuser = mock_diffuser_v1_battery_cartridge()
await init_integration(hass, config_entry, [diffuser])
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
diffuser.perfume_amount = 1
state = hass.states.get("number.genie_perfume_amount")
@@ -64,7 +67,7 @@ async def test_set_number_value(hass: HomeAssistant) -> None:
blocking=True,
)
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["number.genie_perfume_amount"]},
blocking=True,
@@ -81,7 +84,7 @@ async def test_set_number_value_out_of_range(hass: HomeAssistant) -> None:
config_entry = mock_config_entry(unique_id="number_set_value_out_of_range_test")
diffuser = mock_diffuser(hublot="lot123", perfume_amount=2)
await init_integration(hass, config_entry, [diffuser])
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
state = hass.states.get("number.genie_perfume_amount")
assert state
@@ -95,7 +98,7 @@ async def test_set_number_value_out_of_range(hass: HomeAssistant) -> None:
blocking=True,
)
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["number.genie_perfume_amount"]},
blocking=True,
@@ -114,7 +117,7 @@ async def test_set_number_value_out_of_range(hass: HomeAssistant) -> None:
blocking=True,
)
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["number.genie_perfume_amount"]},
blocking=True,
@@ -131,7 +134,7 @@ async def test_set_number_value_to_float(hass: HomeAssistant) -> None:
config_entry = mock_config_entry(unique_id="number_set_value_to_float_test")
diffuser = mock_diffuser(hublot="lot123", perfume_amount=3)
await init_integration(hass, config_entry, [diffuser])
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
state = hass.states.get("number.genie_perfume_amount")
assert state
@@ -145,7 +148,7 @@ async def test_set_number_value_to_float(hass: HomeAssistant) -> None:
blocking=True,
)
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["number.genie_perfume_amount"]},
blocking=True,

13
tests/components/rituals_perfume_genie/test_select.py
View File

@@ -2,7 +2,10 @@
import pytest
from homeassistant.components.homeassistant import SERVICE_UPDATE_ENTITY
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.select import (
ATTR_OPTION,
ATTR_OPTIONS,
@@ -47,7 +50,7 @@ async def test_select_option(hass: HomeAssistant) -> None:
config_entry = mock_config_entry(unique_id="select_invalid_option_test")
diffuser = mock_diffuser(hublot="lot123", room_size_square_meter=60)
await init_integration(hass, config_entry, [diffuser])
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
diffuser.room_size_square_meter = 30
state = hass.states.get("select.genie_room_size")
@@ -61,7 +64,7 @@ async def test_select_option(hass: HomeAssistant) -> None:
blocking=True,
)
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["select.genie_room_size"]},
blocking=True,
@@ -78,7 +81,7 @@ async def test_select_invalid_option(hass: HomeAssistant) -> None:
config_entry = mock_config_entry(unique_id="select_invalid_option_test")
diffuser = mock_diffuser(hublot="lot123", room_size_square_meter=60)
await init_integration(hass, config_entry, [diffuser])
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
state = hass.states.get("select.genie_room_size")
assert state
@@ -92,7 +95,7 @@ async def test_select_invalid_option(hass: HomeAssistant) -> None:
blocking=True,
)
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["select.genie_room_size"]},
blocking=True,

9
tests/components/rituals_perfume_genie/test_switch.py
View File

@@ -2,7 +2,10 @@
from __future__ import annotations
from homeassistant.components.homeassistant import SERVICE_UPDATE_ENTITY
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.rituals_perfume_genie.const import DOMAIN
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.const import (
@@ -45,7 +48,7 @@ async def test_switch_handle_coordinator_update(hass: HomeAssistant) -> None:
config_entry = mock_config_entry(unique_id="switch_handle_coordinator_update_test")
diffuser = mock_diffuser_v1_battery_cartridge()
await init_integration(hass, config_entry, [diffuser])
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
coordinator = hass.data[DOMAIN][config_entry.entry_id]["lot123v1"]
diffuser.is_on = False
@@ -56,7 +59,7 @@ async def test_switch_handle_coordinator_update(hass: HomeAssistant) -> None:
call_count_before_update = diffuser.update_data.call_count
await hass.services.async_call(
"homeassistant",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ["switch.genie"]},
blocking=True,

9
tests/components/sharkiq/test_vacuum.py
View File

@@ -14,7 +14,10 @@ from sharkiq import AylaApi, SharkIqAuthError, SharkIqNotAuthedError, SharkIqVac
from voluptuous.error import MultipleInvalid
from homeassistant import exceptions
from homeassistant.components.homeassistant import SERVICE_UPDATE_ENTITY
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.sharkiq.const import ATTR_ROOMS, DOMAIN
from homeassistant.components.sharkiq.services import SERVICE_CLEAN_ROOM
from homeassistant.components.sharkiq.vacuum import (
@@ -289,14 +292,14 @@ async def test_coordinator_updates(
"""Test the update coordinator update functions."""
coordinator = hass.data[DOMAIN][ENTRY_ID]
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
with patch.object(
MockShark, "async_update", side_effect=side_effect
) as mock_update:
data = {ATTR_ENTITY_ID: [VAC_ENTITY_ID]}
await hass.services.async_call(
"homeassistant", SERVICE_UPDATE_ENTITY, data, blocking=True
HOMEASSISTANT_DOMAIN, SERVICE_UPDATE_ENTITY, data, blocking=True
)
assert coordinator.last_update_success == success
mock_update.assert_called_once()

10
tests/components/unifiprotect/test_camera.py
View File

@@ -23,6 +23,10 @@ from homeassistant.components.camera import (
async_register_webrtc_provider,
get_camera_from_entity_id,
)
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.unifiprotect.const import (
ATTR_BITRATE,
ATTR_CHANNEL_ID,
@@ -430,15 +434,15 @@ async def test_camera_generic_update(
assert_entity_counts(hass, Platform.CAMERA, 2, 1)
entity_id = "camera.test_camera_high_resolution_channel"
assert await async_setup_component(hass, "homeassistant", {})
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
state = hass.states.get(entity_id)
assert state and state.state == "idle"
ufp.api.update = AsyncMock(return_value=None)
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: entity_id},
blocking=True,
)

10
tests/components/yeelight/test_light.py
View File

@@ -21,6 +21,10 @@ from yeelight import (
from yeelight.flow import Action, Flow
from yeelight.main import _MODEL_SPECS
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
ATTR_BRIGHTNESS_PCT,
@@ -139,7 +143,7 @@ SUPPORT_YEELIGHT = (
async def test_services(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -> None:
"""Test Yeelight services."""
assert await async_setup_component(hass, "homeassistant", {})
assert await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
config_entry = MockConfigEntry(
domain=DOMAIN,
data={
@@ -487,8 +491,8 @@ async def test_services(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -
mocked_bulb.last_properties["power"] = "off"
mocked_bulb.available = True
await hass.services.async_call(
"homeassistant",
"update_entity",
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{ATTR_ENTITY_ID: ENTITY_LIGHT},
blocking=True,
)

11
tests/components/zha/test_number.py
View File

@@ -10,6 +10,10 @@ from zigpy.typing import UNDEFINED
from zigpy.zcl.clusters import general
import zigpy.zcl.foundation as zcl_f
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.number import DOMAIN as NUMBER_DOMAIN
from homeassistant.components.zha.helpers import (
ZHADeviceProxy,
@@ -130,11 +134,14 @@ async def test_number(
# update device value with failed attribute report
cluster.PLUGGED_ATTR_READS["present_value"] = 40.0
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
await hass.async_block_till_done()
await hass.services.async_call(
"homeassistant", "update_entity", {"entity_id": entity_id}, blocking=True
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{"entity_id": entity_id},
blocking=True,
)
assert hass.states.get(entity_id).state == "40.0"
assert "present_value" in cluster.read_attributes.call_args[0][0]

11
tests/components/zha/test_switch.py
View File

@@ -10,6 +10,10 @@ from zigpy.typing import UNDEFINED
from zigpy.zcl.clusters import general
import zigpy.zcl.foundation as zcl_f
from homeassistant.components.homeassistant import (
DOMAIN as HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
)
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.components.zha.helpers import (
ZHADeviceProxy,
@@ -136,11 +140,14 @@ async def test_switch(
assert state
assert state.state == STATE_OFF
await async_setup_component(hass, "homeassistant", {})
await async_setup_component(hass, HOMEASSISTANT_DOMAIN, {})
cluster.read_attributes.reset_mock()
await hass.services.async_call(
"homeassistant", "update_entity", {"entity_id": entity_id}, blocking=True
HOMEASSISTANT_DOMAIN,
SERVICE_UPDATE_ENTITY,
{"entity_id": entity_id},
blocking=True,
)
assert len(cluster.read_attributes.mock_calls) == 1
assert cluster.read_attributes.call_args == call(