302e631984
* added the ability to mock a "long poll" get request by setting up the waiting request and feeding responses to it with this, refactored the qwikswitch test so it doesn't use global variables and is more understandable and maintainable * added import asyncio from merge * added assert that first call with long_poll has empty content * passing json instead of the binary string * use json instead of text in mock requests * refactored to use a proxy * return the proxy also for the http methods other than get * refactored so any side_effect can be used and created the long_poll side effect * simplified by using kwargs needed to move the json->text->content logic from mocker to mockrequest * no need to explicitly define method and url
108 lines
3.1 KiB
Python
108 lines
3.1 KiB
Python
"""Test qwikswitch sensors."""
|
|
import asyncio
|
|
import logging
|
|
|
|
from homeassistant.components.qwikswitch import DOMAIN as QWIKSWITCH
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_START
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.test_util.aiohttp import MockLongPollSideEffect
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
DEVICES = [
|
|
{
|
|
"id": "@000001",
|
|
"name": "Switch 1",
|
|
"type": "rel",
|
|
"val": "OFF",
|
|
"time": "1522777506",
|
|
"rssi": "51%",
|
|
},
|
|
{
|
|
"id": "@000002",
|
|
"name": "Light 2",
|
|
"type": "rel",
|
|
"val": "ON",
|
|
"time": "1522777507",
|
|
"rssi": "45%",
|
|
},
|
|
{
|
|
"id": "@000003",
|
|
"name": "Dim 3",
|
|
"type": "dim",
|
|
"val": "280c00",
|
|
"time": "1522777544",
|
|
"rssi": "62%",
|
|
},
|
|
]
|
|
|
|
|
|
async def test_binary_sensor_device(hass, aioclient_mock):
|
|
"""Test a binary sensor device."""
|
|
config = {
|
|
"qwikswitch": {
|
|
"sensors": {"name": "s1", "id": "@a00001", "channel": 1, "type": "imod"}
|
|
}
|
|
}
|
|
aioclient_mock.get("http://127.0.0.1:2020/&device", json=DEVICES)
|
|
listen_mock = MockLongPollSideEffect()
|
|
aioclient_mock.get("http://127.0.0.1:2020/&listen", side_effect=listen_mock)
|
|
await async_setup_component(hass, QWIKSWITCH, config)
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
|
|
await hass.async_block_till_done()
|
|
|
|
state_obj = hass.states.get("binary_sensor.s1")
|
|
assert state_obj.state == "off"
|
|
|
|
listen_mock.queue_response(
|
|
json={"id": "@a00001", "cmd": "", "data": "4e0e1601", "rssi": "61%"}
|
|
)
|
|
await asyncio.sleep(0.01)
|
|
await hass.async_block_till_done()
|
|
state_obj = hass.states.get("binary_sensor.s1")
|
|
assert state_obj.state == "on"
|
|
|
|
listen_mock.queue_response(
|
|
json={"id": "@a00001", "cmd": "", "data": "4e0e1701", "rssi": "61%"},
|
|
)
|
|
await asyncio.sleep(0.01)
|
|
await hass.async_block_till_done()
|
|
state_obj = hass.states.get("binary_sensor.s1")
|
|
assert state_obj.state == "off"
|
|
|
|
listen_mock.stop()
|
|
|
|
|
|
async def test_sensor_device(hass, aioclient_mock):
|
|
"""Test a sensor device."""
|
|
config = {
|
|
"qwikswitch": {
|
|
"sensors": {
|
|
"name": "ss1",
|
|
"id": "@a00001",
|
|
"channel": 1,
|
|
"type": "qwikcord",
|
|
}
|
|
}
|
|
}
|
|
aioclient_mock.get("http://127.0.0.1:2020/&device", json=DEVICES)
|
|
listen_mock = MockLongPollSideEffect()
|
|
aioclient_mock.get("http://127.0.0.1:2020/&listen", side_effect=listen_mock)
|
|
await async_setup_component(hass, QWIKSWITCH, config)
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
|
|
await hass.async_block_till_done()
|
|
|
|
state_obj = hass.states.get("sensor.ss1")
|
|
assert state_obj.state == "None"
|
|
|
|
listen_mock.queue_response(
|
|
json={"id": "@a00001", "name": "ss1", "type": "rel", "val": "4733800001a00000"},
|
|
)
|
|
await asyncio.sleep(0.01)
|
|
await hass.async_block_till_done()
|
|
state_obj = hass.states.get("sensor.ss1")
|
|
assert state_obj.state == "416"
|
|
|
|
listen_mock.stop()
|