From bd6c36f6a15fb5a05bb33a4aec69c6ef0cc5e0c8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D7=A0=D7=99=D7=A8?=
 <88795475+nrbnlulu@users.noreply.github.com>
Date: Sun, 7 Jul 2024 21:51:59 +0300
Subject: [PATCH] review fiexes.

---
 tests/execution/test_middleware.py | 17 +++++++----------
 1 file changed, 7 insertions(+), 10 deletions(-)

diff --git a/tests/execution/test_middleware.py b/tests/execution/test_middleware.py
index 2c31e1ec..8e260a8b 100644
--- a/tests/execution/test_middleware.py
+++ b/tests/execution/test_middleware.py
@@ -3,18 +3,10 @@
 
 import pytest
 from graphql.execution import Middleware, MiddlewareManager, execute, subscribe
-from graphql.execution import Middleware, MiddlewareManager, execute
 from graphql.language.parser import parse
 from graphql.type import GraphQLField, GraphQLObjectType, GraphQLSchema, GraphQLString
 
 
-def _create_subscription_schema(tp: GraphQLObjectType) -> GraphQLSchema:
-    noop_type = GraphQLObjectType(
-        "Noop", {"noop": GraphQLField(GraphQLString, resolve=lambda *_: "noop")}
-    )
-    return GraphQLSchema(query=noop_type, subscription=tp)
-
-
 def describe_middleware():
     def describe_with_manager():
         def default():
@@ -247,7 +239,6 @@ async def resolve(self, next_, *args, **kwargs):
 
         @pytest.mark.asyncio()
         async def subscription_simple():
-
             async def bar_resolve(_obj, _info):
                 yield "bar"
                 yield "oof"
@@ -268,8 +259,14 @@ async def reverse_middleware(next_, value, info, **kwargs):
                 awaitable_maybe = next_(value, info, **kwargs)
                 return awaitable_maybe[::-1]
 
+            noop_type = GraphQLObjectType(
+                "Noop",
+                {"noop": GraphQLField(GraphQLString, resolve=lambda *_args: "noop")},
+            )
+            schema = GraphQLSchema(query=noop_type, subscription=test_type)
+
             agen = subscribe(
-                _create_subscription_schema(test_type),
+                schema,
                 doc,
                 middleware=MiddlewareManager(reverse_middleware),
             )