Don't allow partial update of input_boolean settings (#78372)

tests/components/input_boolean/test_init.py

@@ -40,7 +40,11 @@ def storage_setup(hass, hass_storage):
                 "data": {"items": [{"id": "from_storage", "name": "from storage"}]},
             }
         else:
-            hass_storage[DOMAIN] = items
+            hass_storage[DOMAIN] = {
+                "key": DOMAIN,
+                "version": 1,
+                "data": {"items": items},
+            }
         if config is None:
             config = {DOMAIN: {}}
         return await async_setup_component(hass, DOMAIN, config)
@@ -332,6 +336,89 @@ async def test_ws_delete(hass, hass_ws_client, storage_setup):
     assert ent_reg.async_get_entity_id(DOMAIN, DOMAIN, input_id) is None
 
 
+async def test_ws_update(hass, hass_ws_client, storage_setup):
+    """Test update WS."""
+
+    settings = {
+        "name": "from storage",
+    }
+    items = [{"id": "from_storage"} | settings]
+    assert await storage_setup(items)
+
+    input_id = "from_storage"
+    input_entity_id = f"{DOMAIN}.{input_id}"
+    ent_reg = er.async_get(hass)
+
+    state = hass.states.get(input_entity_id)
+    assert state is not None
+    assert state.state
+    assert ent_reg.async_get_entity_id(DOMAIN, DOMAIN, input_id) is not None
+
+    client = await hass_ws_client(hass)
+
+    updated_settings = settings | {"name": "new_name", "icon": "mdi:blah"}
+    await client.send_json(
+        {
+            "id": 6,
+            "type": f"{DOMAIN}/update",
+            f"{DOMAIN}_id": f"{input_id}",
+            **updated_settings,
+        }
+    )
+    resp = await client.receive_json()
+    assert resp["success"]
+    assert resp["result"] == {"id": "from_storage"} | updated_settings
+
+    state = hass.states.get(input_entity_id)
+    assert state.attributes["icon"] == "mdi:blah"
+    assert state.attributes["friendly_name"] == "new_name"
+
+    updated_settings = settings | {"name": "new_name_2"}
+    await client.send_json(
+        {
+            "id": 7,
+            "type": f"{DOMAIN}/update",
+            f"{DOMAIN}_id": f"{input_id}",
+            **updated_settings,
+        }
+    )
+    resp = await client.receive_json()
+    assert resp["success"]
+    assert resp["result"] == {"id": "from_storage"} | updated_settings
+
+    state = hass.states.get(input_entity_id)
+    assert "icon" not in state.attributes
+    assert state.attributes["friendly_name"] == "new_name_2"
+
+
+async def test_ws_create(hass, hass_ws_client, storage_setup):
+    """Test create WS."""
+    assert await storage_setup(items=[])
+
+    input_id = "new_input"
+    input_entity_id = f"{DOMAIN}.{input_id}"
+    ent_reg = er.async_get(hass)
+
+    state = hass.states.get(input_entity_id)
+    assert state is None
+    assert ent_reg.async_get_entity_id(DOMAIN, DOMAIN, input_id) is None
+
+    client = await hass_ws_client(hass)
+
+    await client.send_json(
+        {
+            "id": 6,
+            "type": f"{DOMAIN}/create",
+            "name": "New Input",
+        }
+    )
+    resp = await client.receive_json()
+    assert resp["success"]
+
+    state = hass.states.get(input_entity_id)
+    assert state.state
+
+
 async def test_setup_no_config(hass, hass_admin_user):
     """Test component setup with no config."""
     count_start = len(hass.states.async_entity_ids())