Fix failed tests

Author: Oleksandr Poliakov

src/MongoDB.Driver/Core/Misc/StreamExtensionMethods.cs

@@ -77,19 +77,28 @@ public static int Read(this Stream stream, byte[] buffer, int offset, int count,
 
         public static async Task<int> ReadAsync(this Stream stream, byte[] buffer, int offset, int count, TimeSpan timeout, CancellationToken cancellationToken)
         {
-            var readTask = stream.ReadAsync(buffer, offset, count);
+            Task<int> readTask = null;
             try
             {
+                readTask = stream.ReadAsync(buffer, offset, count);
                 return await readTask.WaitAsync(timeout, cancellationToken).ConfigureAwait(false);
             }
+            catch (ObjectDisposedException)
+            {
+                // It's possible to get ObjectDisposedException when the connection pool was closed with interruptInUseConnections set to true.
+                throw new IOException();
+            }
             catch (Exception ex) when (ex is OperationCanceledException or TimeoutException)
             {
                 // await Task.WaitAsync() throws OperationCanceledException in case of cancellation and TimeoutException in case of timeout
                 try
                 {
                     stream.Dispose();
-                    // Should await on the task to avoid UnobservedTaskException
-                    await readTask.ConfigureAwait(false);
+                    if (readTask != null)
+                    {
+                        // Should await on the task to avoid UnobservedTaskException
+                        await readTask.ConfigureAwait(false);
+                    }
                 }
                 catch
                 {
@@ -204,6 +213,7 @@ public static void Write(this Stream stream, byte[] buffer, int offset, int coun
             }
             catch (ObjectDisposedException)
             {
+                // It's possible to get ObjectDisposedException when the connection pool was closed with interruptInUseConnections set to true.
                 throw new IOException();
             }
 
@@ -222,19 +232,28 @@ public static void Write(this Stream stream, byte[] buffer, int offset, int coun
 
         public static async Task WriteAsync(this Stream stream, byte[] buffer, int offset, int count, TimeSpan timeout, CancellationToken cancellationToken)
         {
-            var writeTask = stream.WriteAsync(buffer, offset, count);
+            Task writeTask = null;
             try
             {
+                writeTask = stream.WriteAsync(buffer, offset, count);
                 await writeTask.WaitAsync(timeout, cancellationToken).ConfigureAwait(false);
             }
+            catch (ObjectDisposedException)
+            {
+                // It's possible to get ObjectDisposedException when the connection pool was closed with interruptInUseConnections set to true.
+                throw new IOException();
+            }
             catch (Exception ex) when (ex is OperationCanceledException or TimeoutException)
             {
                 // await Task.WaitAsync() throws OperationCanceledException in case of cancellation and TimeoutException in case of timeout
                 try
                 {
                     stream.Dispose();
                     // Should await on the task to avoid UnobservedTaskException
-                    await writeTask.ConfigureAwait(false);
+                    if (writeTask != null)
+                    {
+                        await writeTask.ConfigureAwait(false);
+                    }
                 }
                 catch
                 {