diff --git a/tests/components/accuweather/test_sensor.py b/tests/components/accuweather/test_sensor.py index 69035d63990..f23cc413991 100644 --- a/tests/components/accuweather/test_sensor.py +++ b/tests/components/accuweather/test_sensor.py @@ -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, ) diff --git a/tests/components/accuweather/test_weather.py b/tests/components/accuweather/test_weather.py index 7e163e40d83..8635d30d300 100644 --- a/tests/components/accuweather/test_weather.py +++ b/tests/components/accuweather/test_weather.py @@ -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, ) diff --git a/tests/components/airly/test_sensor.py b/tests/components/airly/test_sensor.py index 970ec4e0e2b..9da10553ddd 100644 --- a/tests/components/airly/test_sensor.py +++ b/tests/components/airly/test_sensor.py @@ -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, ) diff --git a/tests/components/apcupsd/test_sensor.py b/tests/components/apcupsd/test_sensor.py index 07ac0c90c04..97ea126ccd8 100644 --- a/tests/components/apcupsd/test_sensor.py +++ b/tests/components/apcupsd/test_sensor.py @@ -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", diff --git a/tests/components/devialet/test_media_player.py b/tests/components/devialet/test_media_player.py index fd593a10a98..ab90bd92407 100644 --- a/tests/components/devialet/test_media_player.py +++ b/tests/components/devialet/test_media_player.py @@ -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, diff --git a/tests/components/easyenergy/test_diagnostics.py b/tests/components/easyenergy/test_diagnostics.py index 8b9d850d98c..3820078a00b 100644 --- a/tests/components/easyenergy/test_diagnostics.py +++ b/tests/components/easyenergy/test_diagnostics.py @@ -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, diff --git a/tests/components/easyenergy/test_sensor.py b/tests/components/easyenergy/test_sensor.py index afc3a12d6a2..5ca4740dc6f 100644 --- a/tests/components/easyenergy/test_sensor.py +++ b/tests/components/easyenergy/test_sensor.py @@ -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, diff --git a/tests/components/flo/test_sensor.py b/tests/components/flo/test_sensor.py index 828e4f3b4d5..8c12da7c679 100644 --- a/tests/components/flo/test_sensor.py +++ b/tests/components/flo/test_sensor.py @@ -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, ) diff --git a/tests/components/google_weather/test_sensor.py b/tests/components/google_weather/test_sensor.py index e96089b17c7..73d170e71fd 100644 --- a/tests/components/google_weather/test_sensor.py +++ b/tests/components/google_weather/test_sensor.py @@ -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, ) diff --git a/tests/components/google_weather/test_weather.py b/tests/components/google_weather/test_weather.py index 128bcee02ac..35ab591de6e 100644 --- a/tests/components/google_weather/test_weather.py +++ b/tests/components/google_weather/test_weather.py @@ -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, ) diff --git a/tests/components/hassio/test_init.py b/tests/components/hassio/test_init.py index 7825ba19f91..d9ff4362609 100644 --- a/tests/components/hassio/test_init.py +++ b/tests/components/hassio/test_init.py @@ -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", diff --git a/tests/components/homeassistant_hardware/test_update.py b/tests/components/homeassistant_hardware/test_update.py index cfd19d94bdb..cd2298dfb43 100644 --- a/tests/components/homeassistant_hardware/test_update.py +++ b/tests/components/homeassistant_hardware/test_update.py @@ -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, ) diff --git a/tests/components/modbus/test_cover.py b/tests/components/modbus/test_cover.py index 9f3a64c27e5..915395b41db 100644 --- a/tests/components/modbus/test_cover.py +++ b/tests/components/modbus/test_cover.py @@ -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, ) diff --git a/tests/components/nam/test_sensor.py b/tests/components/nam/test_sensor.py index c1681537c95..dd0fb8d8028 100644 --- a/tests/components/nam/test_sensor.py +++ b/tests/components/nam/test_sensor.py @@ -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, ) diff --git a/tests/components/nws/test_weather.py b/tests/components/nws/test_weather.py index bbf808dbd1f..d59943fdf3f 100644 --- a/tests/components/nws/test_weather.py +++ b/tests/components/nws/test_weather.py @@ -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, ) diff --git a/tests/components/rituals_perfume_genie/test_number.py b/tests/components/rituals_perfume_genie/test_number.py index ddca70649b5..3d0e9ab09e2 100644 --- a/tests/components/rituals_perfume_genie/test_number.py +++ b/tests/components/rituals_perfume_genie/test_number.py @@ -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, diff --git a/tests/components/rituals_perfume_genie/test_select.py b/tests/components/rituals_perfume_genie/test_select.py index a4d97ab83fd..10c9bcf037d 100644 --- a/tests/components/rituals_perfume_genie/test_select.py +++ b/tests/components/rituals_perfume_genie/test_select.py @@ -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, diff --git a/tests/components/rituals_perfume_genie/test_switch.py b/tests/components/rituals_perfume_genie/test_switch.py index 7e6a94906e1..5e0ae292db9 100644 --- a/tests/components/rituals_perfume_genie/test_switch.py +++ b/tests/components/rituals_perfume_genie/test_switch.py @@ -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, diff --git a/tests/components/sharkiq/test_vacuum.py b/tests/components/sharkiq/test_vacuum.py index eab08c12ca1..b8004d01081 100644 --- a/tests/components/sharkiq/test_vacuum.py +++ b/tests/components/sharkiq/test_vacuum.py @@ -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() diff --git a/tests/components/unifiprotect/test_camera.py b/tests/components/unifiprotect/test_camera.py index e721e6b8688..47ffaf59ef5 100644 --- a/tests/components/unifiprotect/test_camera.py +++ b/tests/components/unifiprotect/test_camera.py @@ -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, ) diff --git a/tests/components/yeelight/test_light.py b/tests/components/yeelight/test_light.py index 2da47ef7167..9e8e00c900d 100644 --- a/tests/components/yeelight/test_light.py +++ b/tests/components/yeelight/test_light.py @@ -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, ) diff --git a/tests/components/zha/test_number.py b/tests/components/zha/test_number.py index 111e852b3e6..4f6d0afecc2 100644 --- a/tests/components/zha/test_number.py +++ b/tests/components/zha/test_number.py @@ -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] diff --git a/tests/components/zha/test_switch.py b/tests/components/zha/test_switch.py index ef7ff4e50c5..5b22ec8343e 100644 --- a/tests/components/zha/test_switch.py +++ b/tests/components/zha/test_switch.py @@ -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(