mirror of
https://github.com/home-assistant/core.git
synced 2025-07-21 12:17:07 +00:00
Clean up upc_connect tests (#21150)
This commit is contained in:
Martin Hjelmare
Andrew Sayre
parent
0b77a89a2f
commit
0ab9b006f0
@ -1,273 +1,230 @@
|
||||
"""The tests for the UPC ConnextBox device tracker platform."""
|
||||
import asyncio
|
||||
from unittest.mock import patch
|
||||
import logging
|
||||
|
||||
from asynctest import patch
|
||||
import pytest
|
||||
|
||||
from homeassistant.setup import setup_component
|
||||
from homeassistant.const import (
|
||||
CONF_PLATFORM, CONF_HOST)
|
||||
from homeassistant.components.device_tracker import DOMAIN
|
||||
import homeassistant.components.device_tracker.upc_connect as platform
|
||||
from homeassistant.util.async_ import run_coroutine_threadsafe
|
||||
from homeassistant.const import CONF_HOST, CONF_PLATFORM
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
from tests.common import (
|
||||
get_test_home_assistant, assert_setup_component, load_fixture,
|
||||
mock_component, mock_coro)
|
||||
from tests.common import assert_setup_component, load_fixture, mock_component
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
HOST = "127.0.0.1"
|
||||
|
||||
|
||||
@asyncio.coroutine
|
||||
def async_scan_devices_mock(scanner):
|
||||
async def async_scan_devices_mock(scanner):
|
||||
"""Mock async_scan_devices."""
|
||||
return []
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def mock_load_config():
|
||||
"""Mock device tracker loading config."""
|
||||
with patch('homeassistant.components.device_tracker.async_load_config',
|
||||
return_value=mock_coro([])):
|
||||
yield
|
||||
def setup_comp_deps(hass, mock_device_tracker_conf):
|
||||
"""Set up component dependencies."""
|
||||
mock_component(hass, 'zone')
|
||||
mock_component(hass, 'group')
|
||||
yield
|
||||
|
||||
|
||||
class TestUPCConnect:
|
||||
"""Tests for the Ddwrt device tracker platform."""
|
||||
async def test_setup_platform_timeout_loginpage(hass, caplog, aioclient_mock):
|
||||
"""Set up a platform with timeout on loginpage."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
exc=asyncio.TimeoutError()
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
content=b'successful',
|
||||
)
|
||||
|
||||
def setup_method(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
mock_component(self.hass, 'zone')
|
||||
mock_component(self.hass, 'group')
|
||||
assert await async_setup_component(
|
||||
hass, DOMAIN, {
|
||||
DOMAIN: {CONF_PLATFORM: 'upc_connect', CONF_HOST: HOST}})
|
||||
|
||||
self.host = "127.0.0.1"
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
def teardown_method(self):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
assert 'Error setting up platform' in caplog.text
|
||||
|
||||
@patch('homeassistant.components.device_tracker.upc_connect.'
|
||||
'UPCDeviceScanner.async_scan_devices',
|
||||
return_value=async_scan_devices_mock)
|
||||
def test_setup_platform(self, scan_mock, aioclient_mock):
|
||||
"""Set up a platform."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
content=b'successful'
|
||||
)
|
||||
|
||||
with assert_setup_component(1, DOMAIN):
|
||||
assert setup_component(
|
||||
self.hass, DOMAIN, {DOMAIN: {
|
||||
CONF_PLATFORM: 'upc_connect',
|
||||
CONF_HOST: self.host
|
||||
}})
|
||||
async def test_setup_platform_timeout_webservice(hass, caplog, aioclient_mock):
|
||||
"""Set up a platform with api timeout."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
cookies={'sessionToken': '654321'},
|
||||
content=b'successful',
|
||||
exc=asyncio.TimeoutError()
|
||||
)
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert await async_setup_component(
|
||||
hass, DOMAIN, {
|
||||
DOMAIN: {CONF_PLATFORM: 'upc_connect', CONF_HOST: HOST}})
|
||||
|
||||
@patch('homeassistant.components.device_tracker._LOGGER.error')
|
||||
def test_setup_platform_timeout_webservice(self, mock_error,
|
||||
aioclient_mock):
|
||||
"""Set up a platform with api timeout."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
cookies={'sessionToken': '654321'},
|
||||
content=b'successful',
|
||||
exc=asyncio.TimeoutError()
|
||||
)
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
with assert_setup_component(1, DOMAIN):
|
||||
assert setup_component(
|
||||
self.hass, DOMAIN, {DOMAIN: {
|
||||
CONF_PLATFORM: 'upc_connect',
|
||||
CONF_HOST: self.host
|
||||
}})
|
||||
assert 'Error setting up platform' in caplog.text
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
assert 'Error setting up platform' in \
|
||||
str(mock_error.call_args_list[-1])
|
||||
@patch('homeassistant.components.device_tracker.upc_connect.'
|
||||
'UPCDeviceScanner.async_scan_devices',
|
||||
return_value=async_scan_devices_mock)
|
||||
async def test_setup_platform(scan_mock, hass, aioclient_mock):
|
||||
"""Set up a platform."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
content=b'successful'
|
||||
)
|
||||
|
||||
@patch('homeassistant.components.device_tracker._LOGGER.error')
|
||||
def test_setup_platform_timeout_loginpage(self, mock_error,
|
||||
aioclient_mock):
|
||||
"""Set up a platform with timeout on loginpage."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
exc=asyncio.TimeoutError()
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
content=b'successful',
|
||||
)
|
||||
with assert_setup_component(1, DOMAIN):
|
||||
assert await async_setup_component(
|
||||
hass, DOMAIN, {DOMAIN: {
|
||||
CONF_PLATFORM: 'upc_connect',
|
||||
CONF_HOST: HOST
|
||||
}})
|
||||
|
||||
with assert_setup_component(1, DOMAIN):
|
||||
assert setup_component(
|
||||
self.hass, DOMAIN, {DOMAIN: {
|
||||
CONF_PLATFORM: 'upc_connect',
|
||||
CONF_HOST: self.host
|
||||
}})
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
assert 'Error setting up platform' in \
|
||||
str(mock_error.call_args_list[-1])
|
||||
async def test_scan_devices(hass, aioclient_mock):
|
||||
"""Set up a upc platform and scan device."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
content=b'successful',
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
|
||||
def test_scan_devices(self, aioclient_mock):
|
||||
"""Set up a upc platform and scan device."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
content=b'successful',
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
scanner = await platform.async_get_scanner(
|
||||
hass, {
|
||||
DOMAIN: {CONF_PLATFORM: 'upc_connect', CONF_HOST: HOST}})
|
||||
|
||||
scanner = run_coroutine_threadsafe(platform.async_get_scanner(
|
||||
self.hass, {DOMAIN: {
|
||||
CONF_PLATFORM: 'upc_connect',
|
||||
CONF_HOST: self.host
|
||||
}}
|
||||
), self.hass.loop).result()
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
aioclient_mock.clear_requests()
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
text=load_fixture('upc_connect.xml'),
|
||||
cookies={'sessionToken': '1235678'}
|
||||
)
|
||||
|
||||
aioclient_mock.clear_requests()
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
text=load_fixture('upc_connect.xml'),
|
||||
cookies={'sessionToken': '1235678'}
|
||||
)
|
||||
mac_list = await scanner.async_scan_devices()
|
||||
|
||||
mac_list = run_coroutine_threadsafe(
|
||||
scanner.async_scan_devices(), self.hass.loop).result()
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert aioclient_mock.mock_calls[0][2] == 'token=654321&fun=123'
|
||||
assert mac_list == ['30:D3:2D:0:69:21', '5C:AA:FD:25:32:02',
|
||||
'70:EE:50:27:A1:38']
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert aioclient_mock.mock_calls[0][2] == 'token=654321&fun=123'
|
||||
assert mac_list == ['30:D3:2D:0:69:21', '5C:AA:FD:25:32:02',
|
||||
'70:EE:50:27:A1:38']
|
||||
|
||||
def test_scan_devices_without_session(self, aioclient_mock):
|
||||
"""Set up a upc platform and scan device with no token."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
content=b'successful',
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
async def test_scan_devices_without_session(hass, aioclient_mock):
|
||||
"""Set up a upc platform and scan device with no token."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
content=b'successful',
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
|
||||
scanner = run_coroutine_threadsafe(platform.async_get_scanner(
|
||||
self.hass, {DOMAIN: {
|
||||
CONF_PLATFORM: 'upc_connect',
|
||||
CONF_HOST: self.host
|
||||
}}
|
||||
), self.hass.loop).result()
|
||||
scanner = await platform.async_get_scanner(
|
||||
hass, {
|
||||
DOMAIN: {CONF_PLATFORM: 'upc_connect', CONF_HOST: HOST}})
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
aioclient_mock.clear_requests()
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
text=load_fixture('upc_connect.xml'),
|
||||
cookies={'sessionToken': '1235678'}
|
||||
)
|
||||
aioclient_mock.clear_requests()
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
text=load_fixture('upc_connect.xml'),
|
||||
cookies={'sessionToken': '1235678'}
|
||||
)
|
||||
|
||||
scanner.token = None
|
||||
mac_list = run_coroutine_threadsafe(
|
||||
scanner.async_scan_devices(), self.hass.loop).result()
|
||||
scanner.token = None
|
||||
mac_list = await scanner.async_scan_devices()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 2
|
||||
assert aioclient_mock.mock_calls[1][2] == 'token=654321&fun=123'
|
||||
assert mac_list == ['30:D3:2D:0:69:21', '5C:AA:FD:25:32:02',
|
||||
'70:EE:50:27:A1:38']
|
||||
assert len(aioclient_mock.mock_calls) == 2
|
||||
assert aioclient_mock.mock_calls[1][2] == 'token=654321&fun=123'
|
||||
assert mac_list == ['30:D3:2D:0:69:21', '5C:AA:FD:25:32:02',
|
||||
'70:EE:50:27:A1:38']
|
||||
|
||||
def test_scan_devices_without_session_wrong_re(self, aioclient_mock):
|
||||
"""Set up a upc platform and scan device with no token and wrong."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
content=b'successful',
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
|
||||
scanner = run_coroutine_threadsafe(platform.async_get_scanner(
|
||||
self.hass, {DOMAIN: {
|
||||
CONF_PLATFORM: 'upc_connect',
|
||||
CONF_HOST: self.host
|
||||
}}
|
||||
), self.hass.loop).result()
|
||||
async def test_scan_devices_without_session_wrong_re(hass, aioclient_mock):
|
||||
"""Set up a upc platform and scan device with no token and wrong."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
content=b'successful',
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
scanner = await platform.async_get_scanner(
|
||||
hass, {
|
||||
DOMAIN: {CONF_PLATFORM: 'upc_connect', CONF_HOST: HOST}})
|
||||
|
||||
aioclient_mock.clear_requests()
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
status=400,
|
||||
cookies={'sessionToken': '1235678'}
|
||||
)
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
scanner.token = None
|
||||
mac_list = run_coroutine_threadsafe(
|
||||
scanner.async_scan_devices(), self.hass.loop).result()
|
||||
aioclient_mock.clear_requests()
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
status=400,
|
||||
cookies={'sessionToken': '1235678'}
|
||||
)
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 2
|
||||
assert aioclient_mock.mock_calls[1][2] == 'token=654321&fun=123'
|
||||
assert mac_list == []
|
||||
scanner.token = None
|
||||
mac_list = await scanner.async_scan_devices()
|
||||
|
||||
def test_scan_devices_parse_error(self, aioclient_mock):
|
||||
"""Set up a upc platform and scan device with parse error."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(self.host),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
content=b'successful',
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
assert len(aioclient_mock.mock_calls) == 2
|
||||
assert aioclient_mock.mock_calls[1][2] == 'token=654321&fun=123'
|
||||
assert mac_list == []
|
||||
|
||||
scanner = run_coroutine_threadsafe(platform.async_get_scanner(
|
||||
self.hass, {DOMAIN: {
|
||||
CONF_PLATFORM: 'upc_connect',
|
||||
CONF_HOST: self.host
|
||||
}}
|
||||
), self.hass.loop).result()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
async def test_scan_devices_parse_error(hass, aioclient_mock):
|
||||
"""Set up a upc platform and scan device with parse error."""
|
||||
aioclient_mock.get(
|
||||
"http://{}/common_page/login.html".format(HOST),
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
content=b'successful',
|
||||
cookies={'sessionToken': '654321'}
|
||||
)
|
||||
|
||||
aioclient_mock.clear_requests()
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(self.host),
|
||||
text="Blablebla blabalble",
|
||||
cookies={'sessionToken': '1235678'}
|
||||
)
|
||||
scanner = await platform.async_get_scanner(
|
||||
hass, {
|
||||
DOMAIN: {CONF_PLATFORM: 'upc_connect', CONF_HOST: HOST}})
|
||||
|
||||
mac_list = run_coroutine_threadsafe(
|
||||
scanner.async_scan_devices(), self.hass.loop).result()
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert aioclient_mock.mock_calls[0][2] == 'token=654321&fun=123'
|
||||
assert scanner.token is None
|
||||
assert mac_list == []
|
||||
aioclient_mock.clear_requests()
|
||||
aioclient_mock.post(
|
||||
"http://{}/xml/getter.xml".format(HOST),
|
||||
text="Blablebla blabalble",
|
||||
cookies={'sessionToken': '1235678'}
|
||||
)
|
||||
|
||||
mac_list = await scanner.async_scan_devices()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert aioclient_mock.mock_calls[0][2] == 'token=654321&fun=123'
|
||||
assert scanner.token is None
|
||||
assert mac_list == []
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user