Viewbagdatasupport

TestStack.FluentMVCTesting.Mvc3/TestStack.FluentMVCTesting.Mvc3.csproj

@@ -36,6 +36,9 @@
     <SignAssembly>false</SignAssembly>
   </PropertyGroup>
   <ItemGroup>
+    <Reference Include="ImpromptuInterface">
+      <HintPath>..\packages\ImpromptuInterface.6.0.6\lib\net40\ImpromptuInterface.dll</HintPath>
+    </Reference>
     <Reference Include="Microsoft.Web.Infrastructure, Version=1.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
       <Private>True</Private>
       <HintPath>..\packages\Microsoft.Web.Infrastructure.1.0.0.0\lib\net40\Microsoft.Web.Infrastructure.dll</HintPath>

TestStack.FluentMVCTesting.Mvc3/packages.config

@@ -1,5 +1,6 @@
 <?xml version="1.0" encoding="utf-8"?>
 <packages>
+  <package id="ImpromptuInterface" version="6.0.6" targetFramework="net40" />
   <package id="Microsoft.AspNet.Mvc" version="3.0.20105.1" targetFramework="net40" />
   <package id="Microsoft.AspNet.Razor" version="1.0.20105.408" targetFramework="net40" />
   <package id="Microsoft.AspNet.WebPages" version="1.0.20105.408" targetFramework="net40" />

TestStack.FluentMVCTesting.Tests/ControllerExtensionsTests.cs

@@ -1,4 +1,6 @@
-using NUnit.Framework;
+//
+using NUnit.Framework;
+//
 using TestStack.FluentMVCTesting.Tests.TestControllers;
 
 namespace TestStack.FluentMVCTesting.Tests

TestStack.FluentMVCTesting.Tests/ControllerResultTestTests.cs

@@ -1,9 +1,12 @@
-using System;
+//
+using System;
 using System.Collections.Generic;
 using System.Linq.Expressions;
 using System.Net;
 using System.Web.Mvc;
+//
 using NUnit.Framework;
+//
 using TestStack.FluentMVCTesting.Tests.TestControllers;
 
 namespace TestStack.FluentMVCTesting.Tests

TestStack.FluentMVCTesting.Tests/ModelErrorTestTests.cs

@@ -1,8 +1,11 @@
-using System;
+using System;//
+
 using System.Collections.Generic;
 using System.Linq.Expressions;
 using System.Web.Mvc;
+//
 using NSubstitute;
+//
 using NUnit.Framework;
 
 namespace TestStack.FluentMVCTesting.Tests

TestStack.FluentMVCTesting.Tests/ModelTestTests.cs

@@ -1,4 +1,5 @@
-using NUnit.Framework;
+//
+using NUnit.Framework;
 
 namespace TestStack.FluentMVCTesting.Tests
 {

TestStack.FluentMVCTesting.Tests/ViewResultTestTests.cs

@@ -1,4 +1,8 @@
-using System.Web.Mvc;
+//
+using System;
+using System.Collections.Generic;
+using System.Web.Mvc;
+//
 using NUnit.Framework;
 
 namespace TestStack.FluentMVCTesting.Tests
@@ -19,6 +23,69 @@ public void Setup()
             _viewResultTest = new ViewResultTest(_viewResult, new ViewTestController());
         }
 
