"""Tests for config/script.""" from http import HTTPStatus import json from unittest.mock import patch import pytest from homeassistant.bootstrap import async_setup_component from homeassistant.components import config from homeassistant.core import HomeAssistant from homeassistant.helpers import entity_registry as er from tests.components.blueprint.conftest import stub_blueprint_populate # noqa: F401 @pytest.fixture(autouse=True) async def setup_script(hass, script_config, stub_blueprint_populate): # noqa: F811 """Set up script integration.""" assert await async_setup_component(hass, "script", {"script": script_config}) @pytest.mark.parametrize("script_config", ({},)) async def test_get_script_config(hass: HomeAssistant, hass_client, hass_config_store): """Test getting script config.""" with patch.object(config, "SECTIONS", ["script"]): await async_setup_component(hass, "config", {}) client = await hass_client() hass_config_store["scripts.yaml"] = { "sun": {"alias": "Sun"}, "moon": {"alias": "Moon"}, } resp = await client.get("/api/config/script/config/moon") assert resp.status == HTTPStatus.OK result = await resp.json() assert result == {"alias": "Moon"} @pytest.mark.parametrize("script_config", ({},)) async def test_update_script_config( hass: HomeAssistant, hass_client, hass_config_store ): """Test updating script config.""" with patch.object(config, "SECTIONS", ["script"]): await async_setup_component(hass, "config", {}) assert sorted(hass.states.async_entity_ids("script")) == [] client = await hass_client() orig_data = {"sun": {"alias": "Sun"}, "moon": {"alias": "Moon"}} hass_config_store["scripts.yaml"] = orig_data resp = await client.post( "/api/config/script/config/moon", data=json.dumps({"alias": "Moon updated", "sequence": []}), ) await hass.async_block_till_done() assert sorted(hass.states.async_entity_ids("script")) == ["script.moon"] assert resp.status == HTTPStatus.OK result = await resp.json() assert result == {"result": "ok"} new_data = hass_config_store["scripts.yaml"] assert list(new_data["moon"]) == ["alias", "sequence"] assert new_data["moon"] == {"alias": "Moon updated", "sequence": []} @pytest.mark.parametrize("script_config", ({},)) async def test_update_script_config_with_error( hass: HomeAssistant, hass_client, hass_config_store, caplog ): """Test updating script config with errors.""" with patch.object(config, "SECTIONS", ["script"]): await async_setup_component(hass, "config", {}) assert sorted(hass.states.async_entity_ids("script")) == [] client = await hass_client() orig_data = {"sun": {}, "moon": {}} hass_config_store["scripts.yaml"] = orig_data resp = await client.post( "/api/config/script/config/moon", data=json.dumps({}), ) await hass.async_block_till_done() assert sorted(hass.states.async_entity_ids("script")) == [] assert resp.status != HTTPStatus.OK result = await resp.json() validation_error = "required key not provided @ data['sequence']" assert result == {"message": f"Message malformed: {validation_error}"} # Assert the validation error is not logged assert validation_error not in caplog.text @pytest.mark.parametrize("script_config", ({},)) async def test_update_remove_key_script_config( hass: HomeAssistant, hass_client, hass_config_store ): """Test updating script config while removing a key.""" with patch.object(config, "SECTIONS", ["script"]): await async_setup_component(hass, "config", {}) assert sorted(hass.states.async_entity_ids("script")) == [] client = await hass_client() orig_data = {"sun": {"key": "value"}, "moon": {"key": "value"}} hass_config_store["scripts.yaml"] = orig_data resp = await client.post( "/api/config/script/config/moon", data=json.dumps({"sequence": []}), ) await hass.async_block_till_done() assert sorted(hass.states.async_entity_ids("script")) == ["script.moon"] assert resp.status == HTTPStatus.OK result = await resp.json() assert result == {"result": "ok"} new_data = hass_config_store["scripts.yaml"] assert list(new_data["moon"]) == ["sequence"] assert new_data["moon"] == {"sequence": []} @pytest.mark.parametrize( "script_config", ( { "one": {"alias": "Light on", "sequence": []}, "two": {"alias": "Light off", "sequence": []}, }, ), ) async def test_delete_script(hass: HomeAssistant, hass_client, hass_config_store): """Test deleting a script.""" with patch.object(config, "SECTIONS", ["script"]): await async_setup_component(hass, "config", {}) assert sorted(hass.states.async_entity_ids("script")) == [ "script.one", "script.two", ] ent_reg = er.async_get(hass) assert len(ent_reg.entities) == 2 client = await hass_client() orig_data = {"one": {}, "two": {}} hass_config_store["scripts.yaml"] = orig_data resp = await client.delete("/api/config/script/config/two") await hass.async_block_till_done() assert sorted(hass.states.async_entity_ids("script")) == [ "script.one", ] assert resp.status == HTTPStatus.OK result = await resp.json() assert result == {"result": "ok"} assert hass_config_store["scripts.yaml"] == {"one": {}} assert len(ent_reg.entities) == 1