Support for checking for temp data key existence.

Author: AlexArchive

TestStack.FluentMVCTesting.Tests/ControllerExtensionsTests.cs

@@ -1,4 +1,5 @@
-using NUnit.Framework;
+using System;
+using NUnit.Framework;
 using TestStack.FluentMVCTesting.Tests.TestControllers;
 
 namespace TestStack.FluentMVCTesting.Tests
@@ -41,5 +42,25 @@ public void Throw_exception_for_child_action_call_to_non_child_action()
             var exception = Assert.Throws<InvalidControllerActionException>(() => _controller.WithCallToChild(c => c.SomeAction()));
             Assert.That(exception.Message, Is.EqualTo("Expected action SomeAction of controller ControllerExtensionsController to be a child action, but it didn't have the ChildActionOnly attribute."));
         }
+
+        [Test]
+        public void Check_for_existent_temp_data_property()
+        {
+            const string key = "";
+            _controller.TempData[key] = "";
+
+            _controller.ShouldHaveTempDataProperty(key);
+        }
+
+        [Test]
+        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)));
+        }
     }
 }

TestStack.FluentMvcTesting/ControllerExtensions.cs

@@ -59,5 +59,16 @@ public static ControllerResultTest<T> WithCallToChild<T, TAction>(this T control
 
             return controller.WithCallTo(actionCall);
         }
+
+        public static void ShouldHaveTempDataProperty(this Controller controller, string key)
+        {
+            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));
+            }
+        }
     }
 }

TestStack.FluentMvcTesting/Exceptions.cs

@@ -3,6 +3,11 @@
 namespace TestStack.FluentMVCTesting
 {
 
+    public class TempDataAssertionException : Exception
+    {
+        public TempDataAssertionException(string message) : base(message) { }
+    }
+
     public class ActionResultAssertionException : Exception
     {
         public ActionResultAssertionException(string message) : base(message) { }