com.github.tomakehurst.wiremock.servlet.WireMockWebContextListener Maven / Gradle / Ivy
/*
* Copyright (C) 2011 Thomas Akehurst
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.github.tomakehurst.wiremock.servlet;
import com.github.tomakehurst.wiremock.common.Notifier;
import com.github.tomakehurst.wiremock.common.ServletContextFileSource;
import com.github.tomakehurst.wiremock.common.Slf4jNotifier;
import com.github.tomakehurst.wiremock.core.MappingsSaver;
import com.github.tomakehurst.wiremock.core.WireMockApp;
import com.github.tomakehurst.wiremock.extension.ResponseDefinitionTransformer;
import com.github.tomakehurst.wiremock.extension.ResponseTransformer;
import com.github.tomakehurst.wiremock.http.AdminRequestHandler;
import com.github.tomakehurst.wiremock.http.BasicResponseRenderer;
import com.github.tomakehurst.wiremock.http.ProxyResponseRenderer;
import com.github.tomakehurst.wiremock.http.StubRequestHandler;
import com.github.tomakehurst.wiremock.http.StubResponseRenderer;
import com.github.tomakehurst.wiremock.matching.RequestMatcherExtension;
import com.github.tomakehurst.wiremock.standalone.JsonFileMappingsSource;
import com.google.common.base.Optional;
import javax.servlet.ServletContext;
import javax.servlet.ServletContextEvent;
import javax.servlet.ServletContextListener;
import java.util.Collections;
import static com.google.common.base.MoreObjects.firstNonNull;
public class WireMockWebContextListener implements ServletContextListener {
private static final String FILES_ROOT = "__files";
private static final String APP_CONTEXT_KEY = "WireMockApp";
private static final String FILE_SOURCE_ROOT_KEY = "WireMockFileSourceRoot";
@Override
public void contextInitialized(ServletContextEvent sce) {
ServletContext context = sce.getServletContext();
String fileSourceRoot = context.getInitParameter(FILE_SOURCE_ROOT_KEY);
ServletContextFileSource fileSource = new ServletContextFileSource(context, fileSourceRoot);
Optional maxRequestJournalEntries = readMaxRequestJournalEntries(context);
boolean verboseLoggingEnabled = Boolean.parseBoolean(
firstNonNull(context.getInitParameter("verboseLoggingEnabled"), "true"));
JsonFileMappingsSource defaultMappingsLoader = new JsonFileMappingsSource(fileSource.child("mappings"));
MappingsSaver mappingsSaver = new NotImplementedMappingsSaver();
WireMockApp wireMockApp = new WireMockApp(
false,
defaultMappingsLoader,
mappingsSaver,
false,
maxRequestJournalEntries,
Collections.emptyMap(),
Collections.emptyMap(),
fileSource,
new NotImplementedContainer()
);
AdminRequestHandler adminRequestHandler = new AdminRequestHandler(wireMockApp, new BasicResponseRenderer());
StubRequestHandler stubRequestHandler = new StubRequestHandler(wireMockApp,
new StubResponseRenderer(fileSource.child(FILES_ROOT),
wireMockApp.getGlobalSettingsHolder(),
new ProxyResponseRenderer(),
Collections.emptyList()));
context.setAttribute(APP_CONTEXT_KEY, wireMockApp);
context.setAttribute(StubRequestHandler.class.getName(), stubRequestHandler);
context.setAttribute(AdminRequestHandler.class.getName(), adminRequestHandler);
context.setAttribute(Notifier.KEY, new Slf4jNotifier(verboseLoggingEnabled));
}
/**
* @param context Servlet context for parameter reading
* @return Maximum number of entries or absent
*/
private Optional readMaxRequestJournalEntries(ServletContext context) {
String str = context.getInitParameter("maxRequestJournalEntries");
if(str == null) {
return Optional.absent();
}
return Optional.of(Integer.parseInt(str));
}
@Override
public void contextDestroyed(ServletContextEvent sce) {
}
}