84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
"""Tests for the cloud component."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from hass_nabucasa import Cloud
|
|
|
|
from homeassistant.components import cloud
|
|
from homeassistant.components.cloud import const, prefs as cloud_prefs
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
PIPELINE_DATA = {
|
|
"items": [
|
|
{
|
|
"conversation_engine": "conversation_engine_1",
|
|
"conversation_language": "language_1",
|
|
"id": "01GX8ZWBAQYWNB1XV3EXEZ75DY",
|
|
"language": "language_1",
|
|
"name": "Home Assistant Cloud",
|
|
"stt_engine": "cloud",
|
|
"stt_language": "language_1",
|
|
"tts_engine": "cloud",
|
|
"tts_language": "language_1",
|
|
"tts_voice": "Arnold Schwarzenegger",
|
|
"wake_word_entity": None,
|
|
"wake_word_id": None,
|
|
},
|
|
{
|
|
"conversation_engine": "conversation_engine_2",
|
|
"conversation_language": "language_2",
|
|
"id": "01GX8ZWBAQTKFQNK4W7Q4CTRCX",
|
|
"language": "language_2",
|
|
"name": "name_2",
|
|
"stt_engine": "stt_engine_2",
|
|
"stt_language": "language_2",
|
|
"tts_engine": "tts_engine_2",
|
|
"tts_language": "language_2",
|
|
"tts_voice": "The Voice",
|
|
"wake_word_entity": None,
|
|
"wake_word_id": None,
|
|
},
|
|
{
|
|
"conversation_engine": "conversation_engine_3",
|
|
"conversation_language": "language_3",
|
|
"id": "01GX8ZWBAQSV1HP3WGJPFWEJ8J",
|
|
"language": "language_3",
|
|
"name": "name_3",
|
|
"stt_engine": None,
|
|
"stt_language": None,
|
|
"tts_engine": None,
|
|
"tts_language": None,
|
|
"tts_voice": None,
|
|
"wake_word_entity": None,
|
|
"wake_word_id": None,
|
|
},
|
|
],
|
|
"preferred_item": "01GX8ZWBAQYWNB1XV3EXEZ75DY",
|
|
}
|
|
|
|
|
|
async def mock_cloud(hass, config=None):
|
|
"""Mock cloud."""
|
|
# The homeassistant integration is needed by cloud. It's not in it's requirements
|
|
# because it's always setup by bootstrap. Set it up manually in tests.
|
|
assert await async_setup_component(hass, "homeassistant", {})
|
|
|
|
assert await async_setup_component(hass, cloud.DOMAIN, {"cloud": config or {}})
|
|
cloud_inst: Cloud = hass.data["cloud"]
|
|
with patch("hass_nabucasa.Cloud.run_executor", AsyncMock(return_value=None)):
|
|
await cloud_inst.initialize()
|
|
|
|
|
|
def mock_cloud_prefs(hass, prefs={}):
|
|
"""Fixture for cloud component."""
|
|
prefs_to_set = {
|
|
const.PREF_ALEXA_SETTINGS_VERSION: cloud_prefs.ALEXA_SETTINGS_VERSION,
|
|
const.PREF_ENABLE_ALEXA: True,
|
|
const.PREF_ENABLE_GOOGLE: True,
|
|
const.PREF_GOOGLE_SECURE_DEVICES_PIN: None,
|
|
const.PREF_GOOGLE_SETTINGS_VERSION: cloud_prefs.GOOGLE_SETTINGS_VERSION,
|
|
}
|
|
prefs_to_set.update(prefs)
|
|
hass.data[cloud.DOMAIN].client._prefs._prefs = prefs_to_set
|
|
return hass.data[cloud.DOMAIN].client._prefs
|