56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
"""The tests for the Demo roller shutter platform."""
|
|
import unittest
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
from homeassistant.components.rollershutter import demo
|
|
from tests.common import fire_time_changed, get_test_home_assistant
|
|
|
|
|
|
class TestRollershutterDemo(unittest.TestCase):
|
|
"""Test the Demo roller shutter."""
|
|
|
|
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 down everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_move_up(self):
|
|
"""Test moving the rollershutter up."""
|
|
entity = demo.DemoRollershutter(self.hass, 'test', 100)
|
|
entity.move_up()
|
|
|
|
fire_time_changed(self.hass, dt_util.utcnow())
|
|
self.hass.block_till_done()
|
|
self.assertEqual(90, entity.current_position)
|
|
|
|
def test_move_down(self):
|
|
"""Test moving the rollershutter down."""
|
|
entity = demo.DemoRollershutter(self.hass, 'test', 0)
|
|
entity.move_down()
|
|
|
|
fire_time_changed(self.hass, dt_util.utcnow())
|
|
self.hass.block_till_done()
|
|
self.assertEqual(10, entity.current_position)
|
|
|
|
def test_move_position(self):
|
|
"""Test moving the rollershutter to a specific position."""
|
|
entity = demo.DemoRollershutter(self.hass, 'test', 0)
|
|
entity.move_position(10)
|
|
|
|
fire_time_changed(self.hass, dt_util.utcnow())
|
|
self.hass.block_till_done()
|
|
self.assertEqual(10, entity.current_position)
|
|
|
|
def test_stop(self):
|
|
"""Test stopping the rollershutter."""
|
|
entity = demo.DemoRollershutter(self.hass, 'test', 0)
|
|
entity.move_down()
|
|
entity.stop()
|
|
|
|
fire_time_changed(self.hass, dt_util.utcnow())
|
|
self.hass.block_till_done()
|
|
self.assertEqual(0, entity.current_position)
|