128 lines
4.5 KiB
Python
128 lines
4.5 KiB
Python
"""The tests for the StatsD feeder."""
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
import homeassistant.core as ha
|
|
import homeassistant.components.statsd as statsd
|
|
from homeassistant.const import STATE_ON, STATE_OFF, EVENT_STATE_CHANGED
|
|
|
|
|
|
class TestStatsd(unittest.TestCase):
|
|
"""Test the StatsD component."""
|
|
|
|
@mock.patch('statsd.StatsClient')
|
|
def test_statsd_setup_full(self, mock_connection):
|
|
"""Test setup with all data."""
|
|
config = {
|
|
'statsd': {
|
|
'host': 'host',
|
|
'port': 123,
|
|
'sample_rate': 1,
|
|
'prefix': 'foo',
|
|
}
|
|
}
|
|
hass = mock.MagicMock()
|
|
self.assertTrue(statsd.setup(hass, config))
|
|
mock_connection.assert_called_once_with(
|
|
host='host',
|
|
port=123,
|
|
prefix='foo')
|
|
|
|
self.assertTrue(hass.bus.listen.called)
|
|
self.assertEqual(EVENT_STATE_CHANGED,
|
|
hass.bus.listen.call_args_list[0][0][0])
|
|
|
|
@mock.patch('statsd.StatsClient')
|
|
def test_statsd_setup_defaults(self, mock_connection):
|
|
"""Test setup with defaults."""
|
|
config = {
|
|
'statsd': {
|
|
'host': 'host',
|
|
}
|
|
}
|
|
hass = mock.MagicMock()
|
|
self.assertTrue(statsd.setup(hass, config))
|
|
mock_connection.assert_called_once_with(
|
|
host='host',
|
|
port=statsd.DEFAULT_PORT,
|
|
prefix=statsd.DEFAULT_PREFIX)
|
|
self.assertTrue(hass.bus.listen.called)
|
|
|
|
@mock.patch('statsd.StatsClient')
|
|
def test_event_listener_defaults(self, mock_client):
|
|
"""Test event listener."""
|
|
config = {
|
|
'statsd': {
|
|
'host': 'host',
|
|
}
|
|
}
|
|
hass = mock.MagicMock()
|
|
statsd.setup(hass, config)
|
|
self.assertTrue(hass.bus.listen.called)
|
|
handler_method = hass.bus.listen.call_args_list[0][0][1]
|
|
|
|
valid = {'1': 1,
|
|
'1.0': 1.0,
|
|
STATE_ON: 1,
|
|
STATE_OFF: 0}
|
|
for in_, out in valid.items():
|
|
state = mock.MagicMock(state=in_,
|
|
attributes={"attribute key": 3.2})
|
|
handler_method(mock.MagicMock(data={'new_state': state}))
|
|
mock_client.return_value.gauge.assert_has_calls([
|
|
mock.call(state.entity_id, out, statsd.DEFAULT_RATE),
|
|
])
|
|
|
|
mock_client.return_value.gauge.reset_mock()
|
|
|
|
mock_client.return_value.incr.assert_called_once_with(
|
|
state.entity_id, rate=statsd.DEFAULT_RATE)
|
|
mock_client.return_value.incr.reset_mock()
|
|
|
|
for invalid in ('foo', '', object):
|
|
handler_method(mock.MagicMock(data={
|
|
'new_state': ha.State('domain.test', invalid, {})}))
|
|
self.assertFalse(mock_client.return_value.gauge.called)
|
|
self.assertFalse(mock_client.return_value.incr.called)
|
|
|
|
@mock.patch('statsd.StatsClient')
|
|
def test_event_listener_attr_details(self, mock_client):
|
|
"""Test event listener."""
|
|
config = {
|
|
'statsd': {
|
|
'host': 'host',
|
|
'log_attributes': True
|
|
}
|
|
}
|
|
hass = mock.MagicMock()
|
|
statsd.setup(hass, config)
|
|
self.assertTrue(hass.bus.listen.called)
|
|
handler_method = hass.bus.listen.call_args_list[0][0][1]
|
|
|
|
valid = {'1': 1,
|
|
'1.0': 1.0,
|
|
STATE_ON: 1,
|
|
STATE_OFF: 0}
|
|
for in_, out in valid.items():
|
|
state = mock.MagicMock(state=in_,
|
|
attributes={"attribute key": 3.2})
|
|
handler_method(mock.MagicMock(data={'new_state': state}))
|
|
mock_client.return_value.gauge.assert_has_calls([
|
|
mock.call("%s.state" % state.entity_id,
|
|
out, statsd.DEFAULT_RATE),
|
|
mock.call("%s.attribute_key" % state.entity_id,
|
|
3.2, statsd.DEFAULT_RATE),
|
|
])
|
|
|
|
mock_client.return_value.gauge.reset_mock()
|
|
|
|
mock_client.return_value.incr.assert_called_once_with(
|
|
state.entity_id, rate=statsd.DEFAULT_RATE)
|
|
mock_client.return_value.incr.reset_mock()
|
|
|
|
for invalid in ('foo', '', object):
|
|
handler_method(mock.MagicMock(data={
|
|
'new_state': ha.State('domain.test', invalid, {})}))
|
|
self.assertFalse(mock_client.return_value.gauge.called)
|
|
self.assertFalse(mock_client.return_value.incr.called)
|