org.webpieces.webserver.test.MockChannelManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http-webserver-test Show documentation
The full webpieces server AS A library
The newest version!
package org.webpieces.webserver.test;

import org.webpieces.nio.api.ChannelManager;
import org.webpieces.nio.api.SSLEngineFactory;
import org.webpieces.nio.api.channels.*;
import org.webpieces.nio.api.handlers.ConnectionListener;
import org.webpieces.nio.api.handlers.DataListener;
import org.webpieces.nio.api.handlers.DatagramListener;

import javax.net.ssl.SSLEngine;
import java.util.concurrent.CompletableFuture;

public class MockChannelManager implements ChannelManager {

	private ConnectionListener httpConnectionListener;
	private ConnectionListener httpsConnectionListener;
	private ConnectionListener backendConnectionListener;

	@Override
	public TCPServerChannel createTCPServerChannel(String id, ConnectionListener connectionListener) {
		if(id.startsWith("backend")) {
			backendConnectionListener = connectionListener;
			return new MockServerChannel();
		}
		httpConnectionListener = connectionListener;
		return new MockServerChannel();
	}

	@Override
	public TCPServerChannel createTCPServerChannel(String id, ConnectionListener connectionListener,
			SSLEngineFactory factory) {
		if(id.startsWith("backend")) {
			backendConnectionListener = connectionListener;
			return new MockServerChannel();
		}
		httpsConnectionListener = connectionListener;
		return new MockServerChannel();
	}

	public CompletableFuture simulateHttpsConnect(Channel channel) {
		return httpsConnectionListener.connected(channel, true);
	}
	
	public CompletableFuture simulateHttpConnect(Channel channel) {
		return httpConnectionListener.connected(channel, true);
	}	
	
	@Override
	public TCPChannel createTCPChannel(String id) {
		return null;
	}

	@Override
	public TCPChannel createTCPChannel(String id, SSLEngine engine) {
		return null;
	}

	@Override
	public UDPChannel createUDPChannel(String id) {
		return null;
	}

	@Override
	public DatagramChannel createDatagramChannel(String id, int bufferSize, DatagramListener listener) {
		return null;
	}

	@Override
	public void stop() {
	}

	public ConnectionListener getHttpConnection() {
		return httpConnectionListener;
	}

	public ConnectionListener getHttpsConnection() {
		return httpsConnectionListener;
	}
	
	public ConnectionListener getBackendConnection() {
		return backendConnectionListener;
	}

	@Override
	public String getName() {
		return "mockChanMgr2";
	}

	@Override
	public TCPServerChannel createTCPUpgradableChannel(String id, ConnectionListener connectionListener,
			SSLEngineFactory factory) {
		throw new UnsupportedOperationException("not implemented but could easily be with a one liner");
	}
}