Added resolving of ViewComponent arguments from the lambda expression (#66)

Author: ivaylokenov

src/MyTested.AspNetCore.Mvc.Controllers/Builders/Controllers/ControllerActionCallBuilder.cs

@@ -56,6 +56,20 @@ protected override void ProcessAndValidateMethod(LambdaExpression methodCall, Me
             }
         }
 
+        private void SetActionDescriptor(MethodInfo methodInfo)
+        {
+            var controllerContext = this.TestContext.ComponentContext;
+            if (controllerContext.ActionDescriptor?.MethodInfo == null)
+            {
+                var controllerActionDescriptorCache = this.Services.GetService<IControllerActionDescriptorCache>();
+                if (controllerActionDescriptorCache != null)
+                {
+                    controllerContext.ActionDescriptor
+                        = controllerActionDescriptorCache.TryGetActionDescriptor(methodInfo);
+                }
+            }
+        }
+
         private void ValidateModelState(LambdaExpression actionCall)
         {
             var arguments = ExpressionParser.ResolveMethodArguments(actionCall).ToArray();
@@ -71,19 +85,5 @@ private void ValidateModelState(LambdaExpression actionCall)
                 });
             }
         }
-
-        private void SetActionDescriptor(MethodInfo methodInfo)
-        {
-            var controllerContext = this.TestContext.ComponentContext;
-            if (controllerContext.ActionDescriptor?.MethodInfo == null)
-            {
-                var controllerActionDescriptorCache = this.Services.GetService<IControllerActionDescriptorCache>();
-                if (controllerActionDescriptorCache != null)
-                {
-                    controllerContext.ActionDescriptor
-                        = controllerActionDescriptorCache.TryGetActionDescriptor(methodInfo);
-                }
-            }
-        }
     }
 }

src/MyTested.AspNetCore.Mvc.ViewComponents/Builders/ViewComponents/ViewComponentInvocationBuilder.cs

@@ -6,6 +6,10 @@
     using System.Threading.Tasks;
     using Contracts.Invocations;
     using Invocations;
+    using Internal.Contracts;
+    using Microsoft.Extensions.DependencyInjection;
+    using Utilities;
+    using Utilities.Extensions;
 
     public partial class ViewComponentBuilder<TViewComponent>
     {
@@ -23,8 +27,31 @@ public IViewComponentResultTestBuilder<TActionResult> InvokedWith<TActionResult>
             return new ViewComponentResultTestBuilder<TActionResult>(this.TestContext);
         }
 
-        protected override void ProcessAndValidateMethod(LambdaExpression methodCall, MethodInfo methodInfo)
+        protected override void ProcessAndValidateMethod(LambdaExpression invocationCall, MethodInfo methodInfo)
         {
+            this.SetViewComponentDescriptor(methodInfo);
+            this.ExtractArguments(invocationCall);
+        }
+
+        private void SetViewComponentDescriptor(MethodInfo methodInfo)
+        {
+            var viewComponentContext = this.TestContext.ViewComponentContext;
+            if (viewComponentContext.ViewComponentDescriptor?.MethodInfo == null)
+            {
+                var viewComponentDescriptorCache = this.Services.GetService<IViewComponentDescriptorCache>();
+                if (viewComponentDescriptorCache == null)
+                {
+                    viewComponentContext.ViewComponentDescriptor
+                        = viewComponentDescriptorCache.TryGetViewComponentDescriptor(methodInfo);
+                }
+            }
+        }
+
+        private void ExtractArguments(LambdaExpression invocationCall)
+        {
+            ExpressionParser
+                .ResolveMethodArguments(invocationCall)
+                .ForEach(arg => this.TestContext.ViewComponentContext.Arguments.Add(arg.Name, arg.Value));
         }
     }
 }