Added support for checking a file content result's content type.

Partial implementation of #3.

Author: AlexArchive

TestStack.FluentMVCTesting.Tests/ControllerResultTestTests.cs

@@ -30,6 +30,7 @@ class ControllerResultTestShould
             ReturnType<FileStreamResult>(t => t.ShouldRenderFileStream()),
             ReturnType<FileContentResult>(t => t.ShouldRenderFileContents()),
             ReturnType<FileContentResult>(t => t.ShouldRenderFileContents(new byte[0])),
+            ReturnType<FileContentResult>(t => t.ShouldRenderFileContents(new byte[0], "")),
             ReturnType<FileStreamResult>(t => t.ShouldRenderFileStream("")),
             ReturnType<FilePathResult>(t => t.ShouldRenderFilePath()),
             ReturnType<FilePathResult>(t => t.ShouldRenderFilePath("")),
@@ -364,6 +365,23 @@ public void Check_for_file_content_result_and_check_invalid_binary_content()
             Assert.True(string.Join(",", ControllerResultTestController.FileContents).All(exception.Message.Contains));
         }
 
+        [Test]
+        public void Check_for_file_content_result_and_check_binary_content_and_check_content_type()
+        {
+            _controller.WithCallTo(c => c.File()).ShouldRenderFileContents(ControllerResultTestController.FileContents, ControllerResultTestController.FileContentType);
+        }
+
+        [Test]
+        public void Check_for_file_content_result_and_check_invalid_content_type()
+        {
+            const string contentType = "application/dummy";
+
+            var exception = Assert.Throws<ActionResultAssertionException>(() =>
+                _controller.WithCallTo(c => c.File()).ShouldRenderFileContents(ControllerResultTestController.FileContents, contentType));
+
+            Assert.That(exception.Message, Is.EqualTo(string.Format("Expected file to be of content type '{0}', but instead was given '{1}'.", contentType, ControllerResultTestController.FileContentType)));
+        }
+
         [Test]
         public void Check_for_file_result()
         {

TestStack.FluentMvcTesting/ControllerResultTest.cs

@@ -7,6 +7,7 @@
 using System.Text.RegularExpressions;
 using System.Web.Mvc;
 using System.Web.Routing;
+using System.Web.UI.WebControls;
 
 namespace TestStack.FluentMVCTesting
 {
@@ -230,7 +231,7 @@ public FileResult ShouldRenderAnyFile(string contentType = null)
             return fileResult;
         }
 
-        public FileContentResult ShouldRenderFileContents(byte[] contents = null)
+        public FileContentResult ShouldRenderFileContents(byte[] contents = null, string contentType = null)
         {
             ValidateActionReturnType<FileContentResult>();
 
@@ -244,6 +245,11 @@ public FileContentResult ShouldRenderFileContents(byte[] contents = null)
                     string.Join(",", fileResult.FileContents)));
             }
 
+            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;
         }