"""Tests for the HomeKit component.""" import unittest from unittest.mock import call, patch import voluptuous as vol from homeassistant import setup from homeassistant.core import Event from homeassistant.components.homekit import ( CONF_PIN_CODE, BRIDGE_NAME, HOMEKIT_FILE, HomeKit, valid_pin) from homeassistant.const import ( CONF_PORT, EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STOP) from tests.common import get_test_home_assistant HOMEKIT_PATH = 'homeassistant.components.homekit' CONFIG_MIN = {'homekit': {}} CONFIG = { 'homekit': { CONF_PORT: 11111, CONF_PIN_CODE: '987-65-432', } } class TestHomeKit(unittest.TestCase): """Test setup of HomeKit component and HomeKit class.""" def setUp(self): """Setup things to be run when tests are started.""" self.hass = get_test_home_assistant() def tearDown(self): """Stop down everything that was started.""" self.hass.stop() @patch(HOMEKIT_PATH + '.HomeKit.start_driver') @patch(HOMEKIT_PATH + '.HomeKit.setup_bridge') @patch(HOMEKIT_PATH + '.HomeKit.__init__') def test_setup_min(self, mock_homekit, mock_setup_bridge, mock_start_driver): """Test async_setup with minimal config option.""" mock_homekit.return_value = None self.assertTrue(setup.setup_component( self.hass, 'homekit', CONFIG_MIN)) mock_homekit.assert_called_once_with(self.hass, 51826) mock_setup_bridge.assert_called_with(b'123-45-678') mock_start_driver.assert_not_called() self.hass.start() self.hass.block_till_done() self.assertEqual(mock_start_driver.call_count, 1) @patch(HOMEKIT_PATH + '.HomeKit.start_driver') @patch(HOMEKIT_PATH + '.HomeKit.setup_bridge') @patch(HOMEKIT_PATH + '.HomeKit.__init__') def test_setup_parameters(self, mock_homekit, mock_setup_bridge, mock_start_driver): """Test async_setup with full config option.""" mock_homekit.return_value = None self.assertTrue(setup.setup_component( self.hass, 'homekit', CONFIG)) mock_homekit.assert_called_once_with(self.hass, 11111) mock_setup_bridge.assert_called_with(b'987-65-432') def test_validate_pincode(self): """Test async_setup with invalid config option.""" schema = vol.Schema(valid_pin) for value in ('', '123-456-78', 'a23-45-678', '12345678', 1234): with self.assertRaises(vol.MultipleInvalid): schema(value) for value in ('123-45-678', '234-56-789'): self.assertTrue(schema(value)) @patch('pyhap.accessory_driver.AccessoryDriver') @patch('pyhap.accessory.Bridge.add_accessory') @patch(HOMEKIT_PATH + '.import_types') @patch(HOMEKIT_PATH + '.get_accessory') def test_homekit_pyhap_interaction( self, mock_get_accessory, mock_import_types, mock_add_accessory, mock_acc_driver): """Test interaction between the HomeKit class and pyhap.""" mock_get_accessory.side_effect = ['TemperatureSensor', 'Window'] homekit = HomeKit(self.hass, 51826) homekit.setup_bridge(b'123-45-678') self.assertEqual(homekit.bridge.display_name, BRIDGE_NAME) self.hass.states.set('demo.demo1', 'on') self.hass.states.set('demo.demo2', 'off') self.hass.start() self.hass.block_till_done() with patch('homeassistant.util.get_local_ip', return_value='127.0.0.1'): homekit.start_driver(Event(EVENT_HOMEASSISTANT_START)) ip_address = '127.0.0.1' path = self.hass.config.path(HOMEKIT_FILE) self.assertEqual(mock_get_accessory.call_count, 2) self.assertEqual(mock_import_types.call_count, 1) self.assertEqual(mock_acc_driver.mock_calls, [call(homekit.bridge, 51826, ip_address, path), call().start()]) self.assertEqual(mock_add_accessory.mock_calls, [call('TemperatureSensor'), call('Window')]) self.hass.bus.fire(EVENT_HOMEASSISTANT_STOP) self.hass.block_till_done() self.assertEqual(mock_acc_driver.mock_calls[2], call().stop()) self.assertEqual(len(mock_acc_driver.mock_calls), 3)