Merge pull request #18 from TestStack/file-stream-result-support

Added support for checking for a file stream result

Author: robdmoore

TestStack.FluentMVCTesting.Tests/ControllerResultTestTests.cs

@@ -25,6 +25,7 @@ class ControllerResultTestShould
             ReturnType<PartialViewResult>(t => t.ShouldRenderPartialView("")),
             ReturnType<PartialViewResult>(t => t.ShouldRenderDefaultPartialView()),
             ReturnType<FileContentResult>(t => t.ShouldRenderFile()),
+            ReturnType<FileStreamResult>(t => t.ShouldRenderFileStream()),
             ReturnType<HttpStatusCodeResult>(t => t.ShouldGiveHttpStatus()),
             ReturnType<JsonResult>(t => t.ShouldReturnJson()),
         };
@@ -311,6 +312,18 @@ public void Check_for_file_result_and_check_content_type()
             _controller.WithCallTo(c => c.EmptyFile()).ShouldRenderFile(ControllerResultTestController.FileContentType);
         }
 
+        [Test]
+        public void Check_for_file_stream_result()
+        {
+            _controller.WithCallTo(c => c.EmptyStream()).ShouldRenderFileStream();
+        }
+
+        [Test]
+        public void Check_for_file_stream_result_and_check_content_type()
+        {
+            _controller.WithCallTo(c => c.EmptyStream()).ShouldRenderFileStream(ControllerResultTestController.FileContentType);
+        }
+
         #endregion
 
         #region HTTP Status tests

TestStack.FluentMVCTesting.Tests/TestControllers/ControllerResultTestController.cs

@@ -1,4 +1,5 @@
-using System.Web.Mvc;
+using System.IO;
+using System.Web.Mvc;
 
 namespace TestStack.FluentMVCTesting.Tests.TestControllers
 {
@@ -135,6 +136,12 @@ public ActionResult EmptyFile()
             return File(content, FileContentType);
         }
 
+        public ActionResult EmptyStream()
+        {
+            var content = new MemoryStream();
+            return File(content, FileContentType);
+        }
+
         public ActionResult NamedView()
         {
             return View(ViewName);

TestStack.FluentMvcTesting/ControllerResultTest.cs

@@ -224,6 +224,20 @@ public FileContentResult ShouldRenderFile(string contentType = null)
             return fileResult;
         }
 
+        public FileStreamResult ShouldRenderFileStream(string contentType = null)
+        {
+            ValidateActionReturnType<FileStreamResult>();
+
+            var fileResult = (FileStreamResult)_actionResult;
+
+            if (contentType != null && fileResult.ContentType != contentType)
+            {
+                throw new ActionResultAssertionException(string.Format("Expected file to be of content type '{0}', but instead was given '{1}'.", contentType, fileResult.ContentType));
+            }
+
+            return fileResult;
+        }
+
         #endregion
 
         #region Http Status