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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of undertow Show documentation
The newest version!
package com.github.skjolber.mockito.rest.spring;

import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import jakarta.servlet.ServletException;
import org.springframework.web.context.ContextLoaderListener;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.servlet.DispatcherServlet;

import io.undertow.Handlers;
import io.undertow.Undertow;
import io.undertow.server.handlers.PathHandler;
import io.undertow.servlet.Servlets;
import io.undertow.servlet.api.DeploymentInfo;
import io.undertow.servlet.api.DeploymentManager;
import io.undertow.servlet.api.InstanceFactory;
import io.undertow.servlet.api.ListenerInfo;
import io.undertow.servlet.api.ServletInfo;
import io.undertow.servlet.util.ImmediateInstanceFactory;

public class UndertowMockitoEndpointServerInstance implements MockitoEndpointServerInstance {

	private static final String SPRING_DISPATCHER_MAPPING_URL = "/*";
	private static final String CONTEXT_PATH = "/";

	protected boolean started = true;
	protected List servers = new ArrayList<>();

	/**
	 * 
	 * Destroy endpoints.
	 * 
	 */

	public void destroy() throws Exception {
		synchronized (this) {
			started = false;

			for (Undertow server : servers) {
				server.stop();
			}

			servers.clear();
		}
	}
	
	/**
	 * 
	 * Stop endpoints.
	 * 
	 */

	public void stop() throws Exception {
		synchronized (this) {
			if(started) {
				started = false;
	
				for (Undertow server : servers) {
					server.stop();
				}
			}
		}
	}

	/**
	 * 
	 * (Re)start endpoints.
	 * 
	 */

	public void start() throws Exception {
		synchronized (this) {
			if(!started) {
				started = true;
				for (Undertow server : servers) {
					server.start();
				}
			}
		}
	}

	public Map, Object> add(List> mockTargetBeans, List> defaultContextBeans, URL url) throws Exception {
		MockitoSpringApplicationListener configuration = new MockitoSpringApplicationListener();
		configuration.setContextBeans(defaultContextBeans);
		configuration.setMockTargetBeans(mockTargetBeans);

		MockitoSpringWebApplicationContext dispatcherContext = new MockitoSpringWebApplicationContext(mockTargetBeans);

		// web config must be loaded after beans
		for(Class bean : defaultContextBeans) {
			dispatcherContext.register(bean);
		}

		dispatcherContext.addApplicationListener(configuration);

		Undertow undertow = configureUndertow(dispatcherContext, url);

		servers.add(undertow);

		undertow.start();

		return configuration.getAll();
	}

	private Undertow configureUndertow(MockitoSpringWebApplicationContext context, URL url) throws ServletException {
		// https://github.com/yarosla/spring-undertow/blob/master/src/main/java/ys/undertow/UndertowMain.java
		DeploymentInfo servletBuilder = Servlets.deployment()
				.setClassLoader(Undertow.class.getClassLoader())
				.setContextPath(url.getPath())
				.setDeploymentName("mock")
				.addServlet(createDispatcherServlet(context))
				.addListener(createContextLoaderListener(context));

		DeploymentManager manager = Servlets.defaultContainer().addDeployment(servletBuilder);
		manager.deploy();

		PathHandler path = Handlers.path(Handlers.redirect("/"))
				.addPrefixPath(CONTEXT_PATH, manager.start());

		return Undertow.builder()
				.addHttpListener(url.getPort(), url.getHost())
				.setHandler(path)
				.build();
	}

	private static ListenerInfo createContextLoaderListener(WebApplicationContext context) {
		InstanceFactory factory = new ImmediateInstanceFactory<>(new ContextLoaderListener(context));
		return new ListenerInfo(ContextLoaderListener.class, factory);
	}

	private static ServletInfo createDispatcherServlet(WebApplicationContext context) {
		InstanceFactory factory = new ImmediateInstanceFactory<>(new DispatcherServlet(context));
		return Servlets.servlet("DispatcherServlet", DispatcherServlet.class, factory)
				.addMapping(SPRING_DISPATCHER_MAPPING_URL)
				.setLoadOnStartup(1);
	}	
}