CSHARP-1439: Add document validation support to create collection methods.

Author: rstam

src/MongoDB.Driver.Core.Tests/Core/Operations/CreateCollectionOperationTests.cs

@@ -230,6 +230,65 @@ public void CreateCommand_should_return_expected_result_when_UsePowerOf2Sizes_is
             result.Should().Be(expectedResult);
         }
 
+        [Test]
+        public void CreateCommand_should_return_expected_result_when_ValidationAction_is_set(
+            [Values(DocumentValidationAction.Error, DocumentValidationAction.Warn)]
+            DocumentValidationAction value)
+        {
+            var subject = new CreateCollectionOperation(_collectionNamespace, _messageEncoderSettings)
+            {
+                ValidationAction = value
+            };
+            var expectedResult = new BsonDocument
+            {
+                { "create", _collectionNamespace.CollectionName },
+                { "validationAction", value.ToString().ToLower() }
+            };
+
+            var result = subject.CreateCommand();
+
+            result.Should().Be(expectedResult);
+        }
+
+        [Test]
+        public void CreateCommand_should_return_expected_result_when_ValidationLevel_is_set(
+            [Values(DocumentValidationLevel.Moderate, DocumentValidationLevel.Off)]
+            DocumentValidationLevel value)
+        {
+            var subject = new CreateCollectionOperation(_collectionNamespace, _messageEncoderSettings)
+            {
+                ValidationLevel = value
+            };
+            var expectedResult = new BsonDocument
+            {
+                { "create", _collectionNamespace.CollectionName },
+                { "validationLevel", value.ToString().ToLower() }
+            };
+
+            var result = subject.CreateCommand();
+
+            result.Should().Be(expectedResult);
+        }
+
+        [Test]
+        public void CreateCommand_should_return_expected_result_when_Validator_is_set()
+        {
+            var value = new BsonDocument("x", 1);
+            var subject = new CreateCollectionOperation(_collectionNamespace, _messageEncoderSettings)
+            {
+                Validator = value
+            };
+            var expectedResult = new BsonDocument
+            {
+                { "create", _collectionNamespace.CollectionName },
+                { "validator", value }
+            };
+
+            var result = subject.CreateCommand();
+
+            result.Should().Be(expectedResult);
+        }
+
         [Test]
         [RequiresServer("DropCollection")]
         public void Execute_should_create_collection(
@@ -455,6 +514,46 @@ public void UsePowerOf2Sizes_should_work()
             subject.UsePowerOf2Sizes.Should().BeTrue();
         }
 
+        [Test]
+        public void ValidationAction_should_work(
+            [Values(null, DocumentValidationAction.Error, DocumentValidationAction.Warn)]
+            DocumentValidationAction? value)
+        {
+            var subject = new CreateCollectionOperation(_collectionNamespace, _messageEncoderSettings);
+            subject.ValidationAction.Should().BeNull();
+
+            subject.ValidationAction = value;
+
+            subject.ValidationAction.Should().Be(value);
+        }
+
+        [Test]
+        public void ValidationLevel_should_work(
+            [Values(null, DocumentValidationLevel.Moderate, DocumentValidationLevel.Off)]
+            DocumentValidationLevel? value)
+        {
+            var subject = new CreateCollectionOperation(_collectionNamespace, _messageEncoderSettings);
+            subject.ValidationLevel.Should().BeNull();
+
+            subject.ValidationLevel = value;
+
+            subject.ValidationLevel.Should().Be(value);
+        }
+
+        [Test]
+        public void Validator_should_work(
+            [Values(null, "{ x : 1 }")]
+            string json)
+        {
+            var subject = new CreateCollectionOperation(_collectionNamespace, _messageEncoderSettings);
+            var value = json == null ? null : BsonDocument.Parse(json);
+            subject.Validator.Should().BeNull();
+
+            subject.Validator = value;
+
+            subject.Validator.Should().Be(value);
+        }
+
         // helper methods
         private void DropCollection()
         {

src/MongoDB.Driver.Core/Core/Operations/CreateCollectionOperation.cs

@@ -38,6 +38,9 @@ public class CreateCollectionOperation : IWriteOperation<BsonDocument>
         private readonly MessageEncoderSettings _messageEncoderSettings;
         private BsonDocument _storageEngine;
         private bool? _usePowerOf2Sizes;
+        private DocumentValidationAction? _validationAction;
+        private DocumentValidationLevel? _validationLevel;
+        private BsonDocument _validator;
 
         // constructors
         /// <summary>
@@ -148,6 +151,42 @@ public bool? UsePowerOf2Sizes
             set { _usePowerOf2Sizes = value; }
         }
 
