Pass kwargs to orig_async_create_task in test (#3529)

hacs · Feb 28, 2024 · a422090 · a422090
1 parent 7d46a52
commit a422090
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/common.py b/tests/common.py
@@ -124,14 +124,14 @@ def async_add_executor_job(target, *args):
 
  return orig_async_add_executor_job(target, *args)
 
- def async_create_task(coroutine, *args):
+ def async_create_task(coroutine, *args, **kwargs):
  """Create task."""
  if isinstance(coroutine, Mock) and not isinstance(coroutine, AsyncMock):
  fut = asyncio.Future()
  fut.set_result(None)
  return fut
 
- return orig_async_create_task(coroutine, *args)
+ return orig_async_create_task(coroutine, *args, **kwargs)
 
  hass.async_add_job = async_add_job
  hass.async_add_executor_job = async_add_executor_job