jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-27 10:47:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve the test class used for testing FlowManager.async_show_progress (#107786)

Browse Source 
* Improve the test class used for testing FlowManager.async_show_progress

* Address review comments
This commit is contained in:
Erik Montnemery 2024-01-14 18:39:45 +01:00 committed by GitHub
parent 7a6dca0987
commit 00165fef5b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/test_data_entry_flow.py
View File

@ -358,30 +358,39 @@ async def test_show_progress(hass: HomeAssistant, manager) -> None:
VERSION = 5
data = None
start_task_two = False
progress_task: asyncio.Task[None] | None = None
task_one: asyncio.Task[None] | None = None
task_two: asyncio.Task[None] | None = None
async def async_step_init(self, user_input=None):
async def long_running_task_one() -> None:
async def long_running_job_one() -> None:
await task_one_evt.wait()
self.start_task_two = True
async def long_running_task_two() -> None:
async def long_running_job_two() -> None:
await task_two_evt.wait()
self.data = {"title": "Hello"}
if not task_one_evt.is_set():
uncompleted_task: asyncio.Task[None] | None = None
if not self.task_one:
self.task_one = hass.async_create_task(long_running_job_one())
progress_action = None
if not self.task_one.done():
progress_action = "task_one"
if not self.progress_task:
self.progress_task = hass.async_create_task(long_running_task_one())
elif not task_two_evt.is_set():
uncompleted_task = self.task_one
if not uncompleted_task:
if not self.task_two:
self.task_two = hass.async_create_task(long_running_job_two())
if not self.task_two.done():
progress_action = "task_two"
if self.start_task_two:
self.progress_task = hass.async_create_task(long_running_task_two())
self.start_task_two = False
if not task_one_evt.is_set() or not task_two_evt.is_set():
uncompleted_task = self.task_two
if uncompleted_task:
assert progress_action
return self.async_show_progress(
progress_action=progress_action,
progress_task=self.progress_task,
progress_task=uncompleted_task,
)
return self.async_show_progress_done(next_step_id="finish")