Add datadog component (#7158)

* Add datadog component

* Improve test_invalid_config datadog test

* Use assert_setup_component for test setup

tests/components/test_datadog.py

@@ -0,0 +1,179 @@
+"""The tests for the Datadog component."""
+from unittest import mock
+import unittest
+
+from homeassistant.const import (
+    EVENT_LOGBOOK_ENTRY,
+    EVENT_STATE_CHANGED,
+    STATE_OFF,
+    STATE_ON
+)
+from homeassistant.setup import setup_component
+import homeassistant.components.datadog as datadog
+import homeassistant.core as ha
+
+from tests.common import (assert_setup_component, get_test_home_assistant)
+
+
+class TestDatadog(unittest.TestCase):
+    """Test the Datadog component."""
+
+    def setUp(self):  # pylint: disable=invalid-name
+        """Setup things to be run when tests are started."""
+        self.hass = get_test_home_assistant()
+
+    def tearDown(self):  # pylint: disable=invalid-name
+        """Stop everything that was started."""
+        self.hass.stop()
+
+    def test_invalid_config(self):
+        """Test invalid configuration."""
+        with assert_setup_component(0):
+            assert not setup_component(self.hass, datadog.DOMAIN, {
+                datadog.DOMAIN: {
+                    'host1': 'host1'
+                }
+            })
+
+    @mock.patch('datadog.initialize')
+    def test_datadog_setup_full(self, mock_connection):
+        """Test setup with all data."""
+        self.hass.bus.listen = mock.MagicMock()
+
+        assert setup_component(self.hass, datadog.DOMAIN, {
+            datadog.DOMAIN: {
+                'host': 'host',
+                'port': 123,
+                'rate': 1,
+                'prefix': 'foo',
+            }
+        })
+
+        self.assertEqual(mock_connection.call_count, 1)
+        self.assertEqual(
+            mock_connection.call_args,
+            mock.call(statsd_host='host', statsd_port=123)
+        )
+
+        self.assertTrue(self.hass.bus.listen.called)
+        self.assertEqual(EVENT_LOGBOOK_ENTRY,
+                         self.hass.bus.listen.call_args_list[0][0][0])
+        self.assertEqual(EVENT_STATE_CHANGED,
+                         self.hass.bus.listen.call_args_list[1][0][0])
+
+    @mock.patch('datadog.initialize')
+    def test_datadog_setup_defaults(self, mock_connection):
+        """Test setup with defaults."""
+        self.hass.bus.listen = mock.MagicMock()
+
+        assert setup_component(self.hass, datadog.DOMAIN, {
+            datadog.DOMAIN: {
+                'host': 'host',
+                'port': datadog.DEFAULT_PORT,
+                'prefix': datadog.DEFAULT_PREFIX,
+            }
+        })
+
+        self.assertEqual(mock_connection.call_count, 1)
+        self.assertEqual(
+            mock_connection.call_args,
+            mock.call(statsd_host='host', statsd_port=8125)
+        )
+        self.assertTrue(self.hass.bus.listen.called)
+
+    @mock.patch('datadog.statsd')
+    def test_logbook_entry(self, mock_client):
+        """Test event listener."""
+        self.hass.bus.listen = mock.MagicMock()
+
+        assert setup_component(self.hass, datadog.DOMAIN, {
+            datadog.DOMAIN: {
+                'host': 'host',
+                'rate': datadog.DEFAULT_RATE,
+            }
+        })
+
+        self.assertTrue(self.hass.bus.listen.called)
+        handler_method = self.hass.bus.listen.call_args_list[0][0][1]
+
+        event = {
+            'domain': 'automation',
+            'entity_id': 'sensor.foo.bar',
+            'message': 'foo bar biz',
+            'name': 'triggered something'
+        }
+        handler_method(mock.MagicMock(data=event))
+
+        self.assertEqual(mock_client.event.call_count, 1)
+        self.assertEqual(
+            mock_client.event.call_args,
+            mock.call(
+                title="Home Assistant",
+                text="%%% \n **{}** {} \n %%%".format(
+                    event['name'],
+                    event['message']
+                ),
+                tags=["entity:sensor.foo.bar", "domain:automation"]
+            )
+        )
+
+        mock_client.event.reset_mock()
+
+    @mock.patch('datadog.statsd')
+    def test_state_changed(self, mock_client):
+        """Test event listener."""
+        self.hass.bus.listen = mock.MagicMock()
+
+        assert setup_component(self.hass, datadog.DOMAIN, {
+            datadog.DOMAIN: {
+                'host': 'host',
+                'prefix': 'ha',
+                'rate': datadog.DEFAULT_RATE,
+            }
+        })
+
+        self.assertTrue(self.hass.bus.listen.called)
+        handler_method = self.hass.bus.listen.call_args_list[1][0][1]
+
+        valid = {
+            '1': 1,
+            '1.0': 1.0,
+            STATE_ON: 1,
+            STATE_OFF: 0
+        }
+
+        attributes = {
+            'elevation': 3.2,
+            'temperature': 5.0
+        }
+
+        for in_, out in valid.items():
+            state = mock.MagicMock(domain="sensor", entity_id="sensor.foo.bar",
+                                   state=in_, attributes=attributes)
+            handler_method(mock.MagicMock(data={'new_state': state}))
+
+            self.assertEqual(mock_client.gauge.call_count, 3)
+
+            for attribute, value in attributes.items():
+                mock_client.gauge.assert_has_calls([
+                    mock.call(
+                        "ha.sensor.{}".format(attribute),
+                        value,
+                        sample_rate=1,
+                        tags=["entity:{}".format(state.entity_id)]
+                    )
+                ])
+
+            self.assertEqual(
+                mock_client.gauge.call_args,
+                mock.call("ha.sensor", out, sample_rate=1, tags=[
+                    "entity:{}".format(state.entity_id)
+                ])
+            )
+
+            mock_client.gauge.reset_mock()
+
+        for invalid in ('foo', '', object):
+            handler_method(mock.MagicMock(data={
+                'new_state': ha.State('domain.test', invalid, {})}))
+            self.assertFalse(mock_client.gauge.called)