Fix race condition in BinaryConnectionTests.

Author: rstam

tests/MongoDB.Driver.Core.Tests/Core/Connections/BinaryConnection_CommandEventTests.cs

@@ -42,7 +42,7 @@ public class BinaryConnection_CommandEventTests : IDisposable
         private DnsEndPoint _endPoint;
         private EventCapturer _capturedEvents;
         private MessageEncoderSettings _messageEncoderSettings = new MessageEncoderSettings();
-        private Stream _stream;
+        private BlockingMemoryStream _stream;
         private Mock<IStreamFactory> _mockStreamFactory;
         private BinaryConnection _subject;
         private IDisposable _operationIdDisposer;

tests/MongoDB.Driver.Core.Tests/Core/Helpers/BlockingMemoryStream.cs

@@ -50,6 +50,11 @@ public override long Length
             }
         }
 
+        public object Lock
+        {
+            get { return _lock; }
+        }
+
         public override long Position
         {
             get

tests/MongoDB.Driver.Core.Tests/Core/Helpers/MessageHelper.cs

@@ -272,16 +272,19 @@ public static List<BsonDocument> TranslateMessagesToBsonDocuments(byte[] bytes)
             return TranslateMessagesToBsonDocuments(TranslateBytesToRequests(bytes));
         }
 
-        public static void WriteResponsesToStream(Stream stream, IEnumerable<ResponseMessage> responses)
+        public static void WriteResponsesToStream(BlockingMemoryStream stream, IEnumerable<ResponseMessage> responses)
         {
-            var startPosition = stream.Position;
-            foreach (var response in responses)
+            lock (stream.Lock)
             {
-                var encoderFactory = new BinaryMessageEncoderFactory(stream, null);
-                var encoder = response.GetEncoder(encoderFactory);
-                encoder.WriteMessage(response);
+                var startPosition = stream.Position;
+                foreach (var response in responses)
+                {
+                    var encoderFactory = new BinaryMessageEncoderFactory(stream, null);
+                    var encoder = response.GetEncoder(encoderFactory);
+                    encoder.WriteMessage(response);
+                }
+                stream.Position = startPosition;
             }
-            stream.Position = startPosition;
         }
 
         private static List<RequestMessage> TranslateBytesToRequests(byte[] bytes)