CSHARP-1424: Support PartialFilterExpression in CreateIndex.

Author: rstam

MongoDB.Driver/Builders/IndexOptionsBuilder.cs

@@ -99,6 +99,16 @@ public static IndexOptionsBuilder SetName(string value)
             return new IndexOptionsBuilder().SetName(value);
         }
 
+        /// <summary>
+        /// Sets the partial filter expression.
+        /// </summary>
+        /// <param name="value">The value.</param>
+        /// <returns>The builder (so method calls can be chained).</returns>
+        public static IndexOptionsBuilder SetPartialFilterExpression(IMongoQuery value)
+        {
+            return new IndexOptionsBuilder().SetPartialFilterExpression(value);
+        }
+
         /// <summary>
         /// Sets whether the index is a sparse index.
         /// </summary>
@@ -260,6 +270,17 @@ public IndexOptionsBuilder SetName(string value)
             return this;
         }
 
+        /// <summary>
+        /// Sets the partial filter expression.
+        /// </summary>
+        /// <param name="value">The value.</param>
+        /// <returns>The builder (so method calls can be chained).</returns>
+        public IndexOptionsBuilder SetPartialFilterExpression(IMongoQuery value)
+        {
+            _document["partialFilterExpression"] = value.ToBsonDocument();
+            return this;
+        }
+
         /// <summary>
         /// Sets whether the index is a sparse index.
         /// </summary>
@@ -445,6 +466,16 @@ public static IndexOptionsBuilder<TDocument> SetName(string value)
             return new IndexOptionsBuilder<TDocument>().SetName(value);
         }
 
+        /// <summary>
+        /// Sets the partial filter expression.
+        /// </summary>
+        /// <param name="value">The value.</param>
+        /// <returns>The builder (so method calls can be chained).</returns>
+        public static IndexOptionsBuilder<TDocument> SetPartialFilterExpression(IMongoQuery value)
+        {
+            return new IndexOptionsBuilder<TDocument>().SetPartialFilterExpression(value);
+        }
+
         /// <summary>
         /// Sets whether the index is a sparse index.
         /// </summary>
@@ -611,6 +642,17 @@ public IndexOptionsBuilder<TDocument> SetName(string value)
             return this;
         }
 
+        /// <summary>
+        /// Sets the partial filter expression.
+        /// </summary>
+        /// <param name="value">The value.</param>
+        /// <returns>The builder (so method calls can be chained).</returns>
+        public IndexOptionsBuilder<TDocument> SetPartialFilterExpression(IMongoQuery value)
+        {
+            _indexOptionsBuilder.SetPartialFilterExpression(value);
+            return this;
+        }
+
         /// <summary>
         /// Sets whether the index is a sparse index.
         /// </summary>

MongoDB.DriverUnitTests/Builders/IndexOptionsBuilderTests.cs

@@ -63,6 +63,14 @@ public void TestName()
             Assert.AreEqual(expected, options.ToJson());
         }
 
+        [Test]
+        public void TestPartialFilterExpression()
+        {
+            var options = IndexOptions.SetPartialFilterExpression(Query.GT("x", 0));
+            string expected = "{ \"partialFilterExpression\" : { \"x\" : { \"$gt\" : 0 } } }";
+            Assert.AreEqual(expected, options.ToJson());
+        }
+
         [Test]
         public void TestSparse()
         {

MongoDB.DriverUnitTests/Builders/IndexOptionsBuilderTypedTests.cs

@@ -71,6 +71,14 @@ public void TestName()
             Assert.AreEqual(expected, options.ToJson());
         }
 
+        [Test]
+        public void TestPartialFilterExpression()
+        {
+            var options = IndexOptions<TestClass>.SetPartialFilterExpression(Query.GT("x", 0));
+            string expected = "{ \"partialFilterExpression\" : { \"x\" : { \"$gt\" : 0 } } }";
+            Assert.AreEqual(expected, options.ToJson());
+        }
+
         [Test]
         public void TestSparse()
         {

MongoDB.DriverUnitTests/MongoCollectionTests.cs

@@ -714,6 +714,21 @@ public void TestCreateIndexWithStorageOptions()
             Assert.AreEqual(2, result.Count);
         }
 
+        [Test]
+        [RequiresServer(MinimumVersion = "3.1.9999")]
+        public void TestCreateIndexWithPartialFilterExpression()
+        {
+            _collection.Drop();
+            var keys = IndexKeys.Ascending("x");
+            var options = IndexOptions<BsonDocument>.SetPartialFilterExpression(Query.GT("x", 0));
+
+            _collection.CreateIndex(keys, options);
+
+            var indexes = _collection.GetIndexes();
+            var index = indexes.Where(i => i.Name == "x_1").Single();
+            Assert.That(index.RawDocument["partialFilterExpression"], Is.EqualTo(BsonDocument.Parse("{ x : { $gt : 0 } }")));
+        }
+
         [Test]
         public void TestDistinct()
         {