Fix tests

mirumee · Feb 14, 2024 · 85fedf2 · 85fedf2
1 parent 68c84f4
commit 85fedf2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_proxy_resolver.py b/tests/test_proxy_resolver.py
@@ -118,9 +118,9 @@ async def test_proxy_resolver_proxies_headers_via_callable(
     schema,
     root_value,
 ):
-    def proxy_headers(_, info, payload):
+    def proxy_headers(context):
         return {
-            "x-auth": info.context["headers"].get("authorization"),
+            "x-auth": context["headers"].get("authorization"),
         }
 
     resolver = ProxyResolver(