Refactor homekit to use a dataclass for entry data (#101738)

This commit is contained in:
J. Nick Koston 2023-10-10 09:14:37 -10:00 committed by GitHub
parent 6c65db2036
commit 265f6653c3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 52 additions and 38 deletions

View file

@ -14,8 +14,6 @@ from homeassistant.components.homekit.const import (
DOMAIN,
FEATURE_ON_OFF,
FEATURE_PLAY_PAUSE,
HOMEKIT_PAIRING_QR,
HOMEKIT_PAIRING_QR_SECRET,
TYPE_FAUCET,
TYPE_OUTLET,
TYPE_SHOWER,
@ -23,6 +21,7 @@ from homeassistant.components.homekit.const import (
TYPE_SWITCH,
TYPE_VALVE,
)
from homeassistant.components.homekit.models import HomeKitEntryData
from homeassistant.components.homekit.util import (
accessory_friendly_name,
async_dismiss_setup_message,
@ -251,8 +250,9 @@ async def test_async_show_setup_msg(
hass, entry.entry_id, "bridge_name", pincode, "X-HM://0"
)
await hass.async_block_till_done()
assert hass.data[DOMAIN][entry.entry_id][HOMEKIT_PAIRING_QR_SECRET]
assert hass.data[DOMAIN][entry.entry_id][HOMEKIT_PAIRING_QR]
entry_data: HomeKitEntryData = hass.data[DOMAIN][entry.entry_id]
assert entry_data.pairing_qr_secret
assert entry_data.pairing_qr
assert len(mock_create.mock_calls) == 1
assert mock_create.mock_calls[0][1][3] == entry.entry_id