play.libs.ws.DummyWSRequest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of framework Show documentation

RePlay is a fork of the Play1 framework, created by Codeborne.

There is a newer version: 2.6.3

package play.libs.ws;

import play.Play;

import java.util.HashMap;
import java.util.Map;

import static java.util.Collections.emptyMap;
import static play.libs.ws.HttpMethod.*;

public class DummyWSRequest extends WSRequest {
  private final Map> requests;
  private final Map> responses;

  public DummyWSRequest(String url, Map> requests, Map> responses) {
    super(url, Play.defaultWebEncoding);
    this.requests = requests;
    this.responses = responses;
  }

  @Override public HttpResponse get() {
    return forMethod(GET);
  }

  @Override public HttpResponse patch() {
    return forMethod(PATCH);
  }

  @Override public HttpResponse post() {
    return forMethod(POST);
  }

  @Override public HttpResponse put() {
    return forMethod(PUT);
  }

  @Override public HttpResponse delete() {
    return forMethod(DELETE);
  }

  @Override public HttpResponse options() {
    return forMethod(OPTIONS);
  }

  @Override public HttpResponse head() {
    return forMethod(HEAD);
  }

  @Override public HttpResponse trace() {
    return forMethod(TRACE);
  }

  private void rememberRequest(HttpMethod get) {
    requests.computeIfAbsent(url, (u) -> new HashMap<>()).put(get, this);
  }

  private Map forURL() {
    return responses.getOrDefault(url, emptyMap());
  }

  private HttpResponse forMethod(HttpMethod method) {
    rememberRequest(method);

    HttpResponse mockedResponse = forURL().get(method);
    if (mockedResponse == null) return notFound(method);
    return mockedResponse;
  }

  private DummyHttpResponse notFound(HttpMethod method) {
    return new DummyHttpResponse(404, "Not found: " + method + " " + url + ", expected responses: " + responses);
  }
}