mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 13:17:32 +00:00
Improve test coverage for ESPHome manager (#120400)
This commit is contained in:
parent
76e890865e
commit
c15718519b
@ -263,6 +263,7 @@ async def _mock_generic_device_entry(
|
|||||||
mock_list_entities_services: tuple[list[EntityInfo], list[UserService]],
|
mock_list_entities_services: tuple[list[EntityInfo], list[UserService]],
|
||||||
states: list[EntityState],
|
states: list[EntityState],
|
||||||
entry: MockConfigEntry | None = None,
|
entry: MockConfigEntry | None = None,
|
||||||
|
hass_storage: dict[str, Any] | None = None,
|
||||||
) -> MockESPHomeDevice:
|
) -> MockESPHomeDevice:
|
||||||
if not entry:
|
if not entry:
|
||||||
entry = MockConfigEntry(
|
entry = MockConfigEntry(
|
||||||
@ -286,6 +287,17 @@ async def _mock_generic_device_entry(
|
|||||||
}
|
}
|
||||||
device_info = DeviceInfo(**(default_device_info | mock_device_info))
|
device_info = DeviceInfo(**(default_device_info | mock_device_info))
|
||||||
|
|
||||||
|
if hass_storage:
|
||||||
|
storage_key = f"{DOMAIN}.{entry.entry_id}"
|
||||||
|
hass_storage[storage_key] = {
|
||||||
|
"version": 1,
|
||||||
|
"minor_version": 1,
|
||||||
|
"key": storage_key,
|
||||||
|
"data": {
|
||||||
|
"device_info": device_info.to_dict(),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
mock_device = MockESPHomeDevice(entry, mock_client, device_info)
|
mock_device = MockESPHomeDevice(entry, mock_client, device_info)
|
||||||
|
|
||||||
def _subscribe_states(callback: Callable[[EntityState], None]) -> None:
|
def _subscribe_states(callback: Callable[[EntityState], None]) -> None:
|
||||||
@ -453,6 +465,7 @@ async def mock_bluetooth_entry_with_legacy_adv(
|
|||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
async def mock_generic_device_entry(
|
async def mock_generic_device_entry(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
|
hass_storage: dict[str, Any],
|
||||||
) -> Callable[
|
) -> Callable[
|
||||||
[APIClient, list[EntityInfo], list[UserService], list[EntityState]],
|
[APIClient, list[EntityInfo], list[UserService], list[EntityState]],
|
||||||
Awaitable[MockConfigEntry],
|
Awaitable[MockConfigEntry],
|
||||||
@ -464,10 +477,17 @@ async def mock_generic_device_entry(
|
|||||||
entity_info: list[EntityInfo],
|
entity_info: list[EntityInfo],
|
||||||
user_service: list[UserService],
|
user_service: list[UserService],
|
||||||
states: list[EntityState],
|
states: list[EntityState],
|
||||||
|
mock_storage: bool = False,
|
||||||
) -> MockConfigEntry:
|
) -> MockConfigEntry:
|
||||||
return (
|
return (
|
||||||
await _mock_generic_device_entry(
|
await _mock_generic_device_entry(
|
||||||
hass, mock_client, {}, (entity_info, user_service), states
|
hass,
|
||||||
|
mock_client,
|
||||||
|
{},
|
||||||
|
(entity_info, user_service),
|
||||||
|
states,
|
||||||
|
None,
|
||||||
|
hass_storage if mock_storage else None,
|
||||||
)
|
)
|
||||||
).entry
|
).entry
|
||||||
|
|
||||||
@ -477,6 +497,7 @@ async def mock_generic_device_entry(
|
|||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
async def mock_esphome_device(
|
async def mock_esphome_device(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
|
hass_storage: dict[str, Any],
|
||||||
) -> Callable[
|
) -> Callable[
|
||||||
[APIClient, list[EntityInfo], list[UserService], list[EntityState]],
|
[APIClient, list[EntityInfo], list[UserService], list[EntityState]],
|
||||||
Awaitable[MockESPHomeDevice],
|
Awaitable[MockESPHomeDevice],
|
||||||
@ -485,19 +506,21 @@ async def mock_esphome_device(
|
|||||||
|
|
||||||
async def _mock_device(
|
async def _mock_device(
|
||||||
mock_client: APIClient,
|
mock_client: APIClient,
|
||||||
entity_info: list[EntityInfo],
|
entity_info: list[EntityInfo] | None = None,
|
||||||
user_service: list[UserService],
|
user_service: list[UserService] | None = None,
|
||||||
states: list[EntityState],
|
states: list[EntityState] | None = None,
|
||||||
entry: MockConfigEntry | None = None,
|
entry: MockConfigEntry | None = None,
|
||||||
device_info: dict[str, Any] | None = None,
|
device_info: dict[str, Any] | None = None,
|
||||||
|
mock_storage: bool = False,
|
||||||
) -> MockESPHomeDevice:
|
) -> MockESPHomeDevice:
|
||||||
return await _mock_generic_device_entry(
|
return await _mock_generic_device_entry(
|
||||||
hass,
|
hass,
|
||||||
mock_client,
|
mock_client,
|
||||||
device_info or {},
|
device_info or {},
|
||||||
(entity_info, user_service),
|
(entity_info or [], user_service or []),
|
||||||
states,
|
states or [],
|
||||||
entry,
|
entry,
|
||||||
|
hass_storage if mock_storage else None,
|
||||||
)
|
)
|
||||||
|
|
||||||
return _mock_device
|
return _mock_device
|
||||||
|
@ -1156,3 +1156,28 @@ async def test_start_reauth(
|
|||||||
assert len(flows) == 1
|
assert len(flows) == 1
|
||||||
flow = flows[0]
|
flow = flows[0]
|
||||||
assert flow["context"]["source"] == "reauth"
|
assert flow["context"]["source"] == "reauth"
|
||||||
|
|
||||||
|
|
||||||
|
async def test_entry_missing_unique_id(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
mock_client: APIClient,
|
||||||
|
mock_esphome_device: Callable[
|
||||||
|
[APIClient, list[EntityInfo], list[UserService], list[EntityState]],
|
||||||
|
Awaitable[MockESPHomeDevice],
|
||||||
|
],
|
||||||
|
) -> None:
|
||||||
|
"""Test the unique id is added from storage if available."""
|
||||||
|
entry = MockConfigEntry(
|
||||||
|
domain=DOMAIN,
|
||||||
|
unique_id=None,
|
||||||
|
data={
|
||||||
|
CONF_HOST: "test.local",
|
||||||
|
CONF_PORT: 6053,
|
||||||
|
CONF_PASSWORD: "",
|
||||||
|
},
|
||||||
|
options={CONF_ALLOW_SERVICE_CALLS: True},
|
||||||
|
)
|
||||||
|
entry.add_to_hass(hass)
|
||||||
|
await mock_esphome_device(mock_client=mock_client, mock_storage=True)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
assert entry.unique_id == "11:22:33:44:55:aa"
|
||||||
|
Loading…
x
Reference in New Issue
Block a user