166 lines
5.0 KiB
Python
166 lines
5.0 KiB
Python
"""Test Group config panel."""
|
|
from http import HTTPStatus
|
|
import json
|
|
from pathlib import Path
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from homeassistant.bootstrap import async_setup_component
|
|
from homeassistant.components import config
|
|
from homeassistant.components.config import group
|
|
from homeassistant.util.file import write_utf8_file
|
|
from homeassistant.util.yaml import dump, load_yaml
|
|
|
|
VIEW_NAME = "api:config:group:config"
|
|
|
|
|
|
async def test_get_device_config(hass, hass_client):
|
|
"""Test getting device config."""
|
|
with patch.object(config, "SECTIONS", ["group"]):
|
|
await async_setup_component(hass, "config", {})
|
|
|
|
client = await hass_client()
|
|
|
|
def mock_read(path):
|
|
"""Mock reading data."""
|
|
return {"hello.beer": {"free": "beer"}, "other.entity": {"do": "something"}}
|
|
|
|
with patch("homeassistant.components.config._read", mock_read):
|
|
resp = await client.get("/api/config/group/config/hello.beer")
|
|
|
|
assert resp.status == HTTPStatus.OK
|
|
result = await resp.json()
|
|
|
|
assert result == {"free": "beer"}
|
|
|
|
|
|
async def test_update_device_config(hass, hass_client):
|
|
"""Test updating device config."""
|
|
with patch.object(config, "SECTIONS", ["group"]):
|
|
await async_setup_component(hass, "config", {})
|
|
|
|
client = await hass_client()
|
|
|
|
orig_data = {
|
|
"hello.beer": {"ignored": True},
|
|
"other.entity": {"polling_intensity": 2},
|
|
}
|
|
|
|
def mock_read(path):
|
|
"""Mock reading data."""
|
|
return orig_data
|
|
|
|
written = []
|
|
|
|
def mock_write(path, data):
|
|
"""Mock writing data."""
|
|
written.append(data)
|
|
|
|
mock_call = AsyncMock()
|
|
|
|
with patch("homeassistant.components.config._read", mock_read), patch(
|
|
"homeassistant.components.config._write", mock_write
|
|
), patch.object(hass.services, "async_call", mock_call):
|
|
resp = await client.post(
|
|
"/api/config/group/config/hello_beer",
|
|
data=json.dumps(
|
|
{"name": "Beer", "entities": ["light.top", "light.bottom"]}
|
|
),
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert resp.status == HTTPStatus.OK
|
|
result = await resp.json()
|
|
assert result == {"result": "ok"}
|
|
|
|
orig_data["hello_beer"]["name"] = "Beer"
|
|
orig_data["hello_beer"]["entities"] = ["light.top", "light.bottom"]
|
|
|
|
assert written[0] == orig_data
|
|
mock_call.assert_called_once_with("group", "reload")
|
|
|
|
|
|
async def test_update_device_config_invalid_key(hass, hass_client):
|
|
"""Test updating device config."""
|
|
with patch.object(config, "SECTIONS", ["group"]):
|
|
await async_setup_component(hass, "config", {})
|
|
|
|
client = await hass_client()
|
|
|
|
resp = await client.post(
|
|
"/api/config/group/config/not a slug", data=json.dumps({"name": "YO"})
|
|
)
|
|
|
|
assert resp.status == HTTPStatus.BAD_REQUEST
|
|
|
|
|
|
async def test_update_device_config_invalid_data(hass, hass_client):
|
|
"""Test updating device config."""
|
|
with patch.object(config, "SECTIONS", ["group"]):
|
|
await async_setup_component(hass, "config", {})
|
|
|
|
client = await hass_client()
|
|
|
|
resp = await client.post(
|
|
"/api/config/group/config/hello_beer", data=json.dumps({"invalid_option": 2})
|
|
)
|
|
|
|
assert resp.status == HTTPStatus.BAD_REQUEST
|
|
|
|
|
|
async def test_update_device_config_invalid_json(hass, hass_client):
|
|
"""Test updating device config."""
|
|
with patch.object(config, "SECTIONS", ["group"]):
|
|
await async_setup_component(hass, "config", {})
|
|
|
|
client = await hass_client()
|
|
|
|
resp = await client.post("/api/config/group/config/hello_beer", data="not json")
|
|
|
|
assert resp.status == HTTPStatus.BAD_REQUEST
|
|
|
|
|
|
async def test_update_config_write_to_temp_file(hass, hass_client, tmpdir):
|
|
"""Test config with a temp file."""
|
|
test_dir = await hass.async_add_executor_job(tmpdir.mkdir, "files")
|
|
group_yaml = Path(test_dir / "group.yaml")
|
|
|
|
with patch.object(group, "GROUP_CONFIG_PATH", group_yaml), patch.object(
|
|
config, "SECTIONS", ["group"]
|
|
):
|
|
await async_setup_component(hass, "config", {})
|
|
|
|
client = await hass_client()
|
|
|
|
orig_data = {
|
|
"hello.beer": {"ignored": True},
|
|
"other.entity": {"polling_intensity": 2},
|
|
}
|
|
contents = dump(orig_data)
|
|
await hass.async_add_executor_job(write_utf8_file, group_yaml, contents)
|
|
|
|
mock_call = AsyncMock()
|
|
|
|
with patch.object(hass.services, "async_call", mock_call):
|
|
resp = await client.post(
|
|
"/api/config/group/config/hello_beer",
|
|
data=json.dumps(
|
|
{"name": "Beer", "entities": ["light.top", "light.bottom"]}
|
|
),
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert resp.status == HTTPStatus.OK
|
|
result = await resp.json()
|
|
assert result == {"result": "ok"}
|
|
|
|
new_data = await hass.async_add_executor_job(load_yaml, group_yaml)
|
|
|
|
assert new_data == {
|
|
**orig_data,
|
|
"hello_beer": {
|
|
"name": "Beer",
|
|
"entities": ["light.top", "light.bottom"],
|
|
},
|
|
}
|
|
mock_call.assert_called_once_with("group", "reload")
|