CSHARP-1437: Added write concern tests for legacy CRUD and FindAndModify methods.

Author: rstam

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

@@ -25,6 +25,7 @@
 using MongoDB.Driver.Builders;
 using MongoDB.Driver.Core;
 using MongoDB.Driver.GeoJsonObjectModel;
+using FluentAssertions;
 using NUnit.Framework;
 
 namespace MongoDB.Driver.Tests
@@ -952,6 +953,58 @@ public void TestFindAndModifyUpsert()
             Assert.AreEqual(1, result.ModifiedDocument["count"].AsInt32);
         }
 
+        [Test]
+        [RequiresServer(ClusterTypes = ClusterTypes.ReplicaSet)]
+        public void TestFindAndModifyReplaceWithWriteConcernError()
+        {
+            _collection.RemoveAll();
+            _collection.Insert(new BsonDocument { { "_id", 1 }, { "x", 1 } });
+            var collectionSettings = new MongoCollectionSettings
+            {
+                WriteConcern = new WriteConcern(9)
+            };
+            var collection = _database.GetCollection(_collection.Name, collectionSettings);
+            var args = new FindAndModifyArgs
+            {
+                Query = Query.EQ("_id", 1),
+                Update = Update.Replace(new BsonDocument { { "_id", 1 }, { "x", 2 } }),
+                VersionReturned = FindAndModifyDocumentVersion.Modified
+            };
+
+            Action action = () => collection.FindAndModify(args);
+
+            var exception = action.ShouldThrow<MongoWriteConcernException>().Which;
+            var commandResult = exception.Result;
+            var result = commandResult["value"].AsBsonDocument;
+            result.Should().Be("{ _id : 1, x : 2 }");
+        }
+
+        [Test]
+        [RequiresServer(ClusterTypes = ClusterTypes.ReplicaSet)]
+        public void TestFindAndModifyUpdateWithWriteConcernError()
+        {
+            _collection.RemoveAll();
+            _collection.Insert(new BsonDocument { { "_id", 1 }, { "x", 1 } });
+            var collectionSettings = new MongoCollectionSettings
+            {
+                WriteConcern = new WriteConcern(9)
+            };
+            var collection = _database.GetCollection(_collection.Name, collectionSettings);
+            var args = new FindAndModifyArgs
+            {
+                Query = Query.EQ("x", 1),
+                Update = Update.Set("x", 2),
+                VersionReturned = FindAndModifyDocumentVersion.Modified
+            };
+
+            Action action = () => collection.FindAndModify(args);
+
+            var exception = action.ShouldThrow<MongoWriteConcernException>().Which;
+            var commandResult = exception.Result;
+            var result = commandResult["value"].AsBsonDocument;
+            result.Should().Be("{ _id : 1, x : 2 }");
+        }
+
         private class FindAndModifyClass
         {
             public ObjectId Id;
@@ -1048,6 +1101,31 @@ public void TestFindAndRemoveWithMaxTime()
             }
         }
 
+        [Test]
+        [RequiresServer(ClusterTypes = ClusterTypes.ReplicaSet)]
+        public void TestFindAndRemoveWithWriteConcernError()
+        {
+            _collection.RemoveAll();
+            _collection.Insert(new BsonDocument("x", 1));
+            var collectionSettings = new MongoCollectionSettings
+            {
+                WriteConcern = new WriteConcern(9)
+            };
+            var collection = _database.GetCollection(_collection.Name, collectionSettings);
+            var args = new FindAndRemoveArgs
+            {
+                Query = Query.EQ("x", 1)              
+            };
+
+            Action action = () => collection.FindAndRemove(args);
+
+            var exception = action.ShouldThrow<MongoWriteConcernException>().Which;
+            var commandResult = exception.Result;
+            var result = commandResult["value"].AsBsonDocument;
+            result["x"].Should().Be(1);
+            _collection.Count().Should().Be(0);
+        }
+
         [Test]
         public void TestFindNearSphericalFalse()
         {
@@ -2281,6 +2359,24 @@ public void TestInsertDuplicateKey()
             CheckExpectedResult(expectedResult, result);
         }
 
+        [Test]
+        [RequiresServer(ClusterTypes = ClusterTypes.ReplicaSet)]
+        public void TestInsertWithWriteConcernError()
+        {
+            _collection.RemoveAll();
+            var document = new BsonDocument { { "_id", 1 }, { "x", 1 } };
+            var collectionSettings = new MongoCollectionSettings
+            {
+                WriteConcern = new WriteConcern(9)
+            };
+            var collection = _database.GetCollection(_collection.Name, collectionSettings);
+
+            Action action = () => collection.Insert(document);
+
+            action.ShouldThrow<MongoWriteConcernException>();
+            _collection.FindOne().Should().Be(document);
+        }
+
         [Test]
         public void TestIsCappedFalse()
         {
@@ -2727,6 +2823,25 @@ public void TestRemoveUnacknowledeged()
             }
         }
 
+        [Test]
+        [RequiresServer(ClusterTypes = ClusterTypes.ReplicaSet)]
+        public void TestRemoveWithWriteConcernError()
+        {
+            _collection.RemoveAll();
+            _collection.Insert(new BsonDocument { { "_id", 1 }, { "x", 1 } });
+            var collectionSettings = new MongoCollectionSettings
+            {
+                WriteConcern = new WriteConcern(9)
+            };
+            var collection = _database.GetCollection(_collection.Name, collectionSettings);
+            var query = Query.EQ("x", 1);
+
+            Action action = () => collection.Remove(query);
+
+            action.ShouldThrow<MongoWriteConcernException>();
+            _collection.Count().Should().Be(0);
+        }
+
         [Test]
         public void TestSetFields()
         {
@@ -2967,6 +3082,26 @@ public void TestUpdateUnacknowledged()
             }
         }
 
+        [Test]
+        [RequiresServer(ClusterTypes = ClusterTypes.ReplicaSet)]
+        public void TestUpdateWithWriteConcernError()
+        {
+            _collection.RemoveAll();
+            _collection.Insert(new BsonDocument { { "_id", 1 }, { "x", 1 } });
+            var collectionSettings = new MongoCollectionSettings
+            {
+                WriteConcern = new WriteConcern(9)
+            };
+            var collection = _database.GetCollection(_collection.Name, collectionSettings);
+            var query = Query.EQ("x", 1);
+            var update = Update.Set("x", 2);
+
+            Action action = () => collection.Update(query, update);
+
+            action.ShouldThrow<MongoWriteConcernException>();
+            _collection.FindOne().Should().Be("{ _id : 1, x : 2 }");
+        }
+
         [Test]
         public void TestUpsertExisting()
         {