Fixed tests

tests/MongoDB.Driver.Tests/AtClusterTimeTests.cs

@@ -18,42 +18,34 @@
 using MongoDB.Bson.Serialization.Attributes;
 using MongoDB.Driver.Core.Clusters;
 using MongoDB.Driver.Core.Misc;
-using MongoDB.Driver.Core.TestHelpers.XunitExtensions;
+using MongoDB.Driver.TestHelpers;
 using Xunit;
 
 namespace MongoDB.Driver.Tests;
 
-[Trait("Category", "Integration")]
-public class AtClusterTimeTests
+public class AtClusterTimeTests : IntegrationTest<AtClusterTimeTests.ClassFixture>
 {
- [Fact]
- public void AtClusterTime_should_work()
+ public AtClusterTimeTests(ClassFixture fixture)
+  : base(fixture, server => server.Supports(Feature.SnapshotReads).ClusterType(ClusterType.ReplicaSet))
  {
- RequireServer.Check().ClusterType(ClusterType.ReplicaSet).Supports(Feature.SnapshotReads);
- const string collectionName = "atClusterTimeTests";
- const string databaseName = "testDb";
-
- using var client = DriverTestConfiguration.Client;
- var database = client.GetDatabase(databaseName);
- database.DropCollection(collectionName);
- var collection = database.GetCollection<TestObject>(collectionName);
+ }
 
- var obj1 = new TestObject { Name = "obj1" };
- collection.InsertOne(obj1);
+ [Fact]
+ public void MainTest()
+ {
+ var client = Fixture.Client;
+ var collection = Fixture.Collection;
 
  BsonTimestamp clusterTime1;
 
- var filterDefinition = Builders<TestObject>.Filter.Empty;
- var sortDefinition = Builders<TestObject>.Sort.Ascending(o => o.Name);
-
  var sessionOptions1 = new ClientSessionOptions
  {
  Snapshot = true
  };
 
  using (var session1 = client.StartSession(sessionOptions1))
  {
- var results = collection.Find(session1, filterDefinition).Sort(sortDefinition).ToList();
+ var results = GetTestObjects(collection, session1);
  AssertOneObj(results);
 
  clusterTime1 = session1.GetSnapshotTime();
@@ -72,7 +64,7 @@ public void AtClusterTime_should_work()
  //Snapshot read session at clusterTime1 should not see obj2
  using (var session2 = client.StartSession(sessionOptions2))
  {
- var results = collection.Find(session2, filterDefinition).Sort(sortDefinition).ToList();
+ var results = GetTestObjects(collection, session2);
  AssertOneObj(results);
 
  var clusterTime2 = session2.GetSnapshotTime();
@@ -87,28 +79,127 @@ public void AtClusterTime_should_work()
  //Snapshot read session without cluster time should see obj2
  using (var session3 = client.StartSession(sessionOptions3))
  {
- var results = collection.Find(session3, filterDefinition).Sort(sortDefinition).ToList();
+ var results = GetTestObjects(collection, session3);
  AssertTwoObjs(results);
 
- var clusterTime3 = session3.WrappedCoreSession.SnapshotTime;
+ var clusterTime3 = session3.GetSnapshotTime();
  Assert.NotEqual(clusterTime3, clusterTime1);
  }
+ }
 
- void AssertOneObj(List<TestObject> objs)
+ [Fact]
+ public void IncreasedTimestamp()
+ {
+ var client = Fixture.Client;
+ var collection = Fixture.Collection;
+
+ BsonTimestamp clusterTime1;
+
+ var sessionOptions1 = new ClientSessionOptions
+ {
+ Snapshot = true
+ };
+
+ using (var session1 = client.StartSession(sessionOptions1))
  {
- Assert.Equal(1, objs.Count);
- Assert.Equal("obj1", objs[0].Name);
+ var results = GetTestObjects(collection, session1);
+ AssertOneObj(results);
+
+ clusterTime1 = session1.GetSnapshotTime();
+ Assert.NotEqual(null, clusterTime1);
  }
 
- void AssertTwoObjs(List<TestObject> objs)
+ var obj2 = new TestObject { Name = "obj2" };
+ collection.InsertOne(obj2);
+
+ var modifiedClusterTime = new BsonTimestamp(clusterTime1.Value + 1);
+ var sessionOptions2 = new ClientSessionOptions
  {
- Assert.Equal(2, objs.Count);
- Assert.Equal("obj1", objs[0].Name);
- Assert.Equal("obj2", objs[1].Name);
+ Snapshot = true,
+ SnapshotTime = modifiedClusterTime
+ };
+
+ //Snapshot read session at clusterTime1+1 should see obj2
+ using (var session2 = client.StartSession(sessionOptions2))
+ {
+ var results = GetTestObjects(collection, session2);
+ AssertTwoObjs(results);
+
+ var clusterTime2 = session2.GetSnapshotTime();
+ Assert.Equal(modifiedClusterTime, clusterTime2);
  }
  }
 
- private class TestObject
+ [Fact]
+ public void DecreasedTimestamp()
+ {
+ var client = Fixture.Client;
+ var collection = Fixture.Collection;
+
+ BsonTimestamp clusterTime1;
+
+ var sessionOptions1 = new ClientSessionOptions
+ {
+ Snapshot = true
+ };
+
+ using (var session1 = client.StartSession(sessionOptions1))
+ {
+ var results = GetTestObjects(collection, session1);
+ AssertOneObj(results);
+
+ clusterTime1 = session1.GetSnapshotTime();
+ Assert.NotEqual(null, clusterTime1);
+ }
+
+ var obj2 = new TestObject { Name = "obj2" };
+ collection.InsertOne(obj2);
+
+ var modifiedClusterTime = new BsonTimestamp(clusterTime1.Value - 1);
+ var sessionOptions2 = new ClientSessionOptions
+ {
+ Snapshot = true,
+ SnapshotTime = modifiedClusterTime
+ };
+
+ //Snapshot read session at clusterTime1-1 should not see obj2
+ using (var session2 = client.StartSession(sessionOptions2))
+ {
+ var results = GetTestObjects(collection, session2);
+ Assert.Equal(0, results.Count);
+
+ var clusterTime2 = session2.GetSnapshotTime();
+ Assert.Equal(modifiedClusterTime, clusterTime2);
+ }
+ }
+
+ List<TestObject> GetTestObjects(IMongoCollection<TestObject> collection, IClientSessionHandle session)
+ {
+ var filterDefinition = Builders<TestObject>.Filter.Empty;
+ var sortDefinition = Builders<TestObject>.Sort.Ascending(o => o.Name);
+ return collection.Find(session, filterDefinition).Sort(sortDefinition).ToList();
+ }
+
+ void AssertOneObj(List<TestObject> objs)
+ {
+ Assert.Equal(1, objs.Count);
+ Assert.Equal("obj1", objs[0].Name);
+ }
+
+ void AssertTwoObjs(List<TestObject> objs)
+ {
+ Assert.Equal(2, objs.Count);
+ Assert.Equal("obj1", objs[0].Name);
+ Assert.Equal("obj2", objs[1].Name);
+ }
+
+ public class ClassFixture : MongoCollectionFixture<TestObject>
+ {
+ public override bool InitializeDataBeforeEachTestCase => true;
+ protected override IEnumerable<TestObject> InitialData => [new() { Name = "obj1" }] ;
+ }
+
+ public class TestObject
  {
  [BsonId]
  public ObjectId Id { get; set; }