Added template methods in JettyHttpServer to allow decoration of the context by subclasses

Author: tomakehurst

src/main/java/com/github/tomakehurst/wiremock/jetty/JettyHttpServer.java

@@ -271,6 +271,8 @@ private ServletContextHandler addMockServiceContext(
  Notifier notifier) {
  ServletContextHandler mockServiceContext = new ServletContextHandler(jettyServer, "/");
 
+ decorateMockServiceContextBeforeConfig(mockServiceContext);
+
  mockServiceContext.setInitParameter("org.eclipse.jetty.servlet.Default.maxCacheSize", "0");
  mockServiceContext.setInitParameter(
  "org.eclipse.jetty.servlet.Default.resourceBase", fileSource.getPath());
@@ -326,13 +328,20 @@ private ServletContextHandler addMockServiceContext(
  addCorsFilter(mockServiceContext);
  }
 
+ decorateMockServiceContextAfterConfig(mockServiceContext);
+
  return mockServiceContext;
  }
 
+ protected void decorateMockServiceContextBeforeConfig(ServletContextHandler mockServiceContext) {}
+ protected void decorateMockServiceContextAfterConfig(ServletContextHandler mockServiceContext) {}
+
  private ServletContextHandler addAdminContext(
  AdminRequestHandler adminRequestHandler, Notifier notifier) {
  ServletContextHandler adminContext = new ServletContextHandler(jettyServer, ADMIN_CONTEXT_ROOT);
 
+ decorateAdminServiceContextBeforeConfig(adminContext);
+
  adminContext.setInitParameter("org.eclipse.jetty.servlet.Default.maxCacheSize", "0");
 
  String javaVendor = System.getProperty("java.vendor");
@@ -371,9 +380,14 @@ private ServletContextHandler addAdminContext(
 
  addCorsFilter(adminContext);
 
+ decorateAdminServiceContextAfterConfig(adminContext);
+
  return adminContext;
  }
 
+ protected void decorateAdminServiceContextBeforeConfig(ServletContextHandler adminServiceContext) {}
+ protected void decorateAdminServiceContextAfterConfig(ServletContextHandler adminServiceContext) {}
+
  private void addCorsFilter(ServletContextHandler context) {
  context.addFilter(buildCorsFilter(), "/*", EnumSet.of(DispatcherType.REQUEST));
  }