Iterators are considered awaitable with the new method, so check only not aiter

Author: SmileyChris

graphql_ws/base_async.py

@@ -143,9 +143,6 @@ async def on_start(self, connection_context, op_id, params):
         execution_result = self.execute(params)
 
         connection_context.register_operation(op_id, execution_result)
-        if is_awaitable(execution_result):
-            execution_result = await execution_result
-
         if hasattr(execution_result, "__aiter__"):
             iterator = await execution_result.__aiter__()
             connection_context.register_operation(op_id, iterator)
@@ -160,6 +157,8 @@ async def on_start(self, connection_context, op_id, params):
                 await self.send_error(connection_context, op_id, e)
         else:
             try:
+                if is_awaitable(execution_result):
+                    execution_result = await execution_result
                 await self.send_execution_result(
                     connection_context, op_id, execution_result
                 )