Support for checking for temp data value.

Author: AlexArchive

TestStack.FluentMVCTesting.Tests/ControllerExtensionsTests.cs

@@ -1,4 +1,4 @@
-using System;
+using System.IO;
 using NUnit.Framework;
 using TestStack.FluentMVCTesting.Tests.TestControllers;
 
@@ -56,11 +56,60 @@ public void Check_for_existent_temp_data_property()
         public void Check_for_non_existent_temp_data_property()
         {
             const string key = "";
+
             var exception = Assert.Throws<TempDataAssertionException>(() =>
                 _controller.ShouldHaveTempDataProperty(key));
 
             Assert.That(exception.Message, Is.EqualTo(string.Format(
                 "Expected TempData to have a non-null value with key \"{0}\", but none found.", key)));
         }
+
+        [Test]
+        public void Check_for_existent_temp_data_property_and_check_value()
+        {
+            const string key = "";
+            const int value = 10;
+            _controller.TempData[key] = value;
+
+            _controller.ShouldHaveTempDataProperty(key, value);
+        }
+
+        [Test]
+        public void Check_for_existent_temp_data_property_and_check_invalid_value()
+        {
+            const string key = "";
+            const int actualValue = 0;
+            const int expectedValue = 1;
+            _controller.TempData[key] = actualValue;
+
+            var exception = Assert.Throws<TempDataAssertionException>(() =>
+                _controller.ShouldHaveTempDataProperty(key, expectedValue));
+
+            Assert.That(exception.Message, Is.EqualTo(string.Format("Expected value for key \"{0}\" to be \"{1}\", but instead found \"{2}\"", key, expectedValue, actualValue)));
+        }
+
+        [Test]
+        public void Check_for_existent_temp_data_property_and_check_invalid_value_of_different_types()
+        {
+            const string key = "";
+            const int actualValue = 0;
+            const string expectedValue = "one";
+            _controller.TempData[key] = actualValue;
+
+            var exception = Assert.Throws<TempDataAssertionException>(() =>
+                _controller.ShouldHaveTempDataProperty(key, expectedValue));
+
+            Assert.That(exception.Message, Is.EqualTo(string.Format("Expected value to be of type {0}, but instead was {1}.", expectedValue.GetType().FullName, actualValue.GetType().FullName)));
+        }
+
+        [Test]
+        public void Check_for_existent_temp_data_property_and_check_value_valid_using_referential_equality()
+        {
+            const string key = "";
+            MemoryStream expectedValue = new MemoryStream();
+            _controller.TempData[key] = expectedValue;
+
+            _controller.ShouldHaveTempDataProperty(key, expectedValue);
+        }
     }
 }

TestStack.FluentMvcTesting/ControllerExtensions.cs

@@ -1,5 +1,7 @@
 using System;
+using System.Collections.Generic;
 using System.Linq.Expressions;
+using System.Reflection.Emit;
 using System.Threading.Tasks;
 using System.Web.Mvc;
 
@@ -60,7 +62,7 @@ public static ControllerResultTest<T> WithCallToChild<T, TAction>(this T control
             return controller.WithCallTo(actionCall);
         }
 
-        public static void ShouldHaveTempDataProperty(this Controller controller, string key)
+        public static void ShouldHaveTempDataProperty(this Controller controller, string key, object value = null)
         {
             var actual = controller.TempData[key];
 
@@ -69,6 +71,22 @@ public static void ShouldHaveTempDataProperty(this Controller controller, string
                 throw new TempDataAssertionException(string.Format(
                     "Expected TempData to have a non-null value with key \"{0}\", but none found.", key));
             }
+
+            if (value == null) return;
+
+            if (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))
+            {
+                throw new TempDataAssertionException(string.Format(
+                    "Expected value for key \"{0}\" to be \"{1}\", but instead found \"{2}\"", key, value, actual));
+            }
         }
     }
 }