core/tests/components/hangouts/test_config_flow.py

"""Tests for the Google Hangouts config flow."""

from unittest.mock import patch

from homeassistant import data_entry_flow
from homeassistant.components.hangouts import config_flow


async def test_flow_works(hass, aioclient_mock):
    """Test config flow without 2fa."""
    flow = config_flow.HangoutsFlowHandler()

    flow.hass = hass

    with patch('hangups.get_auth'):
        result = await flow.async_step_user(
            {'email': 'test@test.com', 'password': '1232456'})
        assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
        assert result['title'] == 'test@test.com'


async def test_flow_works_with_authcode(hass, aioclient_mock):
    """Test config flow without 2fa."""
    flow = config_flow.HangoutsFlowHandler()

    flow.hass = hass

    with patch('hangups.get_auth'):
        result = await flow.async_step_user(
            {'email': 'test@test.com', 'password': '1232456',
             'authorization_code': 'c29tZXJhbmRvbXN0cmluZw=='})
        assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
        assert result['title'] == 'test@test.com'


async def test_flow_works_with_2fa(hass, aioclient_mock):
    """Test config flow with 2fa."""
    from homeassistant.components.hangouts.hangups_utils import Google2FAError

    flow = config_flow.HangoutsFlowHandler()

    flow.hass = hass

    with patch('hangups.get_auth', side_effect=Google2FAError):
        result = await flow.async_step_user(
            {'email': 'test@test.com', 'password': '1232456'})
        assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
        assert result['step_id'] == '2fa'

    with patch('hangups.get_auth'):
        result = await flow.async_step_2fa({'2fa': 123456})
        assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
        assert result['title'] == 'test@test.com'


async def test_flow_with_unknown_2fa(hass, aioclient_mock):
    """Test config flow with invalid 2fa method."""
    from homeassistant.components.hangouts.hangups_utils import GoogleAuthError

    flow = config_flow.HangoutsFlowHandler()

    flow.hass = hass

    with patch('hangups.get_auth',
               side_effect=GoogleAuthError('Unknown verification code input')):
        result = await flow.async_step_user(
            {'email': 'test@test.com', 'password': '1232456'})
        assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
        assert result['errors']['base'] == 'invalid_2fa_method'


async def test_flow_invalid_login(hass, aioclient_mock):
    """Test config flow with invalid 2fa method."""
    from homeassistant.components.hangouts.hangups_utils import GoogleAuthError

    flow = config_flow.HangoutsFlowHandler()

    flow.hass = hass

    with patch('hangups.get_auth',
               side_effect=GoogleAuthError):
        result = await flow.async_step_user(
            {'email': 'test@test.com', 'password': '1232456'})
        assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
        assert result['errors']['base'] == 'invalid_login'


async def test_flow_invalid_2fa(hass, aioclient_mock):
    """Test config flow with 2fa."""
    from homeassistant.components.hangouts.hangups_utils import Google2FAError

    flow = config_flow.HangoutsFlowHandler()

    flow.hass = hass

    with patch('hangups.get_auth', side_effect=Google2FAError):
        result = await flow.async_step_user(
            {'email': 'test@test.com', 'password': '1232456'})
        assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
        assert result['step_id'] == '2fa'

    with patch('hangups.get_auth', side_effect=Google2FAError):
        result = await flow.async_step_2fa({'2fa': 123456})

        assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
        assert result['errors']['base'] == 'invalid_2fa'