Fix Antiforgery support (#326)

Author: ivaylokenov

src/MyTested.AspNetCore.Mvc.Abstractions/Internal/WebFramework.cs

@@ -188,6 +188,26 @@ public static Type LambdaExpressionComparer
                     return type;
                 }
             }
+
+            public static Type IAntiforgeryFeature
+            {
+                get
+                {
+                    var assembly = Assembly.Load(new AssemblyName("Microsoft.AspNetCore.Antiforgery"));
+                    var type = assembly.GetType("Microsoft.AspNetCore.Antiforgery.IAntiforgeryFeature");
+                    return type;
+                }
+            }
+
+            public static Type AntiforgeryFeature
+            {
+                get
+                {
+                    var assembly = Assembly.Load(new AssemblyName("Microsoft.AspNetCore.Antiforgery"));
+                    var type = assembly.GetType("Microsoft.AspNetCore.Antiforgery.AntiforgeryFeature");
+                    return type;
+                }
+            }
         }
 
         // Copied from the ASP.NET Core source code.

src/MyTested.AspNetCore.Mvc.Controllers/Internal/Controllers/ControllerPropertyHelper.cs

@@ -44,15 +44,11 @@ public Func<object, ActionContext> ActionContextGetter
         }
 
         public static ControllerPropertyHelper GetProperties<TController>()
-            where TController : class
-        {
-            return GetProperties(typeof(TController));
-        }
+            where TController : class 
+            => GetProperties(typeof(TController));
 
-        public static ControllerPropertyHelper GetProperties(Type type)
-        {
-            return ControllerPropertiesCache.GetOrAdd(type, _ => new ControllerPropertyHelper(type));
-        }
+        public static ControllerPropertyHelper GetProperties(Type type) 
+            => ControllerPropertiesCache.GetOrAdd(type, _ => new ControllerPropertyHelper(type));
 
         private void TryCreateControllerContextDelegates()
         {

src/MyTested.AspNetCore.Mvc.ViewFeatures/HttpRequestBuilderViewFeaturesExtensions.cs

@@ -1,10 +1,13 @@
 namespace MyTested.AspNetCore.Mvc
 {
+    using System;
     using Builders.Contracts.Http;
     using Builders.Http;
+    using Internal;
     using Microsoft.AspNetCore.Antiforgery;
     using Microsoft.Extensions.DependencyInjection;
     using Microsoft.Extensions.Options;
+    using Utilities.Extensions;
 
     /// <summary>
     /// Contains <see cref="IAntiforgery"/> extension methods for <see cref="IHttpRequestBuilder"/>.
@@ -41,15 +44,20 @@ public static IAndHttpRequestBuilder WithAntiForgeryToken(this IHttpRequestBuild
                 actualHttpRequestBuilder.WithFormField(antiForgeryOptions.FormFieldName, tokens.RequestToken);
             }
 
-            //var generatedAntiforgeryFeature = httpContext.Features.Get<IAntiforgeryFeature>();
+            var antiforgeryFeature = WebFramework.Internals.IAntiforgeryFeature;
 
-            //httpContext.Features.Set<IAntiforgeryFeature>(new AntiforgeryFeature
-            //{
-            //    HaveDeserializedCookieToken = false,
-            //    HaveDeserializedRequestToken = false,
-            //    CookieToken = generatedAntiforgeryFeature.NewCookieToken,
-            //    RequestToken = generatedAntiforgeryFeature.NewRequestToken
-            //});
+            var generatedAntiforgeryFeature = httpContext.Features[antiforgeryFeature].Exposed();
+
+            var newAntiforgeryFeature = Activator
+                .CreateInstance(WebFramework.Internals.AntiforgeryFeature)
+                .Exposed();
+
+            newAntiforgeryFeature.HaveDeserializedCookieToken = false;
+            newAntiforgeryFeature.HaveDeserializedRequestToken = false;
+            newAntiforgeryFeature.CookieToken = generatedAntiforgeryFeature.NewCookieToken;
+            newAntiforgeryFeature.RequestToken = generatedAntiforgeryFeature.NewRequestToken;
+
+            httpContext.Features[antiforgeryFeature] = newAntiforgeryFeature.Object;
 
             return actualHttpRequestBuilder;
         }