core/tests/components/google/test_init.py

"""The tests for the Google Calendar component."""
from unittest.mock import patch

import pytest

import homeassistant.components.google as google
from homeassistant.setup import async_setup_component


@pytest.fixture(name='google_setup')
def mock_google_setup(hass):
    """Mock the google set up functions."""
    p_auth = patch(
        'homeassistant.components.google.do_authentication',
        side_effect=google.do_setup)
    p_service = patch(
        'homeassistant.components.google.GoogleCalendarService.get')
    p_discovery = patch(
        'homeassistant.components.google.discovery.load_platform')
    p_load = patch(
        'homeassistant.components.google.load_config',
        return_value={})
    p_save = patch(
        'homeassistant.components.google.update_config')

    with p_auth, p_load, p_service, p_discovery, p_save:
        yield


async def test_setup_component(hass, google_setup):
    """Test setup component."""
    config = {
        'google': {
            'client_id': 'id',
            'client_secret': 'secret',
        }
    }

    assert await async_setup_component(hass, 'google', config)


async def test_get_calendar_info(hass, test_calendar):
    """Test getting the calendar info."""
    calendar_info = await hass.async_add_executor_job(
        google.get_calendar_info, hass, test_calendar)
    assert calendar_info == {
        'cal_id': 'qwertyuiopasdfghjklzxcvbnm@import.calendar.google.com',
        'entities': [{
            'device_id': 'we_are_we_are_a_test_calendar',
            'name': 'We are, we are, a... Test Calendar',
            'track': True,
            'ignore_availability': True,
        }]
    }


async def test_found_calendar(
        hass, google_setup, mock_next_event, test_calendar):
    """Test when a calendar is found."""
    config = {
        'google': {
            'client_id': 'id',
            'client_secret': 'secret',
            'track_new_calendar': True,
        }
    }
    assert await async_setup_component(hass, 'google', config)
    assert hass.data[google.DATA_INDEX] == {}

    await hass.services.async_call(
        'google', google.SERVICE_FOUND_CALENDARS, test_calendar, blocking=True)

    assert hass.data[google.DATA_INDEX].get(test_calendar['id']) is not None