modify where task is created / ran

Author: sleepyStick

test/asynchronous/test_mongos_load_balancing.py

@@ -55,7 +55,7 @@ def run(self):
 
     class SimpleOp:
         def __init__(self, client):
-            self.task = asyncio.create_task(self.run())
+            self.task: asyncio.Task
             self.client = client
             self.passed = False
 
@@ -64,7 +64,7 @@ async def run(self):
             self.passed = True  # No exception raised.
 
         def start(self):
-            pass
+            self.task = asyncio.create_task(self.run())
 
         async def join(self):
             await self.task

test/test_mongos_load_balancing.py

@@ -55,7 +55,7 @@ def run(self):
 
     class SimpleOp:
         def __init__(self, client):
-            self.task = asyncio.create_task(self.run())
+            self.task: asyncio.Task
             self.client = client
             self.passed = False
 
@@ -64,7 +64,7 @@ def run(self):
             self.passed = True  # No exception raised.
 
         def start(self):
-            pass
+            self.task = asyncio.create_task(self.run())
 
         def join(self):
             self.task