Added domain to MongoQueryProvider

Author: papafe

src/MongoDB.Driver/Linq/Linq3Implementation/MongoQueryProvider.cs

@@ -64,6 +64,7 @@ internal sealed class MongoQueryProvider<TDocument> : MongoQueryProvider
         private readonly IMongoDatabase _database;
         private ExecutableQuery<TDocument> _executedQuery;
         private readonly IBsonSerializer _pipelineInputSerializer;
+        private readonly IBsonSerializationDomain _serializationDomain;
 
         // constructors
         public MongoQueryProvider(
@@ -74,6 +75,7 @@ public MongoQueryProvider(
         {
             _collection = Ensure.IsNotNull(collection, nameof(collection));
             _pipelineInputSerializer = collection.DocumentSerializer;
+            _serializationDomain = collection.Settings.SerializationDomain;
         }
 
         public MongoQueryProvider(
@@ -84,15 +86,18 @@ public MongoQueryProvider(
         {
             _database = Ensure.IsNotNull(database, nameof(database));
             _pipelineInputSerializer = NoPipelineInputSerializer.Instance;
+            _serializationDomain = _database.Settings.SerializationDomain;
         }
 
         internal MongoQueryProvider(
             IBsonSerializer pipelineInputSerializer,
             IClientSessionHandle session,
-            AggregateOptions options)
+            AggregateOptions options,
+            IBsonSerializationDomain serializationDomain)
             : base(session, options)
         {
             _pipelineInputSerializer = Ensure.IsNotNull(pipelineInputSerializer, nameof(pipelineInputSerializer));
+            _serializationDomain = Ensure.IsNotNull(serializationDomain, nameof(serializationDomain));
         }
 
         // public properties
@@ -101,6 +106,7 @@ internal MongoQueryProvider(
         public IMongoDatabase Database => _database;
         public override BsonDocument[] LoggedStages => _executedQuery?.LoggedStages;
         public override IBsonSerializer PipelineInputSerializer => _pipelineInputSerializer;
+        public IBsonSerializationDomain SerializationDomain => _serializationDomain;
 
         // public methods
         public override IQueryable CreateQuery(Expression expression)

src/MongoDB.Driver/Linq/Linq3Implementation/Translators/ExpressionToExecutableQueryTranslators/ExpressionToExecutableQueryTranslator.cs

@@ -31,7 +31,7 @@ public static ExecutableQuery<TDocument, IAsyncCursor<TOutput>> Translate<TDocum
         {
             expression = PartialEvaluator.EvaluatePartially(expression);
 
-            var context = TranslationContext.Create(translationOptions, provider.Collection.Settings.SerializationDomain);
+            var context = TranslationContext.Create(translationOptions, provider.SerializationDomain);
             var pipeline = ExpressionToPipelineTranslator.Translate(context, expression);
 
             return ExecutableQuery.Create(

src/MongoDB.Driver/Linq/Linq3Implementation/Translators/ExpressionToPipelineTranslators/LookupMethodToPipelineTranslator.cs

@@ -286,7 +286,7 @@ private static TranslatedPipeline TranslateDocumentsPipelineGeneric<TLocal, TDoc
             var localSymbol = context.CreateSymbol(localParameter, localAst, localSerializer);
             context = context.WithSymbol(localSymbol);
 
-            var provider = new MongoQueryProvider<TDocument>(documentSerializer, session: null, options: null);
+            var provider = new MongoQueryProvider<TDocument>(documentSerializer, session: null, options: null, serializationDomain: context.SerializationDomain);
             var queryable = new MongoQuery<TDocument, TDocument>(provider);
 
             body = ExpressionReplacer.Replace(body, queryableParameter, Expression.Constant(queryable));

tests/MongoDB.Driver.Tests/Linq/Linq3Implementation/Jira/CSharp5286Tests.cs

@@ -161,10 +161,9 @@ public Type GetActualType(IBsonReader bsonReader, Type nominalType, IBsonSeriali
             }
 
             public BsonValue GetDiscriminator(Type nominalType, Type actualType) => actualType.Name;
-            public BsonValue GetDiscriminator(Type nominalType, Type actualType, IBsonSerializationDomain domain)
-            {
-                throw new NotImplementedException();
-            }
+
+            public BsonValue GetDiscriminator(Type nominalType, Type actualType, IBsonSerializationDomain domain) =>
+                actualType.Name;
 
             public BsonValue[] GetDiscriminatorsForTypeAndSubTypes(Type type)
             {