Skip to content

Commit 5b8feba

Browse files
committed
fix cov
1 parent 00731aa commit 5b8feba

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/execution/test_middleware.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -247,9 +247,10 @@ async def resolve(self, next_, *args, **kwargs):
247247

248248
@pytest.mark.asyncio()
249249
async def subscription_simple():
250+
250251
async def bar_resolve(_obj, _info):
251252
yield "bar"
252-
253+
yield "oof"
253254
test_type = GraphQLObjectType(
254255
"Subscription",
255256
{
@@ -264,8 +265,6 @@ async def bar_resolve(_obj, _info):
264265

265266
async def reverse_middleware(next_, value, info, **kwargs):
266267
awaitable_maybe = next_(value, info, **kwargs)
267-
if inspect.isawaitable(awaitable_maybe):
268-
return (await awaitable_maybe)[::-1]
269268
return awaitable_maybe[::-1]
270269

271270
agen = subscribe(
@@ -276,6 +275,8 @@ async def reverse_middleware(next_, value, info, **kwargs):
276275
assert inspect.isasyncgen(agen)
277276
data = (await agen.__anext__()).data
278277
assert data == {"bar": "rab"}
278+
data = (await agen.__anext__()).data
279+
assert data == {"bar": "foo"}
279280

280281
def describe_without_manager():
281282
def no_middleware():

0 commit comments

Comments
 (0)