112 lines
4.3 KiB
Python
112 lines
4.3 KiB
Python
"""
|
|
tests.components.test_conversation
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Tests Conversation component.
|
|
"""
|
|
# pylint: disable=too-many-public-methods,protected-access
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
import homeassistant.components as core_components
|
|
from homeassistant.components import conversation
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
class TestConversation(unittest.TestCase):
|
|
""" Test the conversation component. """
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
""" Start up ha for testing """
|
|
self.ent_id = 'light.kitchen_lights'
|
|
self.hass = get_test_home_assistant(3)
|
|
self.hass.states.set(self.ent_id, 'on')
|
|
self.assertTrue(core_components.setup(self.hass, {}))
|
|
self.assertTrue(
|
|
conversation.setup(self.hass, {conversation.DOMAIN: {}}))
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
""" Stop down stuff we started. """
|
|
self.hass.stop()
|
|
|
|
def test_turn_on(self):
|
|
""" Setup and perform good turn on requests """
|
|
calls = []
|
|
|
|
def record_call(service):
|
|
calls.append(service)
|
|
|
|
self.hass.services.register('light', 'turn_on', record_call)
|
|
|
|
event_data = {conversation.ATTR_TEXT: 'turn kitchen lights on'}
|
|
self.assertTrue(self.hass.services.call(
|
|
conversation.DOMAIN, 'process', event_data, True))
|
|
|
|
call = calls[-1]
|
|
self.assertEqual('light', call.domain)
|
|
self.assertEqual('turn_on', call.service)
|
|
self.assertEqual([self.ent_id], call.data[ATTR_ENTITY_ID])
|
|
|
|
def test_turn_off(self):
|
|
""" Setup and perform good turn off requests """
|
|
calls = []
|
|
|
|
def record_call(service):
|
|
calls.append(service)
|
|
|
|
self.hass.services.register('light', 'turn_off', record_call)
|
|
|
|
event_data = {conversation.ATTR_TEXT: 'turn kitchen lights off'}
|
|
self.assertTrue(self.hass.services.call(
|
|
conversation.DOMAIN, 'process', event_data, True))
|
|
|
|
call = calls[-1]
|
|
self.assertEqual('light', call.domain)
|
|
self.assertEqual('turn_off', call.service)
|
|
self.assertEqual([self.ent_id], call.data[ATTR_ENTITY_ID])
|
|
|
|
@patch('homeassistant.components.conversation.logging.Logger.error')
|
|
@patch('homeassistant.core.ServiceRegistry.call')
|
|
def test_bad_request_format(self, mock_logger, mock_call):
|
|
""" Setup and perform a badly formatted request """
|
|
event_data = {
|
|
conversation.ATTR_TEXT:
|
|
'what is the answer to the ultimate question of life, ' +
|
|
'the universe and everything'}
|
|
self.assertTrue(self.hass.services.call(
|
|
conversation.DOMAIN, 'process', event_data, True))
|
|
self.assertTrue(mock_logger.called)
|
|
self.assertFalse(mock_call.called)
|
|
|
|
@patch('homeassistant.components.conversation.logging.Logger.error')
|
|
@patch('homeassistant.core.ServiceRegistry.call')
|
|
def test_bad_request_entity(self, mock_logger, mock_call):
|
|
""" Setup and perform requests with bad entity id """
|
|
event_data = {conversation.ATTR_TEXT: 'turn something off'}
|
|
self.assertTrue(self.hass.services.call(
|
|
conversation.DOMAIN, 'process', event_data, True))
|
|
self.assertTrue(mock_logger.called)
|
|
self.assertFalse(mock_call.called)
|
|
|
|
@patch('homeassistant.components.conversation.logging.Logger.error')
|
|
@patch('homeassistant.core.ServiceRegistry.call')
|
|
def test_bad_request_command(self, mock_logger, mock_call):
|
|
""" Setup and perform requests with bad command """
|
|
event_data = {conversation.ATTR_TEXT: 'turn kitchen lights over'}
|
|
self.assertTrue(self.hass.services.call(
|
|
conversation.DOMAIN, 'process', event_data, True))
|
|
self.assertTrue(mock_logger.called)
|
|
self.assertFalse(mock_call.called)
|
|
|
|
@patch('homeassistant.components.conversation.logging.Logger.error')
|
|
@patch('homeassistant.core.ServiceRegistry.call')
|
|
def test_bad_request_notext(self, mock_logger, mock_call):
|
|
""" Setup and perform requests with bad command with no text """
|
|
event_data = {}
|
|
self.assertTrue(self.hass.services.call(
|
|
conversation.DOMAIN, 'process', event_data, True))
|
|
self.assertTrue(mock_logger.called)
|
|
self.assertFalse(mock_call.called)
|