Cancel scope on stream consumer completion

goodboy · May 2, 2021 · 921210a · 921210a
1 parent bc98387
commit 921210a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_streaming.py b/tests/test_streaming.py
@@ -338,6 +338,8 @@ async def consume(task_status=trio.TASK_STATUS_IGNORED):
                         print("all values streamed, BREAKING")
                         break
 
+                cs.cancel()
+
         # TODO: this is justification for a
         # ``ActorNursery.stream_from_actor()`` helper?
         await portal.cancel_actor()