@@ -134,7 +134,7 @@ public void Run(
134
134
135
135
if ( initialData != null )
136
136
{
137
- AddInitialData ( DriverTestConfiguration . Client , initialData ) ;
137
+ AddInitialData ( DriverTestConfiguration . Client , initialData , _entityMap ) ;
138
138
}
139
139
140
140
foreach ( var operation in operations )
@@ -185,7 +185,7 @@ public void Dispose()
185
185
}
186
186
187
187
// private methods
188
- private void AddInitialData ( IMongoClient client , BsonArray initialData )
188
+ private void AddInitialData ( IMongoClient client , BsonArray initialData , UnifiedEntityMap entityMap )
189
189
{
190
190
var mongoCollectionSettings = new MongoCollectionSettings ( ) ;
191
191
#pragma warning disable CS0618 // Type or member is obsolete
@@ -195,6 +195,7 @@ private void AddInitialData(IMongoClient client, BsonArray initialData)
195
195
}
196
196
#pragma warning restore CS0618 // Type or member is obsolete
197
197
198
+ BsonDocument serverTime = null ;
198
199
foreach ( var dataItem in initialData )
199
200
{
200
201
var collectionName = dataItem [ "collectionName" ] . AsString ;
@@ -209,15 +210,20 @@ private void AddInitialData(IMongoClient client, BsonArray initialData)
209
210
_logger . LogDebug ( "Dropping {0}" , collectionName ) ;
210
211
211
212
database . DropCollection ( collectionName ) ;
213
+ var session = client . StartSession ( ) ;
212
214
if ( documents . Any ( ) )
213
215
{
214
- collection . InsertMany ( documents ) ;
216
+ collection . InsertMany ( session , documents ) ;
215
217
}
216
218
else
217
219
{
218
- database . WithWriteConcern ( WriteConcern . WMajority ) . CreateCollection ( collectionName ) ;
220
+ database . WithWriteConcern ( WriteConcern . WMajority ) . CreateCollection ( session , collectionName ) ;
219
221
}
222
+
223
+ serverTime = session . ClusterTime ;
220
224
}
225
+
226
+ entityMap . AdjustSessionsClusterTime ( serverTime ) ;
221
227
}
222
228
223
229
private void AssertEvents ( BsonArray eventItems , UnifiedEntityMap entityMap )
0 commit comments