Waiting for terminating_task to be finished

blast-hardcheese · blast-hardcheese · commit b5105ff939b3 · 2025-04-15T15:34:15.000-07:00
diff --git a/src/replit_river/v2/session.py b/src/replit_river/v2/session.py
@@ -321,6 +321,8 @@ def unbind_connecting_task() -> None:
             )
 
         await self._connecting_task
+        if self._terminating_task:
+            await self._terminating_task
 
     def is_closed(self) -> bool:
         """

Original file line number	Diff line number	Diff line change
`@@ -321,6 +321,8 @@ def unbind_connecting_task() -> None:`
`321`	`321`	`)`
`322`	`322`
`323`	`323`	`await self._connecting_task`
	`324`	`+ if self._terminating_task:`
	`325`	`+ await self._terminating_task`
`324`	`326`
`325`	`327`	`def is_closed(self) -> bool:`
`326`	`328`	`"""`