Add ISessionFactory for MongoSessionProvider

src/Session/ISessionFactory.cs

@@ -0,0 +1,11 @@
+using System.Threading;
+using System.Threading.Tasks;
+using MongoDB.Driver;
+
+namespace MongoDB.Extensions.Session;
+
+public interface ISessionFactory
+{
+    ValueTask<IClientSessionHandle> CreateSessionAsync(
+        CancellationToken cancellationToken);
+}

src/Session/Internal/MongoClientSessionFactory.cs

@@ -0,0 +1,22 @@
+using System.Threading;
+using System.Threading.Tasks;
+using MongoDB.Driver;
+
+namespace MongoDB.Extensions.Session;
+
+internal class MongoClientSessionFactory : ISessionFactory
+{
+    private readonly IMongoClient _client;
+
+    public MongoClientSessionFactory(IMongoClient client)
+    {
+        _client = client;
+    }
+
+    public async ValueTask<IClientSessionHandle> CreateSessionAsync(
+        CancellationToken cancellationToken)
+    {
+        return await _client
+            .StartSessionAsync(cancellationToken: cancellationToken);
+    }
+}

src/Session/Internal/MongoSessionProvider.cs

@@ -6,14 +6,22 @@
 
 namespace MongoDB.Extensions.Session;
 
-public class MongoSessionProvider<TContext, TScope> : ISessionProvider<TScope>
+public class MongoSessionProvider<TContext, TScope> : MongoSessionProvider<TScope>
     where TContext : IMongoDbContext
 {
-    private readonly IMongoClient _mongoClient;
+    public MongoSessionProvider(TContext context) 
+        : base(new MongoClientSessionFactory(context.Client))
+    {
+    }
+}
 
-    public MongoSessionProvider(TContext context)
+public class MongoSessionProvider<TScope> : ISessionProvider<TScope>
+{
+    private readonly ISessionFactory _sessionFactory;
+
+    protected MongoSessionProvider(ISessionFactory sessionFactory)
     {
-        _mongoClient = context.Client;
+        _sessionFactory = sessionFactory;
     }
 
     protected virtual TransactionOptions TransactionOptions { get; } = new(
@@ -25,8 +33,8 @@ public MongoSessionProvider(TContext context)
     public async Task<ITransactionSession> BeginTransactionAsync(
         CancellationToken cancellationToken)
     {
-        IClientSessionHandle clientSession = await _mongoClient
-            .StartSessionAsync(cancellationToken: cancellationToken);
+        IClientSessionHandle clientSession = await _sessionFactory
+            .CreateSessionAsync(cancellationToken);
 
         clientSession.StartTransaction(TransactionOptions);
 
@@ -36,9 +44,9 @@ public async Task<ITransactionSession> BeginTransactionAsync(
     public async Task<ISession> StartSessionAsync(
         CancellationToken cancellationToken)
     {
-        IClientSessionHandle clientSession = await _mongoClient
-            .StartSessionAsync(cancellationToken: cancellationToken);
+        IClientSessionHandle clientSession = await _sessionFactory
+            .CreateSessionAsync(cancellationToken);
 
         return new MongoSession(clientSession, TransactionOptions);
     }
-}
+}