hass-core/tests/components/vera/test_cover.py
Robert Van Gorkom ae22b5187a
Add vera config entries support (#29880)
* Adding vera config entries support.

* Fixing lint error.

* Applying minimal changes necessary to get config entries working.

* Addressing PR feedback by further reducing the scope of the change.

* Addressing PR feedback.

* Fixing pyvera import to make it easier to patch.
Addressing PR feedback regarding creation of controller and scheduling of async config flow actions.

* Updating code owners file.

* Small fixes.

* Adding a user config flow step.

* Adding optional configs for user config flow.

* Updating strings to be more clear to the user.

* Adding options flow.
Fixing some PR feedback.

* Better handling of options.
PR feedback changes.

* Using config registry to update config options.

* Better managing config from file or config from UI
Disabling config through UI if config is provided from a file.
More tests to account for these adjustments.

* Address PR feedback.

* Fixing test, merging with master.

* Disabling all Vera UI for configs managed by configuration.yml.
Adding more tests.

* Updating config based on unique_id.
Addressing additional PR feedback.

* Rebasing off dev.
Addressing feedback.

* Addressing PR feedback.
2020-04-03 09:49:50 +02:00

76 lines
2.7 KiB
Python

"""Vera tests."""
from unittest.mock import MagicMock
import pyvera as pv
from homeassistant.core import HomeAssistant
from .common import ComponentFactory, new_simple_controller_config
async def test_cover(
hass: HomeAssistant, vera_component_factory: ComponentFactory
) -> None:
"""Test function."""
vera_device = MagicMock(spec=pv.VeraCurtain) # type: pv.VeraCurtain
vera_device.device_id = 1
vera_device.name = "dev1"
vera_device.category = pv.CATEGORY_CURTAIN
vera_device.is_closed = False
vera_device.get_level.return_value = 0
entity_id = "cover.dev1_1"
component_data = await vera_component_factory.configure_component(
hass=hass,
controller_config=new_simple_controller_config(devices=(vera_device,)),
)
update_callback = component_data.controller_data.update_callback
assert hass.states.get(entity_id).state == "closed"
assert hass.states.get(entity_id).attributes["current_position"] == 0
await hass.services.async_call(
"cover", "open_cover", {"entity_id": entity_id},
)
await hass.async_block_till_done()
vera_device.open.assert_called()
vera_device.is_open.return_value = True
vera_device.get_level.return_value = 100
update_callback(vera_device)
await hass.async_block_till_done()
assert hass.states.get(entity_id).state == "open"
assert hass.states.get(entity_id).attributes["current_position"] == 100
await hass.services.async_call(
"cover", "set_cover_position", {"entity_id": entity_id, "position": 50},
)
await hass.async_block_till_done()
vera_device.set_level.assert_called_with(50)
vera_device.is_open.return_value = True
vera_device.get_level.return_value = 50
update_callback(vera_device)
await hass.async_block_till_done()
assert hass.states.get(entity_id).state == "open"
assert hass.states.get(entity_id).attributes["current_position"] == 50
await hass.services.async_call(
"cover", "stop_cover", {"entity_id": entity_id},
)
await hass.async_block_till_done()
vera_device.stop.assert_called()
update_callback(vera_device)
await hass.async_block_till_done()
assert hass.states.get(entity_id).state == "open"
assert hass.states.get(entity_id).attributes["current_position"] == 50
await hass.services.async_call(
"cover", "close_cover", {"entity_id": entity_id},
)
await hass.async_block_till_done()
vera_device.close.assert_called()
vera_device.is_open.return_value = False
vera_device.get_level.return_value = 00
update_callback(vera_device)
await hass.async_block_till_done()
assert hass.states.get(entity_id).state == "closed"
assert hass.states.get(entity_id).attributes["current_position"] == 00