mirror of
https://github.com/home-assistant/core.git
synced 2025-07-19 11:17:21 +00:00
Fix check_real location guard (#31890)
This commit is contained in:
parent
4dc4f84b37
commit
e3adbc336d
@ -39,7 +39,7 @@ def check_real(func):
|
|||||||
"""Force a function to require a keyword _test_real to be passed in."""
|
"""Force a function to require a keyword _test_real to be passed in."""
|
||||||
|
|
||||||
@functools.wraps(func)
|
@functools.wraps(func)
|
||||||
def guard_func(*args, **kwargs):
|
async def guard_func(*args, **kwargs):
|
||||||
real = kwargs.pop("_test_real", None)
|
real = kwargs.pop("_test_real", None)
|
||||||
|
|
||||||
if not real:
|
if not real:
|
||||||
@ -47,7 +47,7 @@ def check_real(func):
|
|||||||
'Forgot to mock or pass "_test_real=True" to %s', func.__name__
|
'Forgot to mock or pass "_test_real=True" to %s', func.__name__
|
||||||
)
|
)
|
||||||
|
|
||||||
return func(*args, **kwargs)
|
return await func(*args, **kwargs)
|
||||||
|
|
||||||
return guard_func
|
return guard_func
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user