address review

Author: sleepyStick

gridfs/asynchronous/grid_file.py

@@ -231,7 +231,7 @@ async def get_version(
         try:
             doc = await anext(cursor)
             return AsyncGridOut(self._collection, file_document=doc, session=session)
-        except (StopIteration, StopAsyncIteration):
+        except StopAsyncIteration:
             raise NoFile("no version %d for filename %r" % (version, filename)) from None
 
     async def get_last_version(

test/asynchronous/test_gridfs.py

@@ -75,20 +75,20 @@ def run(self):
                 assert data == b"hello"
 else:
 
-    class JustWrite(asyncio.Task):
+    class JustWrite:
         def __init__(self, fs, n):
             async def run():
                 for _ in range(self.n):
                     file = self.fs.new_file(filename="test")
                     await file.write(b"hello")
                     await file.close()
 
-            asyncio.Task.__init__(self, run())
+            self.task = asyncio.create_task(run())
             self.fs = fs
             self.n = n
             self.daemon = True
 
-    class JustRead(asyncio.Task):
+    class JustRead:
         def __init__(self, fs, n, results):
             async def run():
                 for _ in range(self.n):
@@ -97,7 +97,7 @@ async def run():
                     self.results.append(data)
                     assert data == b"hello"
 
-            asyncio.Task.__init__(self, run())
+            self.task = asyncio.create_task(run())
             self.fs = fs
             self.n = n
             self.results = results

test/test_gridfs.py

@@ -75,20 +75,20 @@ def run(self):
                 assert data == b"hello"
 else:
 
-    class JustWrite(asyncio.Task):
+    class JustWrite:
         def __init__(self, fs, n):
             def run():
                 for _ in range(self.n):
                     file = self.fs.new_file(filename="test")
                     file.write(b"hello")
                     file.close()
 
-            asyncio.Task.__init__(self, run())
+            self.task = asyncio.create_task(run())
             self.fs = fs
             self.n = n
             self.daemon = True
 
-    class JustRead(asyncio.Task):
+    class JustRead:
         def __init__(self, fs, n, results):
             def run():
                 for _ in range(self.n):
@@ -97,7 +97,7 @@ def run():
                     self.results.append(data)
                     assert data == b"hello"
 
-            asyncio.Task.__init__(self, run())
+            self.task = asyncio.create_task(run())
             self.fs = fs
             self.n = n
             self.results = results