evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve homekit_controller tests (#65266)

Browse source
This commit is contained in:
Jc2k 2022-01-30 22:59:39 +00:00 • committed by GitHub
parent eb94fe1ca7
commit 58b8c30221
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
16 changed files with 1487 additions and 650 deletions
Download patch file Download diff file Expand all files Collapse all files

53
tests/components/homekit_controller/test_select.py
View file

@ -12,6 +12,7 @@ def create_service_with_ecobee_mode(accessory: Accessory):
current_mode = service.add_char(CharacteristicsTypes.Vendor.ECOBEE_CURRENT_MODE)
current_mode.value = 0
current_mode.perms.append("ev")
service.add_char(CharacteristicsTypes.Vendor.ECOBEE_SET_HOLD_SCHEDULE)
@ -21,10 +22,9 @@ def create_service_with_ecobee_mode(accessory: Accessory):
async def test_read_current_mode(hass, utcnow):
"""Test that Ecobee mode can be correctly read and show as human readable text."""
helper = await setup_test_component(hass, create_service_with_ecobee_mode)
service = helper.accessory.services.first(service_type=ServicesTypes.THERMOSTAT)
# Helper will be for the primary entity, which is the service. Make a helper for the sensor.
energy_helper = Helper(
ecobee_mode = Helper(
hass,
"select.testdevice_current_mode",
helper.pairing,
@ -32,27 +32,38 @@ async def test_read_current_mode(hass, utcnow):
helper.config_entry,
)
mode = service[CharacteristicsTypes.Vendor.ECOBEE_CURRENT_MODE]
state = await energy_helper.poll_and_get_state()
state = await ecobee_mode.async_update(
ServicesTypes.THERMOSTAT,
{
CharacteristicsTypes.Vendor.ECOBEE_CURRENT_MODE: 0,
},
)
assert state.state == "home"
mode.value = 1
state = await energy_helper.poll_and_get_state()
state = await ecobee_mode.async_update(
ServicesTypes.THERMOSTAT,
{
CharacteristicsTypes.Vendor.ECOBEE_CURRENT_MODE: 1,
},
)
assert state.state == "sleep"
mode.value = 2
state = await energy_helper.poll_and_get_state()
state = await ecobee_mode.async_update(
ServicesTypes.THERMOSTAT,
{
CharacteristicsTypes.Vendor.ECOBEE_CURRENT_MODE: 2,
},
)
assert state.state == "away"
async def test_write_current_mode(hass, utcnow):
"""Test can set a specific mode."""
helper = await setup_test_component(hass, create_service_with_ecobee_mode)
service = helper.accessory.services.first(service_type=ServicesTypes.THERMOSTAT)
helper.accessory.services.first(service_type=ServicesTypes.THERMOSTAT)
# Helper will be for the primary entity, which is the service. Make a helper for the sensor.
energy_helper = Helper(
current_mode = Helper(
hass,
"select.testdevice_current_mode",
helper.pairing,
@ -60,18 +71,16 @@ async def test_write_current_mode(hass, utcnow):
helper.config_entry,
)
service = energy_helper.accessory.services.first(
service_type=ServicesTypes.THERMOSTAT
)
mode = service[CharacteristicsTypes.Vendor.ECOBEE_SET_HOLD_SCHEDULE]
await hass.services.async_call(
"select",
"select_option",
{"entity_id": "select.testdevice_current_mode", "option": "home"},
blocking=True,
)
assert mode.value == 0
current_mode.async_assert_service_values(
ServicesTypes.THERMOSTAT,
{CharacteristicsTypes.Vendor.ECOBEE_SET_HOLD_SCHEDULE: 0},
)
await hass.services.async_call(
"select",
@ -79,7 +88,10 @@ async def test_write_current_mode(hass, utcnow):
{"entity_id": "select.testdevice_current_mode", "option": "sleep"},
blocking=True,
)
assert mode.value == 1
current_mode.async_assert_service_values(
ServicesTypes.THERMOSTAT,
{CharacteristicsTypes.Vendor.ECOBEE_SET_HOLD_SCHEDULE: 1},
)
await hass.services.async_call(
"select",
@ -87,4 +99,7 @@ async def test_write_current_mode(hass, utcnow):
{"entity_id": "select.testdevice_current_mode", "option": "away"},
blocking=True,
)
assert mode.value == 2
current_mode.async_assert_service_values(
ServicesTypes.THERMOSTAT,
{CharacteristicsTypes.Vendor.ECOBEE_SET_HOLD_SCHEDULE: 2},
)