474ac8b09e
* Add basic support for native Hue sensors * Update coveragerc * Simplify attributes * Remove config option * Refactor and document device-ness and update mechanism * Entity docstrings * Remove lingering config for sensors * Whitespace * Remove redundant entity ID generation and hass assignment. * More meaningful variable name. * Add new 'not-darkness' pseudo-sensor. * Refactor sensors into separate binary, non-binary, and shared modules. * formatting * make linter happy. * Refactor again, fix update mechanism, and address comments. * Remove unnecessary assignment * Small fixes. * docstring * Another refactor: only call API once and make testing easier * Tests & test fixes * Flake & lint * Use gather and dispatcher * Remove unnecessary whitespace change. * Move component related stuff out of the shared module * Remove unused remnant of failed approach. * Increase test coverage * Don't get too upset if we're already trying to update an entity before it has finished adding * relative imports
98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
"""Test Hue bridge."""
|
|
from unittest.mock import Mock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
from homeassistant.components.hue import bridge, errors
|
|
|
|
from tests.common import mock_coro
|
|
|
|
|
|
async def test_bridge_setup():
|
|
"""Test a successful setup."""
|
|
hass = Mock()
|
|
entry = Mock()
|
|
api = Mock()
|
|
entry.data = {'host': '1.2.3.4', 'username': 'mock-username'}
|
|
hue_bridge = bridge.HueBridge(hass, entry, False, False)
|
|
|
|
with patch.object(bridge, 'get_bridge', return_value=mock_coro(api)):
|
|
assert await hue_bridge.async_setup() is True
|
|
|
|
assert hue_bridge.api is api
|
|
forward_entries = set(
|
|
c[1][1]
|
|
for c in
|
|
hass.config_entries.async_forward_entry_setup.mock_calls
|
|
)
|
|
assert len(hass.config_entries.async_forward_entry_setup.mock_calls) == 3
|
|
assert forward_entries == set(['light', 'binary_sensor', 'sensor'])
|
|
|
|
|
|
async def test_bridge_setup_invalid_username():
|
|
"""Test we start config flow if username is no longer whitelisted."""
|
|
hass = Mock()
|
|
entry = Mock()
|
|
entry.data = {'host': '1.2.3.4', 'username': 'mock-username'}
|
|
hue_bridge = bridge.HueBridge(hass, entry, False, False)
|
|
|
|
with patch.object(bridge, 'get_bridge',
|
|
side_effect=errors.AuthenticationRequired):
|
|
assert await hue_bridge.async_setup() is False
|
|
|
|
assert len(hass.async_create_task.mock_calls) == 1
|
|
assert len(hass.config_entries.flow.async_init.mock_calls) == 1
|
|
assert hass.config_entries.flow.async_init.mock_calls[0][2]['data'] == {
|
|
'host': '1.2.3.4'
|
|
}
|
|
|
|
|
|
async def test_bridge_setup_timeout(hass):
|
|
"""Test we retry to connect if we cannot connect."""
|
|
hass = Mock()
|
|
entry = Mock()
|
|
entry.data = {'host': '1.2.3.4', 'username': 'mock-username'}
|
|
hue_bridge = bridge.HueBridge(hass, entry, False, False)
|
|
|
|
with patch.object(
|
|
bridge, 'get_bridge', side_effect=errors.CannotConnect
|
|
), pytest.raises(ConfigEntryNotReady):
|
|
await hue_bridge.async_setup()
|
|
|
|
|
|
async def test_reset_if_entry_had_wrong_auth():
|
|
"""Test calling reset when the entry contained wrong auth."""
|
|
hass = Mock()
|
|
entry = Mock()
|
|
entry.data = {'host': '1.2.3.4', 'username': 'mock-username'}
|
|
hue_bridge = bridge.HueBridge(hass, entry, False, False)
|
|
|
|
with patch.object(bridge, 'get_bridge',
|
|
side_effect=errors.AuthenticationRequired):
|
|
assert await hue_bridge.async_setup() is False
|
|
|
|
assert len(hass.async_create_task.mock_calls) == 1
|
|
|
|
assert await hue_bridge.async_reset()
|
|
|
|
|
|
async def test_reset_unloads_entry_if_setup():
|
|
"""Test calling reset while the entry has been setup."""
|
|
hass = Mock()
|
|
entry = Mock()
|
|
entry.data = {'host': '1.2.3.4', 'username': 'mock-username'}
|
|
hue_bridge = bridge.HueBridge(hass, entry, False, False)
|
|
|
|
with patch.object(bridge, 'get_bridge', return_value=mock_coro(Mock())):
|
|
assert await hue_bridge.async_setup() is True
|
|
|
|
assert len(hass.services.async_register.mock_calls) == 1
|
|
assert len(hass.config_entries.async_forward_entry_setup.mock_calls) == 3
|
|
|
|
hass.config_entries.async_forward_entry_unload.return_value = \
|
|
mock_coro(True)
|
|
assert await hue_bridge.async_reset()
|
|
|
|
assert len(hass.config_entries.async_forward_entry_unload.mock_calls) == 3
|
|
assert len(hass.services.async_remove.mock_calls) == 1
|