evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add WS API for debugging previous assist_pipeline runs (#91541)

Browse source 
* Add WS API for debugging previous assist_pipeline runs

* Improve typing
This commit is contained in:
Erik Montnemery 2023-04-17 17:48:02 +02:00 • committed by GitHub
parent b597415b01
commit 0ecd23baee
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 564 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/components/assist_pipeline/test_pipeline.py
View file

@ -5,6 +5,7 @@ from homeassistant.components.assist_pipeline.const import DOMAIN
from homeassistant.components.assist_pipeline.pipeline import (
STORAGE_KEY,
STORAGE_VERSION,
PipelineData,
PipelineStorageCollection,
)
from homeassistant.core import HomeAssistant
@ -42,7 +43,8 @@ async def test_load_datasets(hass: HomeAssistant, init_components) -> None:
]
pipeline_ids = []
store1: PipelineStorageCollection = hass.data[DOMAIN]
pipeline_data: PipelineData = hass.data[DOMAIN]
store1 = pipeline_data.pipeline_store
for pipeline in pipelines:
pipeline_ids.append((await store1.async_create_item(pipeline)).id)
assert len(store1.data) == 3
@ -103,6 +105,7 @@ async def test_loading_datasets_from_storage(
assert await async_setup_component(hass, "assist_pipeline", {})
store: PipelineStorageCollection = hass.data[DOMAIN]
pipeline_data: PipelineData = hass.data[DOMAIN]
store = pipeline_data.pipeline_store
assert len(store.data) == 3
assert store.async_get_preferred_item() == "01GX8ZWBAQYWNB1XV3EXEZ75DY"