7d5c1ede72
* Allow host/ipv6 address for broadlink service This matches switch config and is a regression fix * Restore padding of packets for broadlink * Drop unused import * Fix comment on test
108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
"""The tests for the broadlink component."""
|
|
from datetime import timedelta
|
|
from base64 import b64decode
|
|
from unittest.mock import MagicMock, patch, call
|
|
|
|
import pytest
|
|
|
|
from homeassistant.util.dt import utcnow
|
|
from homeassistant.components.broadlink import async_setup_service, data_packet
|
|
from homeassistant.components.broadlink.const import (
|
|
DOMAIN, SERVICE_LEARN, SERVICE_SEND)
|
|
|
|
DUMMY_IR_PACKET = ("JgBGAJKVETkRORA6ERQRFBEUERQRFBE5ETkQOhAVEBUQFREUEBUQ"
|
|
"OhEUERQRORE5EBURFBA6EBUQOhE5EBUQFRA6EDoRFBEADQUAAA==")
|
|
DUMMY_HOST = "192.168.0.2"
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def dummy_broadlink():
|
|
"""Mock broadlink module so we don't have that dependency on tests."""
|
|
broadlink = MagicMock()
|
|
with patch.dict('sys.modules', {
|
|
'broadlink': broadlink,
|
|
}):
|
|
yield broadlink
|
|
|
|
|
|
async def test_padding(hass):
|
|
"""Verify that non padding strings are allowed."""
|
|
assert data_packet('Jg') == b'&'
|
|
assert data_packet('Jg=') == b'&'
|
|
assert data_packet('Jg==') == b'&'
|
|
|
|
|
|
async def test_send(hass):
|
|
"""Test send service."""
|
|
mock_device = MagicMock()
|
|
mock_device.send_data.return_value = None
|
|
|
|
async_setup_service(hass, DUMMY_HOST, mock_device)
|
|
await hass.async_block_till_done()
|
|
|
|
await hass.services.async_call(DOMAIN, SERVICE_SEND, {
|
|
"host": DUMMY_HOST,
|
|
"packet": (DUMMY_IR_PACKET)
|
|
})
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_device.send_data.call_count == 1
|
|
assert mock_device.send_data.call_args == call(
|
|
b64decode(DUMMY_IR_PACKET))
|
|
|
|
|
|
async def test_learn(hass):
|
|
"""Test learn service."""
|
|
mock_device = MagicMock()
|
|
mock_device.enter_learning.return_value = None
|
|
mock_device.check_data.return_value = b64decode(DUMMY_IR_PACKET)
|
|
|
|
with patch.object(hass.components.persistent_notification,
|
|
'async_create') as mock_create:
|
|
|
|
async_setup_service(hass, DUMMY_HOST, mock_device)
|
|
await hass.async_block_till_done()
|
|
|
|
await hass.services.async_call(DOMAIN, SERVICE_LEARN, {
|
|
"host": DUMMY_HOST,
|
|
})
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_device.enter_learning.call_count == 1
|
|
assert mock_device.enter_learning.call_args == call()
|
|
|
|
assert mock_create.call_count == 1
|
|
assert mock_create.call_args == call(
|
|
"Received packet is: {}".format(DUMMY_IR_PACKET),
|
|
title='Broadlink switch')
|
|
|
|
|
|
async def test_learn_timeout(hass):
|
|
"""Test learn service."""
|
|
mock_device = MagicMock()
|
|
mock_device.enter_learning.return_value = None
|
|
mock_device.check_data.return_value = None
|
|
|
|
async_setup_service(hass, DUMMY_HOST, mock_device)
|
|
await hass.async_block_till_done()
|
|
|
|
now = utcnow()
|
|
|
|
with patch.object(hass.components.persistent_notification,
|
|
'async_create') as mock_create, \
|
|
patch('homeassistant.components.broadlink.utcnow') as mock_utcnow:
|
|
|
|
mock_utcnow.side_effect = [now, now + timedelta(20)]
|
|
|
|
await hass.services.async_call(DOMAIN, SERVICE_LEARN, {
|
|
"host": DUMMY_HOST,
|
|
})
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_device.enter_learning.call_count == 1
|
|
assert mock_device.enter_learning.call_args == call()
|
|
|
|
assert mock_create.call_count == 1
|
|
assert mock_create.call_args == call(
|
|
"No signal was received",
|
|
title='Broadlink switch')
|