mirror of
https://github.com/home-assistant/core.git
synced 2025-12-20 19:09:45 +00:00
296 lines
8.9 KiB
Python
296 lines
8.9 KiB
Python
"""Test configuration and mocks for the SmartThings component."""
|
|
|
|
from collections.abc import Generator
|
|
import time
|
|
from typing import Any
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from pysmartthings import (
|
|
DeviceHealth,
|
|
DeviceResponse,
|
|
DeviceStatus,
|
|
LocationResponse,
|
|
RoomResponse,
|
|
SceneResponse,
|
|
Subscription,
|
|
)
|
|
from pysmartthings.models import HealthStatus, InstalledApp
|
|
import pytest
|
|
|
|
from homeassistant.components.application_credentials import (
|
|
ClientCredential,
|
|
async_import_client_credential,
|
|
)
|
|
from homeassistant.components.smartthings import CONF_INSTALLED_APP_ID, OLD_DATA
|
|
from homeassistant.components.smartthings.const import (
|
|
CONF_LOCATION_ID,
|
|
CONF_REFRESH_TOKEN,
|
|
DOMAIN,
|
|
SCOPES,
|
|
)
|
|
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_CLIENT_ID, CONF_CLIENT_SECRET
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry, load_fixture
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[AsyncMock]:
|
|
"""Override async_setup_entry."""
|
|
with patch(
|
|
"homeassistant.components.smartthings.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
yield mock_setup_entry
|
|
|
|
|
|
@pytest.fixture(name="expires_at")
|
|
def mock_expires_at() -> int:
|
|
"""Fixture to set the oauth token expiration time."""
|
|
return time.time() + 3600
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def setup_credentials(hass: HomeAssistant) -> None:
|
|
"""Fixture to setup credentials."""
|
|
assert await async_setup_component(hass, "application_credentials", {})
|
|
await async_import_client_credential(
|
|
hass,
|
|
DOMAIN,
|
|
ClientCredential("CLIENT_ID", "CLIENT_SECRET"),
|
|
DOMAIN,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_smartthings() -> Generator[AsyncMock]:
|
|
"""Mock a SmartThings client."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.smartthings.SmartThings",
|
|
autospec=True,
|
|
) as mock_client,
|
|
patch(
|
|
"homeassistant.components.smartthings.config_flow.SmartThings",
|
|
new=mock_client,
|
|
),
|
|
):
|
|
client = mock_client.return_value
|
|
client.get_scenes.return_value = SceneResponse.from_json(
|
|
load_fixture("scenes.json", DOMAIN)
|
|
).items
|
|
client.get_locations.return_value = LocationResponse.from_json(
|
|
load_fixture("locations.json", DOMAIN)
|
|
).items
|
|
client.get_rooms.return_value = RoomResponse.from_json(
|
|
load_fixture("rooms.json", DOMAIN)
|
|
).items
|
|
client.create_subscription.return_value = Subscription.from_json(
|
|
load_fixture("subscription.json", DOMAIN)
|
|
)
|
|
client.get_device_health.return_value = DeviceHealth.from_json(
|
|
load_fixture("device_health.json", DOMAIN)
|
|
)
|
|
client.get_installed_app.return_value = InstalledApp.from_json(
|
|
load_fixture("installed_app.json", DOMAIN)
|
|
)
|
|
yield client
|
|
|
|
|
|
@pytest.fixture(
|
|
params=[
|
|
"aq_sensor_3_ikea",
|
|
"aeotec_ms6",
|
|
"da_ac_airsensor_01001",
|
|
"da_ac_rac_000001",
|
|
"da_ac_rac_000003",
|
|
"da_ac_rac_100001",
|
|
"da_ac_rac_01001",
|
|
"da_ac_cac_01001",
|
|
"multipurpose_sensor",
|
|
"contact_sensor",
|
|
"base_electric_meter",
|
|
"smart_plug",
|
|
"vd_stv_2017_k",
|
|
"c2c_arlo_pro_3_switch",
|
|
"yale_push_button_deadbolt_lock",
|
|
"ge_in_wall_smart_dimmer",
|
|
"centralite",
|
|
"da_ref_normal_000001",
|
|
"da_ref_normal_01011",
|
|
"da_ref_normal_01011_onedoor",
|
|
"da_ref_normal_01001",
|
|
"vd_network_audio_002s",
|
|
"vd_network_audio_003s",
|
|
"vd_sensor_light_2023",
|
|
"iphone",
|
|
"da_sac_ehs_000001_sub",
|
|
"da_sac_ehs_000001_sub_1",
|
|
"da_sac_ehs_000002_sub",
|
|
"da_ac_ehs_01001",
|
|
"da_wm_dw_000001",
|
|
"da_wm_wd_01011",
|
|
"da_wm_wd_000001",
|
|
"da_wm_wd_000001_1",
|
|
"da_wm_wm_01011",
|
|
"da_wm_wm_100001",
|
|
"da_wm_wm_100002",
|
|
"da_wm_wm_000001",
|
|
"da_wm_wm_000001_1",
|
|
"da_wm_sc_000001",
|
|
"da_wm_dw_01011",
|
|
"da_rvc_normal_000001",
|
|
"da_rvc_map_01011",
|
|
"da_ks_microwave_0101x",
|
|
"da_ks_cooktop_000001",
|
|
"da_ks_cooktop_31001",
|
|
"da_ks_range_0101x",
|
|
"da_ks_oven_01061",
|
|
"da_ks_oven_0107x",
|
|
"da_ks_walloven_0107x",
|
|
"da_ks_hood_01001",
|
|
"hue_color_temperature_bulb",
|
|
"hue_rgbw_color_bulb",
|
|
"c2c_shade",
|
|
"sonos_player",
|
|
"aeotec_home_energy_meter_gen5",
|
|
"virtual_water_sensor",
|
|
"virtual_thermostat",
|
|
"virtual_valve",
|
|
"sensibo_airconditioner_1",
|
|
"ecobee_sensor",
|
|
"ecobee_thermostat",
|
|
"ecobee_thermostat_offline",
|
|
"sensi_thermostat",
|
|
"fake_fan",
|
|
"generic_fan_3_speed",
|
|
"heatit_ztrm3_thermostat",
|
|
"heatit_zpushwall",
|
|
"generic_ef00_v1",
|
|
"gas_detector",
|
|
"bosch_radiator_thermostat_ii",
|
|
"im_speaker_ai_0001",
|
|
"im_smarttag2_ble_uwb",
|
|
"abl_light_b_001",
|
|
"tplink_p110",
|
|
"ikea_kadrilj",
|
|
"aux_ac",
|
|
"hw_q80r_soundbar",
|
|
"gas_meter",
|
|
"lumi",
|
|
"tesla_powerwall",
|
|
]
|
|
)
|
|
def device_fixture(
|
|
mock_smartthings: AsyncMock, request: pytest.FixtureRequest
|
|
) -> Generator[str]:
|
|
"""Return every device."""
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture
|
|
def devices(mock_smartthings: AsyncMock, device_fixture: str) -> Generator[AsyncMock]:
|
|
"""Return a specific device."""
|
|
mock_smartthings.get_devices.return_value = DeviceResponse.from_json(
|
|
load_fixture(f"devices/{device_fixture}.json", DOMAIN)
|
|
).items
|
|
mock_smartthings.get_device_status.return_value = DeviceStatus.from_json(
|
|
load_fixture(f"device_status/{device_fixture}.json", DOMAIN)
|
|
).components
|
|
return mock_smartthings
|
|
|
|
|
|
@pytest.fixture
|
|
def unavailable_device(devices: AsyncMock) -> AsyncMock:
|
|
"""Mock an unavailable device."""
|
|
devices.get_device_health.return_value.state = HealthStatus.OFFLINE
|
|
return devices
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry(expires_at: int) -> MockConfigEntry:
|
|
"""Mock a config entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="My home",
|
|
unique_id="397678e5-9995-4a39-9d9f-ae6ba310236c",
|
|
data={
|
|
"auth_implementation": DOMAIN,
|
|
"token": {
|
|
"access_token": "mock-access-token",
|
|
"refresh_token": "mock-refresh-token",
|
|
"expires_at": expires_at,
|
|
"scope": " ".join(SCOPES),
|
|
"access_tier": 0,
|
|
"installed_app_id": "5aaaa925-2be1-4e40-b257-e4ef59083324",
|
|
},
|
|
CONF_LOCATION_ID: "397678e5-9995-4a39-9d9f-ae6ba310236c",
|
|
CONF_INSTALLED_APP_ID: "123",
|
|
},
|
|
version=3,
|
|
minor_version=3,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def old_data() -> dict[str, Any]:
|
|
"""Return old data for config entry."""
|
|
return {
|
|
OLD_DATA: {
|
|
CONF_ACCESS_TOKEN: "mock-access-token",
|
|
CONF_REFRESH_TOKEN: "mock-refresh-token",
|
|
CONF_CLIENT_ID: "CLIENT_ID",
|
|
CONF_CLIENT_SECRET: "CLIENT_SECRET",
|
|
CONF_LOCATION_ID: "397678e5-9995-4a39-9d9f-ae6ba310236c",
|
|
CONF_INSTALLED_APP_ID: "123aa123-2be1-4e40-b257-e4ef59083324",
|
|
}
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_migrated_config_entry(
|
|
expires_at: int, old_data: dict[str, Any]
|
|
) -> MockConfigEntry:
|
|
"""Mock a config entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="My home",
|
|
unique_id="397678e5-9995-4a39-9d9f-ae6ba310236c",
|
|
data={
|
|
"auth_implementation": DOMAIN,
|
|
"token": {
|
|
"access_token": "mock-access-token",
|
|
"refresh_token": "mock-refresh-token",
|
|
"expires_at": expires_at,
|
|
"scope": " ".join(SCOPES),
|
|
"access_tier": 0,
|
|
"installed_app_id": "5aaaa925-2be1-4e40-b257-e4ef59083324",
|
|
},
|
|
CONF_LOCATION_ID: "397678e5-9995-4a39-9d9f-ae6ba310236c",
|
|
CONF_INSTALLED_APP_ID: "123",
|
|
**old_data,
|
|
},
|
|
version=3,
|
|
minor_version=2,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_old_config_entry() -> MockConfigEntry:
|
|
"""Mock the old config entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title="My home",
|
|
unique_id="appid123-2be1-4e40-b257-e4ef59083324_397678e5-9995-4a39-9d9f-ae6ba310236c",
|
|
data={
|
|
CONF_ACCESS_TOKEN: "mock-access-token",
|
|
CONF_REFRESH_TOKEN: "mock-refresh-token",
|
|
CONF_CLIENT_ID: "CLIENT_ID",
|
|
CONF_CLIENT_SECRET: "CLIENT_SECRET",
|
|
CONF_LOCATION_ID: "397678e5-9995-4a39-9d9f-ae6ba310236c",
|
|
CONF_INSTALLED_APP_ID: "123aa123-2be1-4e40-b257-e4ef59083324",
|
|
},
|
|
version=2,
|
|
)
|