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

Merge action and condition traces (#47373)

Browse source 
* Merge action and condition traces

* Update __init__.py

* Add typing to AutomationTrace

* Make trace_get prepare a new trace by default

* Correct typing of trace_cv

* Fix tests
This commit is contained in:
Erik Montnemery 2021-03-06 12:57:21 +01:00 • committed by GitHub
parent 022184176a
commit 2f9d03d115
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 191 additions and 284 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/helpers/test_script.py
View file

@ -17,7 +17,7 @@ from homeassistant import exceptions
import homeassistant.components.scene as scene
from homeassistant.const import ATTR_ENTITY_ID, SERVICE_TURN_ON
from homeassistant.core import Context, CoreState, callback
from homeassistant.helpers import config_validation as cv, script
from homeassistant.helpers import config_validation as cv, script, trace
from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util
@ -45,8 +45,8 @@ def assert_element(trace_element, expected_element, path):
def assert_action_trace(expected):
"""Assert a trace condition sequence is as expected."""
action_trace = script.action_trace_get()
script.action_trace_clear()
action_trace = trace.trace_get(clear=False)
trace.trace_clear()
expected_trace_keys = list(expected.keys())
assert list(action_trace.keys()) == expected_trace_keys
for trace_key_index, key in enumerate(expected_trace_keys):