Support for checking for temp data value with predicate.

Author: AlexArchive

TestStack.FluentMVCTesting.Tests/ControllerExtensionsTests.cs

@@ -111,5 +111,28 @@ public void Check_for_existent_temp_data_property_and_check_value_valid_using_re
 
             _controller.ShouldHaveTempDataProperty(key, expectedValue);
         }
+
+        [Test]
+        public void Check_for_existent_temp_data_property_and_check_value_using_valid_predicate()
+        {
+            const string key = "";
+            const int value = 1;
+            _controller.TempData[key] = value;
+
+            _controller
+                .ShouldHaveTempDataProperty<int>(key, x => x == value);
+        }
+
+        [Test]
+        public void Check_for_existent_temp_data_property_and_check_value_using_invalid_predicate()
+        {
+            const string key = "";
+            _controller.TempData[key] = 1;
+
+            var exception = Assert.Throws<TempDataAssertionException>(() =>
+                _controller.ShouldHaveTempDataProperty<int>(key, x => x == 0));
+
+            Assert.That(exception.Message, Is.EqualTo("Expected view model to pass the given condition, but it failed."));
+        }
     }
 }

TestStack.FluentMvcTesting/ControllerExtensions.cs

@@ -72,21 +72,35 @@ public static void ShouldHaveTempDataProperty(this Controller controller, string
                     "Expected TempData to have a non-null value with key \"{0}\", but none found.", key));
             }
 
-            if (value == null) return;
-
-            if (actual.GetType() != value.GetType())
+            if (value != null && actual.GetType() != value.GetType())
             {
                 throw new TempDataAssertionException(string.Format(
                     "Expected value to be of type {0}, but instead was {1}.",
                     value.GetType().FullName,
                     controller.TempData[key].GetType().FullName));
-                }
+            }
 
-            if (!value.Equals(actual))
+            if (value != null && !value.Equals(actual))
             {
                 throw new TempDataAssertionException(string.Format(
                     "Expected value for key \"{0}\" to be \"{1}\", but instead found \"{2}\"", key, value, actual));
             }
         }
+
+        public static void ShouldHaveTempDataProperty<TValue>(this Controller controller, string key, Func<TValue, bool> predicate)
+        {
+            var actual = controller.TempData[key];
+
+            if (actual == null)
+            {
+                throw new TempDataAssertionException(string.Format(
+                    "Expected TempData to have a non-null value with key \"{0}\", but none found.", key));
+            }
+
+            if (!predicate((TValue)actual))
+            {
+                throw new TempDataAssertionException("Expected view model to pass the given condition, but it failed.");
+            }
+        }
     }
 }