d3bab30dbe
* Add cloud status * Expose certificate details * store & reset last state * Fix tests * update tests * update req * fix lint
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
"""Test the cloud component."""
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from homeassistant.const import (
|
|
EVENT_HOMEASSISTANT_STOP, EVENT_HOMEASSISTANT_START)
|
|
from homeassistant.components import cloud
|
|
from homeassistant.components.cloud.const import DOMAIN
|
|
|
|
from tests.common import mock_coro
|
|
|
|
|
|
async def test_constructor_loads_info_from_config():
|
|
"""Test non-dev mode loads info from SERVERS constant."""
|
|
hass = MagicMock(data={})
|
|
|
|
with patch(
|
|
"homeassistant.components.cloud.prefs.CloudPreferences."
|
|
"async_initialize",
|
|
return_value=mock_coro()
|
|
):
|
|
result = await cloud.async_setup(hass, {
|
|
'cloud': {
|
|
cloud.CONF_MODE: cloud.MODE_DEV,
|
|
'cognito_client_id': 'test-cognito_client_id',
|
|
'user_pool_id': 'test-user_pool_id',
|
|
'region': 'test-region',
|
|
'relayer': 'test-relayer',
|
|
}
|
|
})
|
|
assert result
|
|
|
|
cl = hass.data['cloud']
|
|
assert cl.mode == cloud.MODE_DEV
|
|
assert cl.cognito_client_id == 'test-cognito_client_id'
|
|
assert cl.user_pool_id == 'test-user_pool_id'
|
|
assert cl.region == 'test-region'
|
|
assert cl.relayer == 'test-relayer'
|
|
|
|
|
|
async def test_remote_services(hass, mock_cloud_fixture):
|
|
"""Setup cloud component and test services."""
|
|
cloud = hass.data[DOMAIN]
|
|
|
|
assert hass.services.has_service(DOMAIN, 'remote_connect')
|
|
assert hass.services.has_service(DOMAIN, 'remote_disconnect')
|
|
|
|
with patch(
|
|
"hass_nabucasa.remote.RemoteUI.connect", return_value=mock_coro()
|
|
) as mock_connect:
|
|
await hass.services.async_call(DOMAIN, "remote_connect", blocking=True)
|
|
|
|
assert mock_connect.called
|
|
assert cloud.client.remote_autostart
|
|
|
|
with patch(
|
|
"hass_nabucasa.remote.RemoteUI.disconnect", return_value=mock_coro()
|
|
) as mock_disconnect:
|
|
await hass.services.async_call(
|
|
DOMAIN, "remote_disconnect", blocking=True)
|
|
|
|
assert mock_disconnect.called
|
|
assert not cloud.client.remote_autostart
|
|
|
|
|
|
async def test_startup_shutdown_events(hass, mock_cloud_fixture):
|
|
"""Test if the cloud will start on startup event."""
|
|
with patch(
|
|
"hass_nabucasa.Cloud.start", return_value=mock_coro()
|
|
) as mock_start:
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_start.called
|
|
|
|
with patch(
|
|
"hass_nabucasa.Cloud.stop", return_value=mock_coro()
|
|
) as mock_stop:
|
|
hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_stop.called
|