com.github.skjolber.mockito.rest.spring.MockitoSpringContainerStarter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of mockito-spring-rest Show documentation

REST web-service mocking utility for Spring

There is a newer version: 1.0.1

package com.github.skjolber.mockito.rest.spring;

import java.util.List;

import javax.servlet.ServletRegistration;

import org.eclipse.jetty.server.handler.ContextHandler.Context;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.util.component.AbstractLifeCycle;
import org.eclipse.jetty.webapp.WebAppContext;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ApplicationContextEvent;
import org.springframework.web.context.ContextLoaderListener;
import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
import org.springframework.web.servlet.DispatcherServlet;

public class MockitoSpringContainerStarter extends AbstractLifeCycle implements ServletContextHandler.ServletContainerInitializerCaller {

	private WebAppContext context;
	private List> mockTargetBeans;
	private List> contextBeans;
	private ApplicationListener listener;

	public MockitoSpringContainerStarter(WebAppContext context, List> beans, List> configurationBeans, ApplicationListener listener) {
		this.context = context;
		this.mockTargetBeans = beans;
		this.contextBeans = configurationBeans;
		this.listener = listener;
	}

	public void doStart() {
		Context container = context.getServletContext();
		container.setExtendedListenerTypes(true);
		
		// Create the 'root' Spring application context
		AnnotationConfigWebApplicationContext rootContext = new AnnotationConfigWebApplicationContext();
		// Manage the lifecycle of the root application context
		container.addListener(new ContextLoaderListener(rootContext));

		// Create the dispatcher servlet's Spring application context
		MockitoSpringFactoryWebApplicationContext dispatcherContext = new MockitoSpringFactoryWebApplicationContext(mockTargetBeans);

		// web config must be loaded after beans
		for(Class bean : contextBeans) {
			dispatcherContext.register(bean);
		}
		
		dispatcherContext.addApplicationListener(listener);
		
		// Register and map the dispatcher servlet
		ServletRegistration.Dynamic dispatcher = container.addServlet("dispatcher", new DispatcherServlet(dispatcherContext));
		dispatcher.setLoadOnStartup(1);
		dispatcher.addMapping("/");
	}
	
}