From 97e867052d2b88dc487e473ac681aaef27303247 Mon Sep 17 00:00:00 2001
From: Ryan Kraus <rmkraus@gmail.com>
Date: Sat, 30 Jan 2016 22:01:10 -0500
Subject: [PATCH] Added tests for command sensor

Added tests to create and check basic functionality of command sensor.
---
 .../components/sensor/test_command_sensor.py  | 45 +++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 tests/components/sensor/test_command_sensor.py

diff --git a/tests/components/sensor/test_command_sensor.py b/tests/components/sensor/test_command_sensor.py
new file mode 100644
index 00000000000..7059e715ffb
--- /dev/null
+++ b/tests/components/sensor/test_command_sensor.py
@@ -0,0 +1,45 @@
+"""
+tests.components.sensor.command_sensor
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+Tests command sensor.
+"""
+
+import unittest
+
+import homeassistant.core as ha
+from homeassistant.components.sensor import command_sensor
+
+
+class TestCommandSensorSensor(unittest.TestCase):
+    """ Test the Template sensor. """
+
+    def setUp(self):
+        self.hass = ha.HomeAssistant()
+
+        self.config = {'name': 'Test',
+                       'unit_of_measurement': 'in',
+                       'command': 'echo 5',
+                       'value_template': '{{ value }}'}
+
+    def tearDown(self):
+        """ Stop down stuff we started. """
+        self.hass.stop()
+
+    def test_setup(self):
+        """ Test sensor setup """
+        devices = []
+
+        def add_dev_callback(devs):
+            """ callback to add device """
+            for dev in devs:
+                devices.append(dev)
+
+        command_sensor.setup_platform(
+            self.hass, self.config, add_dev_callback)
+
+        self.assertEqual(1, len(devices))
+        entity = devices[0]
+        self.assertEqual('Test', entity.name)
+        self.assertEqual('in', entity.unit_of_measurement)
+        self.assertEqual('5', entity.state)