Updated to new statsd library and added state change counters (#2429)

tests/components/test_statsd.py

@@ -10,9 +10,8 @@ from homeassistant.const import STATE_ON, STATE_OFF, EVENT_STATE_CHANGED
 class TestStatsd(unittest.TestCase):
     """Test the StatsD component."""
 
-    @mock.patch('statsd.Connection')
-    @mock.patch('statsd.Gauge')
-    def test_statsd_setup_full(self, mock_gauge, mock_connection):
+    @mock.patch('statsd.StatsClient')
+    def test_statsd_setup_full(self, mock_connection):
         """Test setup with all data."""
         config = {
             'statsd': {
@@ -24,18 +23,17 @@ class TestStatsd(unittest.TestCase):
         }
         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)
+        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.Connection')
-    @mock.patch('statsd.Gauge')
-    def test_statsd_setup_defaults(self, mock_gauge, mock_connection):
+    @mock.patch('statsd.StatsClient')
+    def test_statsd_setup_defaults(self, mock_connection):
         """Test setup with defaults."""
         config = {
             'statsd': {
@@ -47,15 +45,11 @@ class TestStatsd(unittest.TestCase):
         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)
+            prefix=statsd.DEFAULT_PREFIX)
         self.assertTrue(hass.bus.listen.called)
 
-    @mock.patch('statsd.Connection')
-    @mock.patch('statsd.Gauge')
-    def test_event_listener(self, mock_gauge, mock_connection):
+    @mock.patch('statsd.StatsClient')
+    def test_event_listener(self, mock_client):
         """Test event listener."""
         config = {
             'statsd': {
@@ -74,11 +68,16 @@ class TestStatsd(unittest.TestCase):
         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()
+            mock_client.return_value.gauge.assert_called_once_with(
+                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_gauge.return_value.send.called)
+            self.assertFalse(mock_client.return_value.gauge.called)
+            self.assertFalse(mock_client.return_value.incr.called)