609d7ebea5
* Add event loop to the core * Add block_till_done to HA core object * Fix some tests * Linting core * Fix statemachine tests * Core test fixes * fix block_till_done to wait for loop and queue to empty * fix test_core for passing, and correct start/stop/block_till_done * Fix remote tests * Fix tests: block_till_done * Fix linting * Fix more tests * Fix final linting * Fix remote test * remove unnecessary import * reduce sleep to avoid slowing down the tests excessively * fix remaining tests to wait for non-threadsafe operations * Add async_ doc strings for event loop / coroutine info * Fix command line test to block for the right timeout * Fix py3.4.2 loop var access * Fix SERVICE_CALL_LIMIT being in effect for other tests * Fix lint errors * Fix lint error with proper placement * Fix slave start to not start a timer * Add asyncio compatible listeners. * Increase min Python version to 3.4.2 * Move async backports to util * Add backported async tests * Fix linting * Simplify Python version check * Fix lint * Remove unneeded try/except and queue listener appproriately. * Fix tuple vs. list unorderable error on version compare. * Fix version tests
69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
"""The tests for the MQTT device tracker platform."""
|
|
import unittest
|
|
from unittest.mock import patch
|
|
import logging
|
|
import os
|
|
|
|
from homeassistant.bootstrap import _setup_component
|
|
from homeassistant.components import device_tracker
|
|
from homeassistant.const import CONF_PLATFORM
|
|
|
|
from tests.common import (
|
|
get_test_home_assistant, mock_mqtt_component, fire_mqtt_message)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class TestComponentsDeviceTrackerMQTT(unittest.TestCase):
|
|
"""Test MQTT device tracker platform."""
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
"""Setup things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
mock_mqtt_component(self.hass)
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
"""Stop everything that was started."""
|
|
try:
|
|
os.remove(self.hass.config.path(device_tracker.YAML_DEVICES))
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
def test_ensure_device_tracker_platform_validation(self): \
|
|
# pylint: disable=invalid-name
|
|
"""Test if platform validation was done."""
|
|
def mock_setup_scanner(hass, config, see):
|
|
"""Check that Qos was added by validation."""
|
|
self.assertTrue('qos' in config)
|
|
|
|
with patch('homeassistant.components.device_tracker.mqtt.'
|
|
'setup_scanner', side_effect=mock_setup_scanner) as mock_sp:
|
|
|
|
dev_id = 'paulus'
|
|
topic = '/location/paulus'
|
|
self.hass.config.components = ['mqtt', 'zone']
|
|
assert _setup_component(self.hass, device_tracker.DOMAIN, {
|
|
device_tracker.DOMAIN: {
|
|
CONF_PLATFORM: 'mqtt',
|
|
'devices': {dev_id: topic}
|
|
}
|
|
})
|
|
assert mock_sp.call_count == 1
|
|
|
|
def test_new_message(self):
|
|
"""Test new message."""
|
|
dev_id = 'paulus'
|
|
enttiy_id = device_tracker.ENTITY_ID_FORMAT.format(dev_id)
|
|
topic = '/location/paulus'
|
|
location = 'work'
|
|
|
|
self.hass.config.components = ['mqtt', 'zone']
|
|
assert _setup_component(self.hass, device_tracker.DOMAIN, {
|
|
device_tracker.DOMAIN: {
|
|
CONF_PLATFORM: 'mqtt',
|
|
'devices': {dev_id: topic}
|
|
}
|
|
})
|
|
fire_mqtt_message(self.hass, topic, location)
|
|
self.hass.block_till_done()
|
|
self.assertEqual(location, self.hass.states.get(enttiy_id).state)
|