Isolate inner jetty server component from wrapper with filters

1 files changed, 6 insertions, 0 deletions

diff --git a/jdisc_core/src/main/java/com/yahoo/jdisc/test/ServerProviderConformanceTest.java b/jdisc_core/src/main/java/com/yahoo/jdisc/test/ServerProviderConformanceTest.java
index 1cf4e3dd858..e30bb927204 100644
--- a/jdisc_core/src/main/java/com/yahoo/jdisc/test/ServerProviderConformanceTest.java
+++ b/jdisc_core/src/main/java/com/yahoo/jdisc/test/ServerProviderConformanceTest.java
@@ -11,6 +11,7 @@ import com.yahoo.jdisc.Request;
 import com.yahoo.jdisc.Response;
 import com.yahoo.jdisc.application.BindingSetSelector;
 import com.yahoo.jdisc.application.ContainerBuilder;
+import com.yahoo.jdisc.application.GuiceRepository;
 import com.yahoo.jdisc.handler.AbstractRequestHandler;
 import com.yahoo.jdisc.handler.CompletionHandler;
 import com.yahoo.jdisc.handler.ContentChannel;
@@ -75,6 +76,9 @@ public abstract class ServerProviderConformanceTest {
         Iterable<ByteBuffer> newResponseContent();
 
         void validateResponse(V response) throws Throwable;
+
+        default AutoCloseable configureServerProvider(GuiceRepository guice) { return () -> { }; }
+
     }
 
     /**
@@ -2784,6 +2788,7 @@ public abstract class ServerProviderConformanceTest {
         builder.serverBindings().bind(builder.getInstance(Key.get(String.class, Names.named("serverBinding"))),
                                       requestHandler);
         final T serverProvider = builder.guiceModules().getInstance(adapter.getServerProviderClass());
+        AutoCloseable scaffolding = adapter.configureServerProvider(builder.guiceModules());
         builder.serverProviders().install(serverProvider);
         if (builder.getInstance(Key.get(Boolean.class, Names.named("activateContainer")))) {
             driver.activateContainer(builder);
@@ -2804,6 +2809,7 @@ public abstract class ServerProviderConformanceTest {
             requestHandler.awaitAsyncTasks();
         }
 
+        scaffolding.close();
         serverProvider.close();
         driver.close();
     }