hass-core/tests/components/cloud/test_init.py

"""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."""
    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

    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


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