Added support for async controller actions

Author: robdmoore

TestStack.FluentMVCTesting.Tests/AsyncControllerTests.cs

@@ -0,0 +1,45 @@
+using NUnit.Framework;
+using TestStack.FluentMVCTesting.Tests.TestControllers;
+
+namespace TestStack.FluentMVCTesting.Tests
+{
+    class AsyncControllersShould
+    {
+        private AsyncController _controller;
+
+        [SetUp]
+        public void Setup()
+        {
+            _controller = new AsyncController();
+        }
+
+        [Test]
+        public void Work_correctly_for_valid_check()
+        {
+            _controller.WithCallTo(c => c.AsyncViewAction())
+                .ShouldRenderDefaultView();
+        }
+
+        [Test]
+        public void Work_correctly_for_invalid_check()
+        {
+            Assert.Throws<ActionResultAssertionException>(
+                () => _controller.WithCallTo(c => c.AsyncViewAction()).ShouldGiveHttpStatus()
+            );
+        }
+        [Test]
+        public void Work_correctly_for_valid_child_action_check()
+        {
+            _controller.WithCallToChild(c => c.AsyncChildViewAction())
+                .ShouldRenderDefaultView();
+        }
+
+        [Test]
+        public void Work_correctly_for_invalid_child_action_check()
+        {
+            Assert.Throws<InvalidControllerActionException>(
+                () => _controller.WithCallToChild(c => c.AsyncViewAction())
+            );
+        }
+    }
+}

TestStack.FluentMVCTesting.Tests/TestControllers/AsyncController.cs

@@ -0,0 +1,20 @@
+using System.Threading.Tasks;
+using System.Web.Mvc;
+
+namespace TestStack.FluentMVCTesting.Tests.TestControllers
+{
+    class AsyncController : Controller
+    {
+        public Task<ActionResult> AsyncViewAction()
+        {
+            // Task.FromResult would be better, but I want to support .NET 4
+            return Task.Factory.StartNew<ActionResult>(() => View());
+        }
+
+        [ChildActionOnly]
+        public Task<ActionResult> AsyncChildViewAction()
+        {
+            return Task.Factory.StartNew<ActionResult>(() => View());
+        }
+    }
+}

TestStack.FluentMVCTesting.Tests/TestStack.FluentMVCTesting.Tests.csproj

@@ -77,6 +77,8 @@
     <Reference Include="System.Xml" />
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="AsyncControllerTests.cs" />
+    <Compile Include="TestControllers\AsyncController.cs" />
     <Compile Include="ControllerExtensionsTests.cs" />
     <Compile Include="ControllerResultTestTests.cs" />
     <Compile Include="ModelTestTests.cs" />

TestStack.FluentMvcTesting/ControllerExtensions.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Linq.Expressions;
+using System.Threading.Tasks;
 using System.Web.Mvc;
 
 namespace TestStack.FluentMVCTesting
@@ -24,6 +25,17 @@ public static ControllerResultTest<T> WithCallTo<T, TAction>(this T controller,
             return new ControllerResultTest<T>(controller, actionName, actionResult);
         }
 
+        public static ControllerResultTest<T> WithCallTo<T, TAction>(this T controller, Expression<Func<T, Task<TAction>>> actionCall)
+            where T : Controller
+            where TAction : ActionResult
+        {
+            var actionName = ((MethodCallExpression)actionCall.Body).Method.Name;
+
+            var actionResult = actionCall.Compile().Invoke(controller).Result;
+
+            return new ControllerResultTest<T>(controller, actionName, actionResult);
+        }
+
         public static ControllerResultTest<T> WithCallToChild<T, TAction>(this T controller, Expression<Func<T, TAction>> actionCall)
             where T : Controller
             where TAction : ActionResult
@@ -35,5 +47,17 @@ public static ControllerResultTest<T> WithCallToChild<T, TAction>(this T control
 
             return controller.WithCallTo(actionCall);
         }
+
+        public static ControllerResultTest<T> WithCallToChild<T, TAction>(this T controller, Expression<Func<T, Task<TAction>>> actionCall)
+            where T : Controller
+            where TAction : ActionResult
+        {
+            var action = ((MethodCallExpression)actionCall.Body).Method;
+
+            if (!action.IsDefined(typeof(ChildActionOnlyAttribute), false))
+                throw new InvalidControllerActionException(string.Format("Expected action {0} of controller {1} to be a child action, but it didn't have the ChildActionOnly attribute.", action.Name, controller.GetType().Name));
+
+            return controller.WithCallTo(actionCall);
+        }
     }
 }