From 52572c91b008d14c4980e7eca0d749ca8aa10ec6 Mon Sep 17 00:00:00 2001
From: Rob Winch <rwinch@users.noreply.github.com>
Date: Tue, 22 Jun 2021 15:33:04 -0500
Subject: [PATCH] 
 SecurityDataFetcherExceptionResolver.setAuthenticationTrustResolver

---
 .../graphql/SecurityDataFetcherExceptionResolver.java       | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java b/samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java
index e9ad6115..b57ee5a3 100644
--- a/samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java
+++ b/samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java
@@ -6,6 +6,7 @@ import java.util.List;
 import graphql.GraphQLError;
 import graphql.GraphqlErrorBuilder;
 import graphql.schema.DataFetchingEnvironment;
+import org.springframework.util.Assert;
 import reactor.core.publisher.Mono;
 
 import org.springframework.graphql.execution.DataFetcherExceptionResolver;
@@ -40,6 +41,11 @@ public class SecurityDataFetcherExceptionResolver implements DataFetcherExceptio
 		return Mono.empty();
 	}
 
+	public void setAuthenticationTrustResolver(AuthenticationTrustResolver authenticationTrustResolver) {
+		Assert.notNull(authenticationTrustResolver, "authenticationTrustResolver cannot be null");
+		this.authenticationTrustResolver = authenticationTrustResolver;
+	}
+
 	private Mono<List<GraphQLError>> unauthorized(DataFetchingEnvironment environment) {
 		return Mono.fromCallable(() -> Arrays.asList(
 				GraphqlErrorBuilder.newError(environment)