Fix error filters not being activated (#2774)

ChilliCream · Dec 23, 2020 · da0f109 · da0f109
1 parent 71c9064
commit da0f109
Show file tree

Hide file tree

Showing 5 changed files with 90 additions and 3 deletions.
diff --git a/...te/Core/src/Execution/DependencyInjection/RequestExecutorBuilderExtensions.ErrorFilter.cs b/...te/Core/src/Execution/DependencyInjection/RequestExecutorBuilderExtensions.ErrorFilter.cs
@@ -42,7 +42,7 @@ public static IRequestExecutorBuilder AddErrorFilter<T>(
                 throw new ArgumentNullException(nameof(factory));
             }
 
-            return builder.ConfigureSchemaServices(s => s.AddSingleton(factory));
+            return builder.ConfigureSchemaServices(s => s.AddSingleton<IErrorFilter, T>(factory));
         }
 
         public static IRequestExecutorBuilder AddErrorFilter<T>(

diff --git a/src/HotChocolate/Core/test/Execution.Tests/Errors/ErrorHandlerTests.cs b/src/HotChocolate/Core/test/Execution.Tests/Errors/ErrorHandlerTests.cs
@@ -41,7 +41,7 @@ await ExpectError(
                         }
                         return error;
                     }),
-                
+
                 expectedErrorCount: 2);
         }
 
@@ -58,6 +58,42 @@ await ExpectError(
                     .AddErrorFilter<DummyErrorFilter>());
         }
 
+        [Fact]
+        public async Task AddClassErrorFilter_SchemaBuiltViaServiceExtensions_ErrorFilterWorks()
+        {
+            // arrange
+            var serviceCollection = new ServiceCollection();
+            var schema = await serviceCollection
+                .AddGraphQLServer()
+                .AddErrorFilter<DummyErrorFilter>()
+                .AddQueryType<Query>()
+                .BuildRequestExecutorAsync();
+
+            // act
+            var resp = await schema.ExecuteAsync("{ foo }");
+
+            // assert
+            resp.MatchSnapshot();
+        }
+
+        [Fact]
+        public async Task AddClassErrorFilterUsingFactory_SchemaBuiltViaServiceExtensions_ErrorFilterWorks()
+        {
+            // arrange
+            var serviceCollection = new ServiceCollection();
+            var schema = await serviceCollection
+                .AddGraphQLServer()
+                .AddErrorFilter(f => new DummyErrorFilter())
+                .AddQueryType<Query>()
+                .BuildRequestExecutorAsync();
+
+            // act
+            var resp = await schema.ExecuteAsync("{ foo }");
+
+            // assert
+            resp.MatchSnapshot();
+        }
+
         [Fact]
         public async Task AddClassErrorFilterWithFactory()
         {
@@ -80,7 +116,7 @@ private async Task ExpectError(
 
             await TestHelper.ExpectError(
                 query,
-                b => 
+                b =>
                 {
                     configure(b);
                     b.AddErrorFilter(error =>
@@ -101,5 +137,10 @@ public IError OnError(IError error)
                 return error.WithCode("Foo123");
             }
         }
+
+        public class Query
+        {
+            public string GetFoo() => throw new Exception("FooError");
+        }
     }
 }
diff --git a/...sts.AddClassErrorFilterUsingFactory_SchemaBuiltViaServiceExtensions_ErrorFilterWorks.snap b/...sts.AddClassErrorFilterUsingFactory_SchemaBuiltViaServiceExtensions_ErrorFilterWorks.snap
@@ -0,0 +1,22 @@
+{
+  "errors": [
+    {
+      "message": "Unexpected Execution Error",
+      "locations": [
+        {
+          "line": 1,
+          "column": 3
+        }
+      ],
+      "path": [
+        "foo"
+      ],
+      "extensions": {
+        "code": "Foo123"
+      }
+    }
+  ],
+  "data": {
+    "foo": null
+  }
+}
diff --git a/...rorHandlerTests.AddClassErrorFilter_SchemaBuiltViaServiceExtensions_ErrorFilterWorks.snap b/...rorHandlerTests.AddClassErrorFilter_SchemaBuiltViaServiceExtensions_ErrorFilterWorks.snap
@@ -0,0 +1,22 @@
+{
+  "errors": [
+    {
+      "message": "Unexpected Execution Error",
+      "locations": [
+        {
+          "line": 1,
+          "column": 3
+        }
+      ],
+      "path": [
+        "foo"
+      ],
+      "extensions": {
+        "code": "Foo123"
+      }
+    }
+  ],
+  "data": {
+    "foo": null
+  }
+}
diff --git a/src/HotChocolate/Core/test/Execution.Tests/HotChocolate.Execution.Tests.csproj b/src/HotChocolate/Core/test/Execution.Tests/HotChocolate.Execution.Tests.csproj
@@ -3,9 +3,11 @@
   <PropertyGroup>
     <AssemblyName>HotChocolate.Execution.Tests</AssemblyName>
     <RootNamespace>HotChocolate.Execution</RootNamespace>
+    <TargetFrameworks>net5.0</TargetFrameworks>
   </PropertyGroup>
 
   <ItemGroup>
+    <ProjectReference Include="..\..\..\AspNetCore\src\AspNetCore\HotChocolate.AspNetCore.csproj" />
     <ProjectReference Include="..\..\src\Execution\HotChocolate.Execution.csproj" />
     <ProjectReference Include="..\..\src\Subscriptions.InMemory\HotChocolate.Subscriptions.InMemory.csproj" />
     <ProjectReference Include="..\StarWars\HotChocolate.StarWars.csproj" />