Class HttpTestServer

server/src/test/java/io/crate/http/HttpTestServer.java:71–216

Http server with configurable request handler.

 * Http server with configurable request handler.
 */
public class HttpTestServer {

    private final int port;
    private final boolean fail;

    private final BiConsumer<HttpRequest, JsonGenerator> requestHandler;
    private static final JsonFactory JSON_FACTORY;

    private Channel channel;
    private MultiThreadIoEventLoopGroup group;

    public List<String> responses = new ArrayList<>();

    static {
        JSON_FACTORY = new JsonFactory();
        JSON_FACTORY.configure(JsonParser.Feature.ALLOW_UNQUOTED_FIELD_NAMES, true);
        JSON_FACTORY.configure(JsonWriteFeature.QUOTE_FIELD_NAMES.mappedFeature(), true);
        JSON_FACTORY.configure(JsonParser.Feature.ALLOW_COMMENTS, true);
    }

    @Nullable
    private final Map<String, String> headers;



    /**
     * @param port the port to listen on
     * @param fail of set to true, the server will emit error responses
     * @param requestHandler is function to transform incoming messages to some custom output, written to a pre-configured json generator.
     * @param headers Http headers which are attached to the response
     * It MUST propagate any exception to the server so that server can set status to 500.
     */
    public HttpTestServer(int port, boolean fail, BiConsumer<HttpRequest, JsonGenerator> requestHandler, @Nullable Map<String, String> headers) {
        this.port = port;
        this.fail = fail;
        this.requestHandler = requestHandler;
        this.headers = headers;
    }

    /**
     * @param port the port to listen on
     * @param fail of set to true, the server will emit error responses
     * @param requestHandler is function to transform incoming messages to some custom output, written to a pre-configured json generator.
     * It MUST propagate any exception to the server so that server can set status to 500.
     */
    public HttpTestServer(int port, boolean fail, BiConsumer<HttpRequest, JsonGenerator> requestHandler) {
        this.port = port;
        this.fail = fail;
        this.requestHandler = requestHandler;
        this.headers = null;
    }

    public void run() throws InterruptedException {
        // Configure the server.
        ServerBootstrap bootstrap = new ServerBootstrap();
        group = new MultiThreadIoEventLoopGroup(NioIoHandler.newFactory());
        bootstrap.group(group);
        bootstrap.channel(NioServerSocketChannel.class);