Added unit tests for number of errors in the ModelState (#33)

Author: ivaylokenov

src/MyWebApi.Tests/BuildersTests/ActionsTests/ShouldHaveModelStateTests.cs

@@ -80,6 +80,33 @@ public void ShouldHaveInvalidModelStateShouldBeValidWithInvalidRequestModel()
                 .InvalidModelState();
         }
 
+        [Test]
+        public void ShouldHaveInvalidModelStateShouldBeValidWithInvalidRequestModelAndCorrectNumberOfErrors()
+        {
+            var requestModelWithErrors = TestObjectFactory.GetRequestModelWithErrors();
+
+            MyWebApi
+                .Controller<WebApiController>()
+                .Calling(c => c.ModelStateCheck(requestModelWithErrors))
+                .ShouldHave()
+                .InvalidModelState(2);
+        }
+
+        [Test]
+        [ExpectedException(
+            typeof(ModelErrorAssertionException),
+            ExpectedMessage = "When calling ModelStateCheck action in WebApiController expected to have invalid model state with 5 errors, but contained 2.")]
+        public void ShouldHaveInvalidModelStateShouldBeInvalidWithInvalidRequestModelAndIncorrectNumberOfErrors()
+        {
+            var requestModelWithErrors = TestObjectFactory.GetRequestModelWithErrors();
+
+            MyWebApi
+                .Controller<WebApiController>()
+                .Calling(c => c.ModelStateCheck(requestModelWithErrors))
+                .ShouldHave()
+                .InvalidModelState(5);
+        }
+
         [Test]
         [ExpectedException(
             typeof(ModelErrorAssertionException),
@@ -95,6 +122,21 @@ public void ShouldHaveInvalidModelStateShouldThrowExceptionWithValidRequestModel
                 .InvalidModelState();
         }
 
+        [Test]
+        [ExpectedException(
+            typeof(ModelErrorAssertionException),
+            ExpectedMessage = "When calling ModelStateCheck action in WebApiController expected to have invalid model state with 5 errors, but contained 0.")]
+        public void ShouldHaveInvalidModelStateShouldThrowExceptionWithValidRequestModelAndProvidedNumberOfErrors()
+        {
+            var requestModel = TestObjectFactory.GetValidRequestModel();
+
+            MyWebApi
+                .Controller<WebApiController>()
+                .Calling(c => c.ModelStateCheck(requestModel))
+                .ShouldHave()
+                .InvalidModelState(5);
+        }
+
         [Test]
         public void AndShouldWorkCorrectlyWithValidModelState()
         {

src/MyWebApi/Builders/Actions/ShouldHave/ShouldHaveModelState.cs

@@ -56,14 +56,15 @@ public IAndTestBuilder<TActionResult> ValidModelState()
         public IAndTestBuilder<TActionResult> InvalidModelState(int? numberOfErrors = null)
         {
             var actualModelStateErrors = this.Controller.ModelState.Count;
-            if (actualModelStateErrors == (numberOfErrors ?? 0))
+            if (actualModelStateErrors == 0
+                || (numberOfErrors != null && actualModelStateErrors != numberOfErrors))
             {
                 throw new ModelErrorAssertionException(string.Format(
                     "When calling {0} action in {1} expected to have invalid model state{2}, {3}.",
                     this.ActionName,
                     this.Controller.GetName(),
                     numberOfErrors == null ? string.Empty : string.Format(" with {0} errors", numberOfErrors),
-                    numberOfErrors == null ? "but was in fact valid" : string.Format(" but contained {0}", actualModelStateErrors)));
+                    numberOfErrors == null ? "but was in fact valid" : string.Format("but contained {0}", actualModelStateErrors)));
             }
 
             return this.NewAndTestBuilder();