136 lines
3.8 KiB
Python
136 lines
3.8 KiB
Python
"""The tests the cover command line platform."""
|
|
import os
|
|
from os import path
|
|
import tempfile
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from homeassistant import config as hass_config
|
|
import homeassistant.components.command_line.cover as cmd_rs
|
|
from homeassistant.components.cover import DOMAIN
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
SERVICE_CLOSE_COVER,
|
|
SERVICE_OPEN_COVER,
|
|
SERVICE_RELOAD,
|
|
SERVICE_STOP_COVER,
|
|
)
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.async_mock import patch
|
|
|
|
|
|
@pytest.fixture
|
|
def rs(hass):
|
|
"""Return CommandCover instance."""
|
|
return cmd_rs.CommandCover(
|
|
hass,
|
|
"foo",
|
|
"command_open",
|
|
"command_close",
|
|
"command_stop",
|
|
"command_state",
|
|
None,
|
|
15,
|
|
)
|
|
|
|
|
|
def test_should_poll_new(rs):
|
|
"""Test the setting of polling."""
|
|
assert rs.should_poll is True
|
|
rs._command_state = None
|
|
assert rs.should_poll is False
|
|
|
|
|
|
def test_query_state_value(rs):
|
|
"""Test with state value."""
|
|
with mock.patch("subprocess.check_output") as mock_run:
|
|
mock_run.return_value = b" foo bar "
|
|
result = rs._query_state_value("runme")
|
|
assert "foo bar" == result
|
|
assert mock_run.call_count == 1
|
|
assert mock_run.call_args == mock.call(
|
|
"runme", shell=True, timeout=15 # nosec # shell by design
|
|
)
|
|
|
|
|
|
async def test_state_value(hass):
|
|
"""Test with state value."""
|
|
with tempfile.TemporaryDirectory() as tempdirname:
|
|
path = os.path.join(tempdirname, "cover_status")
|
|
test_cover = {
|
|
"command_state": f"cat {path}",
|
|
"command_open": f"echo 1 > {path}",
|
|
"command_close": f"echo 1 > {path}",
|
|
"command_stop": f"echo 0 > {path}",
|
|
"value_template": "{{ value }}",
|
|
}
|
|
assert (
|
|
await async_setup_component(
|
|
hass,
|
|
DOMAIN,
|
|
{"cover": {"platform": "command_line", "covers": {"test": test_cover}}},
|
|
)
|
|
is True
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert "unknown" == hass.states.get("cover.test").state
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_OPEN_COVER, {ATTR_ENTITY_ID: "cover.test"}, blocking=True
|
|
)
|
|
assert "open" == hass.states.get("cover.test").state
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_CLOSE_COVER, {ATTR_ENTITY_ID: "cover.test"}, blocking=True
|
|
)
|
|
assert "open" == hass.states.get("cover.test").state
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN, SERVICE_STOP_COVER, {ATTR_ENTITY_ID: "cover.test"}, blocking=True
|
|
)
|
|
assert "closed" == hass.states.get("cover.test").state
|
|
|
|
|
|
async def test_reload(hass):
|
|
"""Verify we can reload command_line covers."""
|
|
|
|
test_cover = {
|
|
"command_state": "echo open",
|
|
"value_template": "{{ value }}",
|
|
}
|
|
await async_setup_component(
|
|
hass,
|
|
DOMAIN,
|
|
{"cover": {"platform": "command_line", "covers": {"test": test_cover}}},
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_all()) == 1
|
|
assert hass.states.get("cover.test").state
|
|
|
|
yaml_path = path.join(
|
|
_get_fixtures_base_path(),
|
|
"fixtures",
|
|
"command_line/configuration.yaml",
|
|
)
|
|
with patch.object(hass_config, "YAML_CONFIG_FILE", yaml_path):
|
|
await hass.services.async_call(
|
|
"command_line",
|
|
SERVICE_RELOAD,
|
|
{},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_all()) == 1
|
|
|
|
assert hass.states.get("cover.test") is None
|
|
assert hass.states.get("cover.from_yaml")
|
|
|
|
|
|
def _get_fixtures_base_path():
|
|
return path.dirname(path.dirname(path.dirname(__file__)))
|