From 0a904acd4dd163a7c09c5efd62b7b3dd1baebbeb Mon Sep 17 00:00:00 2001
From: Dan Smith <dsmith@danplanet.com>
Date: Fri, 12 Feb 2016 01:41:21 +0000
Subject: [PATCH] Add some tests for splunk

This also fixes issue #1214, and I think another bug. The splunk
code will just take the value of state.state and try to serialize
it to json if it can't make it into a number. It did this before
I generalized that code. Since json.dumps() will fail on most anything
complicated, I think the right thing to do is *not* try to do that.
---
 homeassistant/components/splunk.py |  4 +-
 tests/components/test_splunk.py    | 93 ++++++++++++++++++++++++++++++
 2 files changed, 95 insertions(+), 2 deletions(-)
 create mode 100644 tests/components/test_splunk.py

diff --git a/homeassistant/components/splunk.py b/homeassistant/components/splunk.py
index da773cbe0f7..e9395ca3587 100644
--- a/homeassistant/components/splunk.py
+++ b/homeassistant/components/splunk.py
@@ -50,7 +50,7 @@ def setup(hass, config):
         uri_scheme = "https://"
     else:
         uri_scheme = "http://"
-    event_collector = uri_scheme + host + ":" + port + \
+    event_collector = uri_scheme + host + ":" + str(port) + \
         "/services/collector/event"
     headers = {'Authorization': 'Splunk ' + token}
 
@@ -65,7 +65,7 @@ def setup(hass, config):
         try:
             _state = state_helper.state_as_number(state)
         except ValueError:
-            _state = state.state
+            return
 
         json_body = [
             {
diff --git a/tests/components/test_splunk.py b/tests/components/test_splunk.py
new file mode 100644
index 00000000000..18814e9d9f7
--- /dev/null
+++ b/tests/components/test_splunk.py
@@ -0,0 +1,93 @@
+"""
+tests.components.test_splunk
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+Tests splunk component.
+"""
+import unittest
+from unittest import mock
+
+import homeassistant.components.splunk as splunk
+from homeassistant.const import STATE_ON, STATE_OFF, EVENT_STATE_CHANGED
+
+
+class TestSplunk(unittest.TestCase):
+    def test_setup_config_full(self):
+        config = {
+            'splunk': {
+                'host': 'host',
+                'port': 123,
+                'token': 'secret',
+                'use_ssl': 'False',
+            }
+        }
+        hass = mock.MagicMock()
+        self.assertTrue(splunk.setup(hass, config))
+        self.assertTrue(hass.bus.listen.called)
+        self.assertEqual(EVENT_STATE_CHANGED,
+                         hass.bus.listen.call_args_list[0][0][0])
+
+    def test_setup_config_defaults(self):
+        config = {
+            'splunk': {
+                'host': 'host',
+                'token': 'secret',
+            }
+        }
+        hass = mock.MagicMock()
+        self.assertTrue(splunk.setup(hass, config))
+        self.assertTrue(hass.bus.listen.called)
+        self.assertEqual(EVENT_STATE_CHANGED,
+                         hass.bus.listen.call_args_list[0][0][0])
+
+
+    def _setup(self, mock_requests):
+        self.mock_post = mock_requests.post
+        self.mock_request_exception = Exception
+        mock_requests.exceptions.RequestException = self.mock_request_exception
+        config = {
+            'splunk': {
+                'host': 'host',
+                'token': 'secret',
+            }
+        }
+        self.hass = mock.MagicMock()
+        splunk.setup(self.hass, config)
+        self.handler_method = self.hass.bus.listen.call_args_list[0][0][1]
+
+    @mock.patch.object(splunk, 'requests')
+    @mock.patch('json.dumps')
+    def test_event_listener(self, mock_dump, mock_requests):
+        mock_dump.side_effect = lambda x: x
+        self._setup(mock_requests)
+
+        valid = {'1': 1,
+                 '1.0': 1.0,
+                 STATE_ON: 1,
+                 STATE_OFF: 0}
+        for in_, out in valid.items():
+            state = mock.MagicMock(state=in_,
+                                   domain='fake',
+                                   object_id='entity',
+                                   attributes={})
+            event = mock.MagicMock(data={'new_state': state},
+                                   time_fired=12345)
+            body = [{
+                'domain': 'fake',
+                'entity_id': 'entity',
+                'attributes': {},
+                'time': '12345',
+                'value': out,
+            }]
+            payload = {'host': 'http://host:8088/services/collector/event',
+                       'event': body}
+            self.handler_method(event)
+            self.mock_post.assert_called_once_with(
+                payload['host'], data=payload,
+                headers={'Authorization': 'Splunk secret'})
+            self.mock_post.reset_mock()
+
+        for invalid in ('foo', '', object):
+            state = mock.MagicMock(state=invalid)
+            self.handler_method(mock.MagicMock(data={'new_state': state}))
+            self.assertFalse(self.mock_post.called)