From e372410f29bc16b19f0357064ff5c8e7514cc5ff Mon Sep 17 00:00:00 2001
From: John Blum <jblum@pivotal.io>
Date: Fri, 5 Oct 2018 17:53:00 -0700
Subject: [PATCH] Add capability to configure the Spring Boot,
 SpringApplicationBuilder.contextClass(..).

---
 ...ingBootApplicationIntegrationTestsSupport.java | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/integration/SpringBootApplicationIntegrationTestsSupport.java b/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/integration/SpringBootApplicationIntegrationTestsSupport.java
index 3402d5d..90de5c8 100644
--- a/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/integration/SpringBootApplicationIntegrationTestsSupport.java
+++ b/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/integration/SpringBootApplicationIntegrationTestsSupport.java
@@ -51,11 +51,16 @@ public abstract class SpringBootApplicationIntegrationTestsSupport
 	@Override
 	protected ConfigurableApplicationContext newApplicationContext(Class<?>... annotatedClasses) {
 
-		return setApplicationContext(processBeforeRun(processBeforeBuild(new SpringApplicationBuilder(annotatedClasses)
-			.initializers(this::processBeforeRefresh)
-			.web(getWebApplicationType()))
-			.build())
-			.run(getArguments()));
+		return setApplicationContext(processBeforeRun(processBeforeBuild(contextClass(
+			new SpringApplicationBuilder(annotatedClasses)
+				.initializers(this::processBeforeRefresh)
+				.web(getWebApplicationType())))
+				.build())
+				.run(getArguments()));
+	}
+
+	protected SpringApplicationBuilder contextClass(SpringApplicationBuilder builder) {
+		return builder;
 	}
 
 	protected SpringApplicationBuilder processBeforeBuild(SpringApplicationBuilder springApplicationBuilder) {