Пример #1
0
    def test_event_listener(self, mock_gauge, mock_connection):
        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_)
            handler_method(mock.MagicMock(data={'new_state': state}))
            mock_gauge.return_value.send.assert_called_once_with(
                state.entity_id, out)
            mock_gauge.return_value.send.reset_mock()

        for invalid in ('foo', '', object):
            handler_method(mock.MagicMock(data={
                'new_state': ha.State('domain.test', invalid, {})}))
            self.assertFalse(mock_gauge.return_value.send.called)
Пример #2
0
    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)
Пример #3
0
    def test_event_listener(self, mock_gauge, mock_connection):
        """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_)
            handler_method(mock.MagicMock(data={'new_state': state}))
            mock_gauge.return_value.send.assert_called_once_with(
                state.entity_id, out)
            mock_gauge.return_value.send.reset_mock()

        for invalid in ('foo', '', object):
            handler_method(
                mock.MagicMock(
                    data={'new_state': ha.State('domain.test', invalid, {})}))
            self.assertFalse(mock_gauge.return_value.send.called)
Пример #4
0
    def test_event_listener(self, mock_gauge, mock_connection):
        """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_)
            handler_method(mock.MagicMock(data={"new_state": state}))
            mock_gauge.return_value.send.assert_called_once_with(state.entity_id, out)
            mock_gauge.return_value.send.reset_mock()

        for invalid in ("foo", "", object):
            handler_method(mock.MagicMock(data={"new_state": ha.State("domain.test", invalid, {})}))
            self.assertFalse(mock_gauge.return_value.send.called)
Пример #5
0
 def test_statsd_setup_full(self, mock_gauge, 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, sample_rate=1, disabled=False)
     mock_gauge.assert_called_once_with("foo", mock_connection.return_value)
     self.assertTrue(hass.bus.listen.called)
     self.assertEqual(EVENT_STATE_CHANGED, hass.bus.listen.call_args_list[0][0][0])
Пример #6
0
 def test_statsd_setup_defaults(self, mock_gauge, 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, sample_rate=statsd.DEFAULT_RATE, disabled=False
     )
     mock_gauge.assert_called_once_with(statsd.DEFAULT_PREFIX, mock_connection.return_value)
     self.assertTrue(hass.bus.listen.called)
Пример #7
0
 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)
Пример #8
0
 def test_statsd_setup_defaults(self, mock_gauge, mock_connection):
     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,
         sample_rate=statsd.DEFAULT_RATE,
         disabled=False)
     mock_gauge.assert_called_once_with(statsd.DEFAULT_PREFIX,
                                        mock_connection.return_value)
     self.assertTrue(hass.bus.listen.called)
Пример #9
0
 def test_statsd_setup_defaults(self, mock_gauge, mock_connection):
     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,
         sample_rate=statsd.DEFAULT_RATE,
         disabled=False)
     mock_gauge.assert_called_once_with(statsd.DEFAULT_PREFIX,
                                        mock_connection.return_value)
     self.assertTrue(hass.bus.listen.called)
Пример #10
0
 def test_statsd_setup_full(self, mock_gauge, mock_connection):
     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,
                                             sample_rate=1,
                                             disabled=False)
     mock_gauge.assert_called_once_with('foo',
                                        mock_connection.return_value)
     self.assertTrue(hass.bus.listen.called)
     self.assertEqual(EVENT_STATE_CHANGED,
                      hass.bus.listen.call_args_list[0][0][0])
Пример #11
0
 def test_statsd_setup_full(self, mock_gauge, mock_connection):
     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,
                                             sample_rate=1,
                                             disabled=False)
     mock_gauge.assert_called_once_with('foo', mock_connection.return_value)
     self.assertTrue(hass.bus.listen.called)
     self.assertEqual(EVENT_STATE_CHANGED,
                      hass.bus.listen.call_args_list[0][0][0])
Пример #12
0
    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])