Added OfType method to ViewTestBuilderViewComponentsExtensions and tests (#148)

Author: stankovT

src/MyTested.AspNetCore.Mvc.ViewComponents.Results/ViewTestBuilderViewComponentsExtensions.cs

@@ -4,6 +4,7 @@
     using Builders.ViewComponentResults;
     using Exceptions;
     using Microsoft.AspNetCore.Mvc.ViewEngines;
+    using System;
     using Utilities;
 
     /// <summary>
@@ -48,18 +49,30 @@ public static IAndViewTestBuilder WithViewEngine(
         public static IAndViewTestBuilder WithViewEngineOfType<TViewEngine>(
             this IViewTestBuilder viewTestBuilder)
             where TViewEngine : IViewEngine
+            => WithViewEngineOfType(viewTestBuilder, typeof(TViewEngine));
+
+        /// <summary>
+        /// Tests whether <see cref="Microsoft.AspNetCore.Mvc.ViewComponents.ViewViewComponentResult"/>
+        /// has the same <see cref="IViewEngine"/> type as the provided one.
+        /// </summary>
+        /// <param name="viewTestBuilder">
+        /// Instance of <see cref="IViewTestBuilder"/> type.
+        /// </param>
+        /// <param name="viewEngineType"></param>
+        /// <returns>The same <see cref="IAndViewTestBuilder"/>.</returns>
+        public static IAndViewTestBuilder WithViewEngineOfType(
+            this IViewTestBuilder viewTestBuilder,Type viewEngineType)
         {
             var actualBuilder = GetActualBuilder(viewTestBuilder);
 
             var actualViewEngineType = actualBuilder.ViewResult?.ViewEngine?.GetType();
-            var expectedViewEngineType = typeof(TViewEngine);
 
             if (actualViewEngineType == null
-                || Reflection.AreDifferentTypes(expectedViewEngineType, actualViewEngineType))
+                || Reflection.AreDifferentTypes(viewEngineType, actualViewEngineType))
             {
                 throw ViewViewComponentResultAssertionException.ForViewEngineType(
                     actualBuilder.TestContext.ExceptionMessagePrefix,
-                    expectedViewEngineType.ToFriendlyTypeName(),
+                    viewEngineType.ToFriendlyTypeName(),
                     actualViewEngineType.ToFriendlyTypeName());
             }
 

test/MyTested.AspNetCore.Mvc.ViewComponents.Results.Test/BuildersTests/ViewComponentResultTests/ViewTestBuilderTests.cs

@@ -23,7 +23,7 @@ public void WithViewEngineShouldNotThrowExceptionWithValidViewEngine()
         }
 
         [Fact]
-        public void WithViewEngineShouldNotThrowExceptionWithNullViewEngine()
+        public void WithViewEngineShouldNotThrowExceptionWithNullViewEngineForGeneric()
         {
             MyViewComponent<ViewResultComponent>
                 .InvokedWith(c => c.Invoke("Test"))
@@ -32,6 +32,16 @@ public void WithViewEngineShouldNotThrowExceptionWithNullViewEngine()
                 .WithViewEngineOfType<CompositeViewEngine>();
         }
 
+        [Fact]
+        public void WithViewEngineShouldNotThrowExceptionWithNullViewEngine()
+        {
+            MyViewComponent<ViewResultComponent>
+                .InvokedWith(c => c.Invoke("Test"))
+                .ShouldReturn()
+                .View("SomeView")
+                .WithViewEngineOfType(typeof(CompositeViewEngine));
+        }
+
         [Fact]
         public void WithViewEngineShouldThrowExceptionWithInvalidViewEngine()
         {
@@ -48,7 +58,7 @@ public void WithViewEngineShouldThrowExceptionWithInvalidViewEngine()
         }
 
         [Fact]
-        public void WithViewEngineOfTypeShouldNotThrowExceptionWithValidViewEngine()
+        public void WithViewEngineOfTypeShouldNotThrowExceptionWithValidViewEngineForGeneric()
         {
             MyViewComponent<ViewEngineComponent>
                 .InvokedWith(c => c.Invoke(new CustomViewEngine()))
@@ -58,7 +68,17 @@ public void WithViewEngineOfTypeShouldNotThrowExceptionWithValidViewEngine()
         }
 
         [Fact]
-        public void WithViewEngineOfTypeShouldThrowExceptionWithInvalidViewEngine()
+        public void WithViewEngineOfTypeShouldNotThrowExceptionWithValidViewEngine()
+        {
+            MyViewComponent<ViewEngineComponent>
+                .InvokedWith(c => c.Invoke(new CustomViewEngine()))
+                .ShouldReturn()
+                .View()
+                .WithViewEngineOfType(typeof(CustomViewEngine));
+        }
+
+        [Fact]
+        public void WithViewEngineOfTypeShouldThrowExceptionWithInvalidViewEngineForGeneric()
         {
             Test.AssertException<ViewViewComponentResultAssertionException>(
                 () =>
@@ -73,7 +93,22 @@ public void WithViewEngineOfTypeShouldThrowExceptionWithInvalidViewEngine()
         }
 
         [Fact]
-        public void WithViewEngineOfTypeShouldNotThrowExceptionWithNullViewEngine()
+        public void WithViewEngineOfTypeShouldThrowExceptionWithInvalidViewEngine()
+        {
+            Test.AssertException<ViewViewComponentResultAssertionException>(
+                () =>
+                {
+                    MyViewComponent<ViewEngineComponent>
+                        .InvokedWith(c => c.Invoke(new CustomViewEngine()))
+                        .ShouldReturn()
+                        .View()
+                        .WithViewEngineOfType(typeof(IViewEngine));
+                },
+                "When invoking ViewEngineComponent expected view result ViewEngine to be of IViewEngine type, but instead received CustomViewEngine.");
+        }
+
+        [Fact]
+        public void WithViewEngineOfTypeShouldNotThrowExceptionWithNullViewEngineForGeneric()
         {
             Test.AssertException<ViewViewComponentResultAssertionException>(
                 () =>
@@ -88,7 +123,22 @@ public void WithViewEngineOfTypeShouldNotThrowExceptionWithNullViewEngine()
         }
 
         [Fact]
-        public void AndAlsoShouldWorkCorrectly()
+        public void WithViewEngineOfTypeShouldNotThrowExceptionWithNullViewEngine()
+        {
+            Test.AssertException<ViewViewComponentResultAssertionException>(
+                () =>
+                {
+                    MyViewComponent<ViewResultComponent>
+                        .InvokedWith(c => c.Invoke(null))
+                        .ShouldReturn()
+                        .View()
+                        .WithViewEngineOfType(typeof(CustomViewEngine));
+                },
+                "When invoking ViewResultComponent expected view result ViewEngine to be of CustomViewEngine type, but instead received CompositeViewEngine.");
+        }
+
+        [Fact]
+        public void AndAlsoShouldWorkCorrectlyForGeneric()
         {
             MyViewComponent<ViewResultComponent>
                 .InvokedWith(c => c.Invoke("All"))
@@ -98,5 +148,17 @@ public void AndAlsoShouldWorkCorrectly()
                 .AndAlso()
                 .WithModelOfType<ResponseModel>();
         }
+
+        [Fact]
+        public void AndAlsoShouldWorkCorrectly()
+        {
+            MyViewComponent<ViewResultComponent>
+                .InvokedWith(c => c.Invoke("All"))
+                .ShouldReturn()
+                .View("SomeView")
+                .WithViewEngineOfType(typeof(CompositeViewEngine))
+                .AndAlso()
+                .WithModelOfType(typeof(ResponseModel));
+        }
     }
 }