+        /// <summary>
+        /// Gets or sets the validation action.
+        /// </summary>
+        /// <value>
+        /// The validation action.
+        /// </value>
+        public DocumentValidationAction? ValidationAction
+        {
+            get { return _validationAction; }
+            set { _validationAction = value; }
+        }
+
+        /// <summary>
+        /// Gets or sets the validation level.
+        /// </summary>
+        /// <value>
+        /// The validation level.
+        /// </value>
+        public DocumentValidationLevel? ValidationLevel
+        {
+            get { return _validationLevel; }
+            set { _validationLevel = value; }
+        }
+
+        /// <summary>
+        /// Gets or sets the validator.
+        /// </summary>
+        /// <value>
+        /// The validator.
+        /// </value>
+        public BsonDocument Validator
+        {
+            get { return _validator; }
+            set { _validator = value; }
+        }
+
         // methods
         internal BsonDocument CreateCommand()
         {
@@ -159,7 +198,10 @@ internal BsonDocument CreateCommand()
                 { "size", () => _maxSize.Value, _maxSize.HasValue },
                 { "max", () => _maxDocuments.Value, _maxDocuments.HasValue },
                 { "flags", () => _usePowerOf2Sizes.Value ? 1 : 0, _usePowerOf2Sizes.HasValue},
-                { "storageEngine", () => _storageEngine, _storageEngine != null }
+                { "storageEngine", () => _storageEngine, _storageEngine != null },
+                { "validator", _validator, _validator != null },
+                { "validationAction", () => _validationAction.Value.ToString().ToLower(), _validationAction.HasValue },
+                { "validationLevel", () => _validationLevel.Value.ToString().ToLower(), _validationLevel.HasValue }
             };
         }
 

src/MongoDB.Driver.Core/Core/Operations/DocumentValidationAction.cs

@@ -0,0 +1,32 @@
+/* Copyright 2015 MongoDB Inc.
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+
+namespace MongoDB.Driver.Core.Operations
+{
+    /// <summary>
+    /// Represents the document validation action.
+    /// </summary>
+    public enum DocumentValidationAction
+    {
+        /// <summary>
+        /// Validation failures result in an error.
+        /// </summary>
+        Error = 0, // the default
+        /// <summary>
+        /// Validation failures result in a warning.
+        /// </summary>
+        Warn
+    }
+}

src/MongoDB.Driver.Core/Core/Operations/DocumentValidationLevel.cs

@@ -0,0 +1,36 @@
+/* Copyright 2015 MongoDB Inc.
+*
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+* http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+
+namespace MongoDB.Driver.Core.Operations
+{
+    /// <summary>
+    /// Represents the document validation level.
+    /// </summary>
+    public enum DocumentValidationLevel
+    {
+        /// <summary>
+        /// Strict document validation.
+        /// </summary>
+        Strict = 0, // the default
+        /// <summary>
+        /// Moderate document validation.
+        /// </summary>
+        Moderate,
+        /// <summary>
+        /// No document validation.
+        /// </summary>
+        Off
+    }
+}

src/MongoDB.Driver.Core/MongoDB.Driver.Core.csproj

@@ -137,6 +137,8 @@
     <Compile Include="Core\Misc\SemaphoreSlimRequest.cs" />
     <Compile Include="Core\Operations\CreateIndexesUsingCommandOperation.cs" />
     <Compile Include="Core\Operations\CreateIndexesUsingInsertOperation.cs" />
+    <Compile Include="Core\Operations\DocumentValidationAction.cs" />
+    <Compile Include="Core\Operations\DocumentValidationLevel.cs" />
     <Compile Include="Core\Operations\ListCollectionsUsingCommandOperation.cs" />
     <Compile Include="Core\Operations\ListCollectionsUsingQueryOperation.cs" />
     <Compile Include="Core\Operations\ListIndexesUsingCommandOperation.cs" />

src/MongoDB.Driver.Tests/MongoDatabaseImplTests.cs

@@ -19,6 +19,7 @@
 using System.Threading.Tasks;
 using FluentAssertions;
 using MongoDB.Bson;
+using MongoDB.Bson.Serialization;
 using MongoDB.Bson.Serialization.Attributes;
 using MongoDB.Driver.Core.Bindings;
 using MongoDB.Driver.Core.Clusters;
@@ -77,7 +78,10 @@ public async Task CreateCollectionAsync_should_execute_the_CreateCollectionOpera
                 MaxDocuments = 10,
                 MaxSize = 11,
                 StorageEngine = storageEngine,
-                UsePowerOf2Sizes = false
+                UsePowerOf2Sizes = false,
+                ValidationAction = DocumentValidationAction.Warn,
+                ValidationLevel = DocumentValidationLevel.Off,
+                Validator = new BsonDocument("x", 1)
             };
             await _subject.CreateCollectionAsync("bar", options, CancellationToken.None);
 
@@ -92,6 +96,12 @@ public async Task CreateCollectionAsync_should_execute_the_CreateCollectionOpera
             op.MaxSize.Should().Be(options.MaxSize);
             op.StorageEngine.Should().Be(storageEngine);
             op.UsePowerOf2Sizes.Should().Be(options.UsePowerOf2Sizes);
+            op.ValidationAction.Should().Be(options.ValidationAction);
+            op.ValidationLevel.Should().Be(options.ValidationLevel);
+            var serializerRegistry = options.SerializerRegistry ?? BsonSerializer.SerializerRegistry;
+            var documentSerializer = options.DocumentSerializer ?? serializerRegistry.GetSerializer<BsonDocument>();
+            var renderedValidator = options.Validator.Render(documentSerializer, serializerRegistry);
+            op.Validator.Should().Be(renderedValidator);
         }
 
         [Test]