CSHARP-2222: Legacy API should use implicit sessions

Author: rstam

src/MongoDB.Driver.Legacy/MongoDatabase.cs

@@ -559,7 +559,7 @@ public virtual MongoCollection<TDefaultDocument> GetCollection<TDefaultDocument>
         public virtual MongoCollection<TDefaultDocument> GetCollection<TDefaultDocument>(
             string collectionName, MongoCollectionSettings collectionSettings)
         {
-            return new MongoCollection<TDefaultDocument>(this, collectionName, collectionSettings);
+            return new MongoCollection<TDefaultDocument>(this, collectionName, collectionSettings, _operationExecutor);
         }
 
         /// <summary>
@@ -638,8 +638,12 @@ public virtual MongoCollection GetCollection(Type defaultDocumentType, string co
         {
             var collectionDefinition = typeof(MongoCollection<>);
             var collectionType = collectionDefinition.MakeGenericType(defaultDocumentType);
-            var constructorInfo = collectionType.GetTypeInfo().GetConstructor(new Type[] { typeof(MongoDatabase), typeof(string), typeof(MongoCollectionSettings) });
-            return (MongoCollection)constructorInfo.Invoke(new object[] { this, collectionName, collectionSettings });
+            var constructorParameterTypes = new Type[] { typeof(MongoDatabase), typeof(string), typeof(MongoCollectionSettings), typeof(IOperationExecutor) };
+            var constructorInfo = collectionType.GetTypeInfo()
+                .GetConstructors(BindingFlags.NonPublic | BindingFlags.Instance)
+                .Where(c => c.GetParameters().Select(p => p.ParameterType).SequenceEqual(constructorParameterTypes))
+                .Single();
+            return (MongoCollection)constructorInfo.Invoke(new object[] { this, collectionName, collectionSettings, _operationExecutor });
         }
 
         /// <summary>

tests/MongoDB.Driver.Legacy.Tests/DisposableMongoClientExtensions.cs

@@ -0,0 +1,31 @@
+/* Copyright 2018-present MongoDB Inc.
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+
+using System;
+using MongoDB.Driver.TestHelpers;
+
+namespace MongoDB.Driver.Legacy.Tests
+{
+    public static class DisposableMongoClientExtensions
+    {
+        public static MongoServer GetServer(this DisposableMongoClient disposableClient)
+        {
+            var client = (MongoClient)disposableClient.Wrapped;
+#pragma warning disable 618
+            return client.GetServer();
+#pragma warning restore
+        }
+    }
+}

tests/MongoDB.Driver.Legacy.Tests/MongoCollectionTests.cs

@@ -31,6 +31,10 @@
 using MongoDB.Driver.Core.TestHelpers.XunitExtensions;
 using MongoDB.Driver.Core.Clusters;
 using MongoDB.Driver.Core.Misc;
+using MongoDB.Driver.TestHelpers;
+using MongoDB.Driver.Core.Events;
+using MongoDB.Driver.Legacy.Tests;
+using MongoDB.Bson.TestHelpers;
 
 namespace MongoDB.Driver.Tests
 {
@@ -423,6 +427,26 @@ public void TestCountZero()
             Assert.Equal(0, count);
         }
 
+        [SkippableFact]
+        public void TestCountUsesImplicitSession()
+        {
+            RequireServer.Check();
+
+            var events = new EventCapturer().Capture<CommandStartedEvent>(x => x.CommandName == "count");
+            using (var client = DriverTestConfiguration.CreateDisposableClient(events))
+            {
+                var server = client.GetServer();
+                var database = server.GetDatabase(DriverTestConfiguration.DatabaseNamespace.DatabaseName);
+                var collection = database.GetCollection<BsonDocument>(DriverTestConfiguration.CollectionNamespace.CollectionName);
+                collection.Count();
+
+                var commandStartedEvent = events.Next().Should().BeOfType<CommandStartedEvent>().Subject;
+                var command = commandStartedEvent.Command;
+                var areSessionsSupported = AreSessionsSupported(client);
+                command.Contains("lsid").Should().Be(areSessionsSupported);
+            }
+        }
+
         [Fact]
         public void TestCountOne()
         {
@@ -1227,7 +1251,7 @@ public void TestFindAndRemoveWithWriteConcernError()
             var collection = _database.GetCollection(_collection.Name, collectionSettings);
             var args = new FindAndRemoveArgs
             {
-                Query = Query.EQ("x", 1)              
+                Query = Query.EQ("x", 1)
             };
 
             BsonDocument modifiedDocument;
@@ -3504,6 +3528,12 @@ public void TestWithWriteConcern()
         }
 
         // private methods
+        private bool AreSessionsSupported(IMongoClient client)
+        {
+            var description = client.Cluster.Description;
+            return description.LogicalSessionTimeout.HasValue;
+        }
+
         private void CheckExpectedResult(ExpectedWriteConcernResult expectedResult, WriteConcernResult result)
         {
             Assert.Equal(expectedResult.DocumentsAffected ?? 0, result.DocumentsAffected);

tests/MongoDB.Driver.Legacy.Tests/MongoDB.Driver.Legacy.Tests.csproj

@@ -85,6 +85,7 @@
     <Compile Include="CommandResults\DatabaseStatsResultTests.cs" />
     <Compile Include="CommandResults\ValidateCollectionResultTests.cs" />
     <Compile Include="DefaultLegacyOperationExecutorTests.cs" />
+    <Compile Include="DisposableMongoClientExtensions.cs" />
     <Compile Include="FailPoint.cs" />
     <Compile Include="Jira\CSharp100Tests.cs" />
     <Compile Include="Jira\CSharp101Tests.cs" />

tests/MongoDB.Driver.Legacy.Tests/MongoDatabaseTests.cs

@@ -17,6 +17,7 @@
 using System.Linq;
 using FluentAssertions;
 using MongoDB.Bson;
+using MongoDB.Bson.TestHelpers;
 using MongoDB.Bson.TestHelpers.XunitExtensions;
 using MongoDB.Driver;
 using MongoDB.Driver.Builders;
@@ -322,6 +323,7 @@ public void TestGetCollection()
             Assert.Equal(_database.Name + "." + collectionName, collection.FullName);
             Assert.Equal(collectionName, collection.Name);
             Assert.Equal(_database.Settings.WriteConcern, collection.Settings.WriteConcern);
+            Assert.Equal(_database._operationExecutor(), collection.OperationExecutor);
         }
 
         [Fact]
@@ -333,6 +335,7 @@ public void TestGetCollectionGeneric()
             Assert.Equal(_database.Name + "." + collectionName, collection.FullName);
             Assert.Equal(collectionName, collection.Name);
             Assert.Equal(_database.Settings.WriteConcern, collection.Settings.WriteConcern);
+            Assert.Equal(_database._operationExecutor(), collection.OperationExecutor);
         }
 
         [Fact]
@@ -589,4 +592,9 @@ private void EnsureCollectionExists(string collectionName)
             _database.CreateCollection(collectionName);
         }
     }
+
+    internal static class MongoDatabaseReflector
+    {
+        public static IOperationExecutor _operationExecutor(this MongoDatabase obj) => (IOperationExecutor)Reflector.GetFieldValue(obj, nameof(_operationExecutor));
+    }
 }

tests/MongoDB.Driver.Legacy.Tests/MongoServerTests.cs

@@ -17,6 +17,7 @@
 using System.Linq;
 using FluentAssertions;
 using MongoDB.Bson;
+using MongoDB.Bson.TestHelpers;
 using MongoDB.Bson.TestHelpers.XunitExtensions;
 using MongoDB.Driver;
 using MongoDB.Driver.Core.Clusters;
@@ -178,6 +179,7 @@ public void TestGetDatabase()
             var database = __server.GetDatabase("test", settings);
             Assert.Equal("test", database.Name);
             Assert.Equal(ReadPreference.Primary, database.Settings.ReadPreference);
+            Assert.Equal(database._operationExecutor(), __server._operationExecutor());
         }
 
         [Fact]
@@ -346,4 +348,9 @@ public void TestWithWriteConcern()
             result.WithWriteConcern(originalWriteConcern).Settings.Should().Be(subject.Settings);
         }
     }
+
+    internal static class MongoServerReflector
+    {
+        public static IOperationExecutor _operationExecutor(this MongoServer obj) => (IOperationExecutor)Reflector.GetFieldValue(obj, nameof(_operationExecutor));
+    }
 }

tests/MongoDB.Driver.TestHelpers/DisposableMongoClient.cs

@@ -34,6 +34,8 @@ public DisposableMongoClient(IMongoClient wrapped)
 
         public MongoClientSettings Settings => wrapped.Settings;
 
+        public IMongoClient Wrapped => wrapped;
+
         public void DropDatabase(string name, CancellationToken cancellationToken = default(CancellationToken))
         {
             wrapped.DropDatabase(name, cancellationToken);

tests/MongoDB.Driver.TestHelpers/DriverTestConfiguration.cs

@@ -14,6 +14,9 @@
 */
 
 using System;
+using MongoDB.Driver.Core;
+using MongoDB.Driver.Core.Configuration;
+using MongoDB.Driver.TestHelpers;
 
 namespace MongoDB.Driver.Tests
 {
@@ -67,6 +70,30 @@ public static DatabaseNamespace DatabaseNamespace
         }
 
         // public static methods
+        public static DisposableMongoClient CreateDisposableClient()
+        {
+            return CreateDisposableClient((MongoClientSettings s) => { });
+        }
+
+        public static DisposableMongoClient CreateDisposableClient(Action<ClusterBuilder> clusterConfigurator)
+        {
+            return CreateDisposableClient((MongoClientSettings s) => s.ClusterConfigurator = clusterConfigurator);
+        }
+
+        public static DisposableMongoClient CreateDisposableClient(Action<MongoClientSettings> clientSettingsConfigurator)
+        {
+            var connectionString = CoreTestConfiguration.ConnectionString.ToString();
+            var clientSettings = MongoClientSettings.FromUrl(new MongoUrl(connectionString));
+            clientSettingsConfigurator(clientSettings);
+            var client = new MongoClient(clientSettings);
+            return new DisposableMongoClient(client);
+        }
+
+        public static DisposableMongoClient CreateDisposableClient(EventCapturer capturer)
+        {
+            return CreateDisposableClient((ClusterBuilder c) => c.Subscribe(capturer));
+        }
+
         public static MongoClientSettings GetClientSettings()
         {
             var connectionString = CoreTestConfiguration.ConnectionString.ToString();

tests/MongoDB.Driver.Tests/CausalConsistencyTests.cs

@@ -209,11 +209,7 @@ public void ReadConcern_should_include_level_when_not_using_the_server_default()
 
         private DisposableMongoClient GetClient(EventCapturer capturer)
         {
-            var connectionString = CoreTestConfiguration.ConnectionString.ToString();
-            var clientSettings = MongoClientSettings.FromUrl(new MongoUrl(connectionString));
-            clientSettings.ClusterConfigurator = cb => cb.Subscribe(capturer);
-
-            return new DisposableMongoClient(new MongoClient(clientSettings));
+            return DriverTestConfiguration.CreateDisposableClient(capturer);
         }
     }
 }

tests/MongoDB.Driver.Tests/RetryableWritesTests.cs

@@ -183,12 +183,11 @@ private DisposableMongoClient GetClient()
 
         private DisposableMongoClient GetClient(Action<ClusterBuilder> clusterConfigurator)
         {
-            var connectionString = CoreTestConfiguration.ConnectionString.ToString();
-            var clientSettings = MongoClientSettings.FromUrl(new MongoUrl(connectionString));
-            clientSettings.RetryWrites = true;
-            clientSettings.ClusterConfigurator = clusterConfigurator;
-
-            return new DisposableMongoClient(new MongoClient(clientSettings));
+            return DriverTestConfiguration.CreateDisposableClient((MongoClientSettings clientSettings) =>
+            {
+                clientSettings.ClusterConfigurator = clusterConfigurator;
+                clientSettings.RetryWrites = true;
+            });
         }
 
         private DisposableMongoClient GetClient(EventCapturer capturer)