From e3adbc336d226392b08e0a3c69a3a80d1d5ae2dd Mon Sep 17 00:00:00 2001 From: Martin Hjelmare Date: Mon, 17 Feb 2020 00:33:09 +0100 Subject: [PATCH] Fix check_real location guard (#31890) --- tests/conftest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 5a3b2494158..04e584cb158 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -39,7 +39,7 @@ def check_real(func): """Force a function to require a keyword _test_real to be passed in.""" @functools.wraps(func) - def guard_func(*args, **kwargs): + async def guard_func(*args, **kwargs): real = kwargs.pop("_test_real", None) if not real: @@ -47,7 +47,7 @@ def check_real(func): 'Forgot to mock or pass "_test_real=True" to %s', func.__name__ ) - return func(*args, **kwargs) + return await func(*args, **kwargs) return guard_func