From 0137998dbabeb2d91f9b150b6af9f40a9e9e1b83 Mon Sep 17 00:00:00 2001 From: Patrick Arminio Date: Thu, 12 Dec 2024 17:55:53 +0000 Subject: [PATCH] Update test --- strawberry/extensions/base_extension.py | 2 +- tests/benchmarks/test_execute_with_extensions.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/strawberry/extensions/base_extension.py b/strawberry/extensions/base_extension.py index 3cc7a0db96..68c40bedd8 100644 --- a/strawberry/extensions/base_extension.py +++ b/strawberry/extensions/base_extension.py @@ -58,7 +58,7 @@ def should_await(self, _next: Callable) -> bool: some performance hits, especially because we know if the resolver is async or not at schema creation time. """ - return _next._is_async + return _next._is_async # type: ignore def resolve( self, diff --git a/tests/benchmarks/test_execute_with_extensions.py b/tests/benchmarks/test_execute_with_extensions.py index 0337ac3407..55a0af4a82 100644 --- a/tests/benchmarks/test_execute_with_extensions.py +++ b/tests/benchmarks/test_execute_with_extensions.py @@ -1,5 +1,4 @@ import asyncio -from inspect import isawaitable from pathlib import Path from typing import Any, Dict, List @@ -21,7 +20,7 @@ def get_results(self) -> AwaitableOrValue[Dict[str, Any]]: class ResolveExtension(SchemaExtension): async def resolve(self, _next, root, info, *args: Any, **kwargs: Any) -> Any: result = _next(root, info, *args, **kwargs) - if isawaitable(result): + if self.should_await(_next): result = await result return result