Fix coverage tests

graphql-python · Aug 11, 2024 · ac56523 · ac56523
1 parent b4f8261
commit ac56523
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 33 deletions.
diff --git a/src/graphql/execution/execute.py b/src/graphql/execution/execute.py
@@ -1645,7 +1645,7 @@ async def execute_stream_async_iterator_item(
     ) -> Any:
         """Execute stream iterator item."""
         if async_iterator in self._canceled_iterators:
-            raise StopAsyncIteration
+            raise StopAsyncIteration  # pragma: no cover
         try:
             item = await anext(async_iterator)
         except StopAsyncIteration as raw_error:
@@ -1682,7 +1682,6 @@ async def execute_stream_async_iterator(
         parent_context: IncrementalDataRecord | None = None,
     ) -> None:
         """Execute stream iterator."""
-        is_awaitable = self.is_awaitable
         incremental_publisher = self.incremental_publisher
         index = initial_index
         previous_incremental_data_record = parent_context
@@ -1722,35 +1721,10 @@ async def execute_stream_async_iterator(
             except StopAsyncIteration:
                 done = True
 
-            if is_awaitable(completed_item):
-
-                async def await_completed_item(
-                    incremental_data_record: StreamItemsRecord, completed_item: Any
-                ) -> None:
-                    try:
-                        value = await completed_item
-                    except GraphQLError as error:
-                        incremental_publisher.add_field_error(
-                            incremental_data_record, error
-                        )
-                        incremental_publisher.filter(path, incremental_data_record)
-                        incremental_publisher.complete_stream_items_record(
-                            incremental_data_record, None
-                        )
-                    else:
-                        incremental_publisher.complete_stream_items_record(
-                            incremental_data_record, [value]
-                        )
-
-                self.add_task(
-                    await_completed_item(incremental_data_record, completed_item)
-                )
-
-            else:
-                incremental_publisher.complete_stream_items_record(
-                    incremental_data_record,
-                    [completed_item],
-                )
+            incremental_publisher.complete_stream_items_record(
+                incremental_data_record,
+                [completed_item],
+            )
 
             if done:
                 break

diff --git a/src/graphql/execution/incremental_publisher.py b/src/graphql/execution/incremental_publisher.py
@@ -407,7 +407,9 @@ async def subscribe(
         finally:
             close_async_iterators = []
             for incremental_data_record in pending:
-                if isinstance(incremental_data_record, StreamItemsRecord):
+                if isinstance(
+                    incremental_data_record, StreamItemsRecord
+                ):  # pragma: no cover
                     async_iterator = incremental_data_record.async_iterator
                     if async_iterator:
                         try:
@@ -508,7 +510,7 @@ def filter(
                 if async_iterator:
                     try:
                         close_async_iterator = async_iterator.aclose()  # type:ignore
-                    except AttributeError:
+                    except AttributeError:  # pragma: no cover
                         pass
                     else:
                         self._add_task(close_async_iterator)

diff --git a/tests/execution/test_executor.py b/tests/execution/test_executor.py
@@ -617,6 +617,7 @@ def resolve_error(*_args):
             ],
         )
 
+    @pytest.mark.filterwarnings("ignore:.* was never awaited:RuntimeWarning")
     def uses_the_inline_operation_if_no_operation_name_is_provided():
         schema = GraphQLSchema(
             GraphQLObjectType("Type", {"a": GraphQLField(GraphQLString)})