From aa8fd48c64b9a5251ba8a1ba2c938377de999428 Mon Sep 17 00:00:00 2001
From: jacalvar <jacalvar@microsoft.com>
Date: Mon, 22 Feb 2016 11:49:15 -0800
Subject: [PATCH] Updated antiforgery ServiceCollectionExtensions

---
 .../ServiceCollectionExtensions.cs            | 29 ++++++++++++-------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs b/src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs
index 4a8fe1cf58..2d5b2b808c 100644
--- a/src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs
+++ b/src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs
@@ -10,9 +10,16 @@ using Microsoft.Extensions.Options;
 
 namespace Microsoft.Extensions.DependencyInjection
 {
+    /// <summary>
+    /// Extension methods for setting up antiforgery services in an <see cref="IServiceCollection" />.
+    /// </summary>
     public static class ServiceCollectionExtensions
     {
-        public static IServiceCollection AddAntiforgery(this IServiceCollection services)
+        /// <summary>
+        /// Adds antiforgery services to the specified <see cref="IServiceCollection" />.
+        /// </summary>
+        /// <param name="services">The <see cref="IServiceCollection" /> to add services to.</param>
+        public static void AddAntiforgery(this IServiceCollection services)
         {
             if (services == null)
             {
@@ -40,27 +47,27 @@ namespace Microsoft.Extensions.DependencyInjection
                 var policy = new AntiforgerySerializationContextPooledObjectPolicy();
                 return provider.Create(policy);
             });
-
-            return services;
         }
 
-        public static IServiceCollection AddAntiforgery(
-            this IServiceCollection services,
-            Action<AntiforgeryOptions> setupAction)
+        /// <summary>
+        /// Adds antiforgery services to the specified <see cref="IServiceCollection" />.
+        /// </summary>
+        /// <param name="services">The <see cref="IServiceCollection" /> to add services to.</param>
+        /// <param name="setupAction">An <see cref="Action{AntiforgeryOptions}"/> to configure the provided <see cref="AntiforgeryOptions"/>.</param>
+        public static void AddAntiforgery(this IServiceCollection services, Action<AntiforgeryOptions> setupAction)
         {
             if (services == null)
             {
                 throw new ArgumentNullException(nameof(services));
             }
 
-            services.AddAntiforgery();
-
-            if (setupAction != null)
+            if (setupAction == null)
             {
-                services.Configure(setupAction);
+                throw new ArgumentNullException(nameof(setupAction));
             }
 
-            return services;
+            services.AddAntiforgery();
+            services.Configure(setupAction);
         }
     }
 }