Additional tests for Command Sensor.
1. Moved template testing out of main test. 2. Added test for bad command.pull/1049/head
parent
97e867052d
commit
6a08f14120
|
@ -17,17 +17,15 @@ class TestCommandSensorSensor(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.hass = ha.HomeAssistant()
|
self.hass = ha.HomeAssistant()
|
||||||
|
|
||||||
self.config = {'name': 'Test',
|
|
||||||
'unit_of_measurement': 'in',
|
|
||||||
'command': 'echo 5',
|
|
||||||
'value_template': '{{ value }}'}
|
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
""" Stop down stuff we started. """
|
""" Stop down stuff we started. """
|
||||||
self.hass.stop()
|
self.hass.stop()
|
||||||
|
|
||||||
def test_setup(self):
|
def test_setup(self):
|
||||||
""" Test sensor setup """
|
""" Test sensor setup """
|
||||||
|
config = {'name': 'Test',
|
||||||
|
'unit_of_measurement': 'in',
|
||||||
|
'command': 'echo 5'}
|
||||||
devices = []
|
devices = []
|
||||||
|
|
||||||
def add_dev_callback(devs):
|
def add_dev_callback(devs):
|
||||||
|
@ -36,10 +34,26 @@ class TestCommandSensorSensor(unittest.TestCase):
|
||||||
devices.append(dev)
|
devices.append(dev)
|
||||||
|
|
||||||
command_sensor.setup_platform(
|
command_sensor.setup_platform(
|
||||||
self.hass, self.config, add_dev_callback)
|
self.hass, config, add_dev_callback)
|
||||||
|
|
||||||
self.assertEqual(1, len(devices))
|
self.assertEqual(1, len(devices))
|
||||||
entity = devices[0]
|
entity = devices[0]
|
||||||
self.assertEqual('Test', entity.name)
|
self.assertEqual('Test', entity.name)
|
||||||
self.assertEqual('in', entity.unit_of_measurement)
|
self.assertEqual('in', entity.unit_of_measurement)
|
||||||
self.assertEqual('5', entity.state)
|
self.assertEqual('5', entity.state)
|
||||||
|
|
||||||
|
def test_template(self):
|
||||||
|
""" Test command sensor with template """
|
||||||
|
data = command_sensor.CommandSensorData('echo 50')
|
||||||
|
|
||||||
|
entity = command_sensor.CommandSensor(
|
||||||
|
self.hass, data, 'test', 'in', '{{ value | multiply(0.1) }}')
|
||||||
|
|
||||||
|
self.assertEqual(5, float(entity.state))
|
||||||
|
|
||||||
|
def test_bad_command(self):
|
||||||
|
""" Test bad command """
|
||||||
|
data = command_sensor.CommandSensorData('asdfasdf')
|
||||||
|
data.update()
|
||||||
|
|
||||||
|
self.assertEqual(None, data.value)
|
||||||
|
|
Loading…
Reference in New Issue