+        [Test]
+        public void Check_string_message_key_on_viewbag()
+        {
+            _viewResult.ViewBag.Message = "Hello View Data Message";
+            Assert.That(_viewResultTest.WithViewBag<string>("Message"), Is.EqualTo("Hello View Data Message"));
+        }
+
+        [Test]
+        public void Check_int_message_key_on_viewbag()
+        {
+            _viewResult.ViewBag.Page = 1;
+            Assert.That(_viewResultTest.WithViewBag<int>("Page"), Is.EqualTo(1));
+        }
+
+        [Test]
+        public void Check_wrong_message_member_name_passed_missing_member_message_for_viewbag()
+        {
+            _viewResult.ViewBag.Message = "Hello View Data Message";
+
+            var exception = Assert.Throws<MissingMemberException>(() =>
+                _viewResultTest.WithViewBag<string>("Message2")
+            );
+
+            Assert.That(exception.Message, Is.EqualTo("Member 'ViewBag.Message2' not found."));
+        }
+
+
+        [Test]
+        public void Check_string_message_key_on_viewdata()
+        {
+            _viewResult.ViewData["Message"] = "Hello View Data Message";
+            Assert.That(_viewResultTest.WithViewData<string>("Message"), Is.EqualTo("Hello View Data Message"));
+        }
+
+        [Test]
+        public void Check_int_message_key_on_viewdata()
+        {
+            _viewResult.ViewData["Page"] = 1;
+            Assert.That(_viewResultTest.WithViewData<int>("Page"), Is.EqualTo(1));
+        }
+
+        [Test]
+        public void Check_wrong_message_key_passed_missing_message_key_for_viewdata()
+        {
+            _viewResult.ViewData["Message"] = "Hello View Data Message";
+
+            var exception = Assert.Throws<KeyNotFoundException>(() =>
+                _viewResultTest.WithViewData<string>("Message2")
+            );
+
+            Assert.That(exception.Message, Is.EqualTo("Exception with ViewData, 'Message2' key not found"));
+        }
+
+        [Test]
+        public void Check_row_count_zero_for_viewdata()
+        {
+            var exception = Assert.Throws<KeyNotFoundException>(() =>
+                _viewResultTest.WithViewData<string>("Message")
+            );
+
+            Assert.That(exception.Message, Is.EqualTo("Exception with ViewData, 'Message' key not found"));
+        }
+
         [Test]
         public void Check_the_type_of_model()
         {

TestStack.FluentMvcTesting/ControllerExtensions.cs

@@ -1,4 +1,5 @@
-using System;
+//
+using System;
 using System.Linq.Expressions;
 using System.Web.Mvc;
 

TestStack.FluentMvcTesting/ControllerResultTest.cs

@@ -1,3 +1,4 @@
+//
 using System;
 using System.Linq.Expressions;
 using System.Net;

TestStack.FluentMvcTesting/Exceptions.cs

@@ -1,4 +1,5 @@
-using System;
+//
+using System;
 
 namespace TestStack.FluentMVCTesting
 {

TestStack.FluentMvcTesting/ModelErrorTest.cs

@@ -1,3 +1,4 @@
+//
 using System;
 using System.Collections.Generic;
 using System.Linq;

TestStack.FluentMvcTesting/ModelTest.cs

@@ -1,3 +1,4 @@
+//
 using System;
 using System.Linq.Expressions;
 using System.Web.Mvc;

TestStack.FluentMvcTesting/TestStack.FluentMVCTesting.csproj

@@ -36,6 +36,9 @@
     <SignAssembly>false</SignAssembly>
   </PropertyGroup>
   <ItemGroup>
+    <Reference Include="ImpromptuInterface">
+      <HintPath>..\packages\ImpromptuInterface.6.0.6\lib\net40\ImpromptuInterface.dll</HintPath>
+    </Reference>
     <Reference Include="Microsoft.Web.Infrastructure, Version=1.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
       <Private>True</Private>
       <HintPath>..\packages\Microsoft.Web.Infrastructure.1.0.0.0\lib\net40\Microsoft.Web.Infrastructure.dll</HintPath>

TestStack.FluentMvcTesting/ViewResultTest.cs

@@ -1,5 +1,10 @@
+//
 using System;
+using System.Collections.Generic;
 using System.Web.Mvc;
+using System.Linq;
+//
+using ImpromptuInterface;
 
 namespace TestStack.FluentMVCTesting
 {
@@ -13,6 +18,33 @@ public ViewResultTest(ViewResultBase viewResult, Controller controller)
             _viewResult = viewResult;
             _controller = controller;
         }
+
+        public TMember WithViewBag<TMember>(string psMemberName)
+        {
+            TMember member = default(TMember);
+            var target = _viewResult.ViewBag;
+
+            IEnumerable<string> memberNames = Impromptu.GetMemberNames(_viewResult.ViewBag, true);
+            if ( !memberNames.Contains(psMemberName) )
+                throw new MissingMemberException("ViewBag",psMemberName);
+
+            member = Impromptu.InvokeGet(target, psMemberName);
+
+            return member;
+        }
+        public TProperty WithViewData<TProperty>(string psKeyName)
+        {
+            if (_viewResult.ViewData == null || _viewResult.ViewData.Count <= 0 || !_viewResult.ViewData.ContainsKey(psKeyName))
+                throw new KeyNotFoundException(string.Format("Exception with ViewData, '{0}' key not found",psKeyName));
+
+            var target = _viewResult.ViewData;
+            TProperty propertyValue = default(TProperty);
+
+            if (target != null)
+                propertyValue = (TProperty)target[psKeyName];
+
+            return propertyValue;
+        }
 
         public ModelTest<TModel> WithModel<TModel>() where TModel : class
         {

TestStack.FluentMvcTesting/packages.config

@@ -1,5 +1,6 @@
 <?xml version="1.0" encoding="utf-8"?>
 <packages>
+  <package id="ImpromptuInterface" version="6.0.6" targetFramework="net40" />
   <package id="Microsoft.AspNet.Mvc" version="4.0.20710.0" targetFramework="net40" />
   <package id="Microsoft.AspNet.Razor" version="2.0.20715.0" targetFramework="net40" />
   <package id="Microsoft.AspNet.WebPages" version="2.0.20710.0" targetFramework="net40" />