com.github.tonivade.zeromock.junit4.MockHttpServerRule Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zeromock-junit4 Show documentation

Mock Http Server (mostly) without dependencies

There is a newer version: 1.0

/*
 * Copyright (c) 2018-2020, Antonio Gabriel Muñoz Conejo 
 * Distributed under the terms of the MIT License
 */
package com.github.tonivade.zeromock.junit4;

import org.junit.rules.ExternalResource;

import com.github.tonivade.purefun.Matcher1;
import com.github.tonivade.zeromock.api.HttpRequest;
import com.github.tonivade.zeromock.api.HttpService;
import com.github.tonivade.zeromock.api.HttpService.MappingBuilder;
import com.github.tonivade.zeromock.api.RequestHandler;
import com.github.tonivade.zeromock.client.AsyncHttpClient;
import com.github.tonivade.zeromock.client.HttpClient;
import com.github.tonivade.zeromock.client.IOHttpClient;
import com.github.tonivade.zeromock.client.TaskHttpClient;
import com.github.tonivade.zeromock.client.UIOHttpClient;
import com.github.tonivade.zeromock.server.MockHttpServer;

public class MockHttpServerRule extends ExternalResource {

  private final MockHttpServer server;

  public MockHttpServerRule() {
     this(0);
  }

  public MockHttpServerRule(int port) {
    server = MockHttpServer.listenAt(port);
  }

  @Override
  protected void before() {
    server.start();
  }

  @Override
  protected void after() {
    server.stop();
  }

  public String getBaseUrl() {
    return "http://localhost:" + server.getPort();
  }

  public HttpClient client() {
    return HttpClient.connectTo(getBaseUrl());
  }

  public AsyncHttpClient asyncClient() {
    return AsyncHttpClient.connectTo(getBaseUrl());
  }

  public IOHttpClient ioClient() {
    return IOHttpClient.connectTo(getBaseUrl());
  }

  public UIOHttpClient uioClient() {
    return UIOHttpClient.connectTo(getBaseUrl());
  }

  public TaskHttpClient taskClient() {
    return TaskHttpClient.connectTo(getBaseUrl());
  }

  public MockHttpServerRule verify(Matcher1 matcher) {
    server.verify(matcher);
    return this;
  }

  public MockHttpServerRule verifyNot(Matcher1 matcher) {
    server.verifyNot(matcher);
    return this;
  }

  public MockHttpServerRule addMapping(Matcher1 matcher, RequestHandler handler) {
    server.when(matcher).then(handler);
    return this;
  }

  public MappingBuilder when(Matcher1 matcher) {
    return new MappingBuilder<>(this::addMapping).when(matcher);
  }

  public MockHttpServerRule mount(String path, HttpService service) {
    server.mount(path, service);
    return this;
  }
}