1
0
mirror of https://github.com/elisspace/core.git synced 2026-09-13 14:37:05 +00:00

Update pyunifi component to use APIError passed from pyunifi 2.13. Better accommodate login failures with wrapper in pyunifi 2.13. (#7899)

* Pyunifi update

* Update pyunifi_test

* Import API Error

* Adjust test_unifi.py to import APIError

* Remove urllib import

* Remove urllib import from test

* Try fix mock

* Remove automations.yaml

* Lint
This commit is contained in:
Caleb
2017-06-17 13:09:27 -05:00
committed by Paulus Schoutsen
parent 363a429c41
commit a2fbc0d2ef
5 changed files with 15 additions and 14 deletions

View File

@@ -1,6 +1,6 @@
"""The tests for the Unifi WAP device tracker platform."""
from unittest import mock
import urllib
from pyunifi.controller import APIError
import pytest
import voluptuous as vol
@@ -13,11 +13,8 @@ from homeassistant.const import (CONF_HOST, CONF_USERNAME, CONF_PASSWORD,
@pytest.fixture
def mock_ctrl():
"""Mock pyunifi."""
module = mock.MagicMock()
with mock.patch.dict('sys.modules', {
'pyunifi.controller': module.controller,
}):
yield module.controller.Controller
with mock.patch('pyunifi.controller.Controller') as mock_control:
yield mock_control
@pytest.fixture
@@ -100,7 +97,7 @@ def test_config_controller_failed(hass, mock_ctrl, mock_scanner):
CONF_PASSWORD: 'password',
}
}
mock_ctrl.side_effect = urllib.error.HTTPError(
mock_ctrl.side_effect = APIError(
'/', 500, 'foo', {}, None)
result = unifi.get_scanner(hass, config)
assert result is False
@@ -122,7 +119,7 @@ def test_scanner_update():
def test_scanner_update_error():
"""Test the scanner update for error."""
ctrl = mock.MagicMock()
ctrl.get_clients.side_effect = urllib.error.HTTPError(
ctrl.get_clients.side_effect = APIError(
'/', 500, 'foo', {}, None)
unifi.UnifiScanner(ctrl)