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

Use import instead of hass in Shelly tests (#67909)

Browse source
This commit is contained in:
Shay Levy 2022-03-09 20:29:20 +02:00 • committed by GitHub
parent e02c21a4de
commit eee0c5372c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/components/shelly/test_cover.py
View file

@ -13,6 +13,7 @@ from homeassistant.components.cover import (
STATE_OPENING,
)
from homeassistant.const import ATTR_ENTITY_ID, STATE_UNKNOWN
from homeassistant.helpers.entity_component import async_update_entity
ROLLER_BLOCK_ID = 1
@ -71,12 +72,12 @@ async def test_block_device_update(hass, coap_wrapper, monkeypatch):
await hass.async_block_till_done()
monkeypatch.setattr(coap_wrapper.device.blocks[ROLLER_BLOCK_ID], "rollerPos", 0)
await hass.helpers.entity_component.async_update_entity("cover.test_name")
await async_update_entity(hass, "cover.test_name")
await hass.async_block_till_done()
assert hass.states.get("cover.test_name").state == STATE_CLOSED
monkeypatch.setattr(coap_wrapper.device.blocks[ROLLER_BLOCK_ID], "rollerPos", 100)
await hass.helpers.entity_component.async_update_entity("cover.test_name")
await async_update_entity(hass, "cover.test_name")
await hass.async_block_till_done()
assert hass.states.get("cover.test_name").state == STATE_OPEN
@ -165,12 +166,12 @@ async def test_rpc_device_update(hass, rpc_wrapper, monkeypatch):
await hass.async_block_till_done()
monkeypatch.setitem(rpc_wrapper.device.status["cover:0"], "state", "closed")
await hass.helpers.entity_component.async_update_entity("cover.test_cover_0")
await async_update_entity(hass, "cover.test_cover_0")
await hass.async_block_till_done()
assert hass.states.get("cover.test_cover_0").state == STATE_CLOSED
monkeypatch.setitem(rpc_wrapper.device.status["cover:0"], "state", "open")
await hass.helpers.entity_component.async_update_entity("cover.test_cover_0")
await async_update_entity(hass, "cover.test_cover_0")
await hass.async_block_till_done()
assert hass.states.get("cover.test_cover_0").state == STATE_OPEN
@ -186,6 +187,6 @@ async def test_rpc_device_no_position_control(hass, rpc_wrapper, monkeypatch):
)
await hass.async_block_till_done()
await hass.helpers.entity_component.async_update_entity("cover.test_cover_0")
await async_update_entity(hass, "cover.test_cover_0")
await hass.async_block_till_done()
assert hass.states.get("cover.test_cover_0").state == STATE_UNKNOWN

9
tests/components/shelly/test_switch.py
View file

@ -7,6 +7,7 @@ from homeassistant.const import (
STATE_OFF,
STATE_ON,
)
from homeassistant.helpers.entity_component import async_update_entity
RELAY_BLOCK_ID = 0
@ -47,16 +48,12 @@ async def test_block_device_update(hass, coap_wrapper, monkeypatch):
await hass.async_block_till_done()
monkeypatch.setattr(coap_wrapper.device.blocks[RELAY_BLOCK_ID], "output", False)
await hass.helpers.entity_component.async_update_entity(
"switch.test_name_channel_1"
)
await async_update_entity(hass, "switch.test_name_channel_1")
await hass.async_block_till_done()
assert hass.states.get("switch.test_name_channel_1").state == STATE_OFF
monkeypatch.setattr(coap_wrapper.device.blocks[RELAY_BLOCK_ID], "output", True)
await hass.helpers.entity_component.async_update_entity(
"switch.test_name_channel_1"
)
await async_update_entity(hass, "switch.test_name_channel_1")
await hass.async_block_till_done()
assert hass.states.get("switch.test_name_channel_1").state == STATE_ON