jio.test.stub.httpserver.BodyStub Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jio-test Show documentation

JIO test library based on Property Based Testing and Java Flight Recording Debuggers

There is a newer version: 3.0.0-RC2

package jio.test.stub.httpserver;

import fun.gen.Gen;

import java.time.Duration;
import java.util.Objects;
import java.util.function.Supplier;

/**
 * A stub that stands in for the body response of an HTTP request.
 */
public non-sealed interface BodyStub extends HttpRespStub {


  /**
   * Creates a body stub where each call returns the value generated by the given generator.
   *
   * @param gen The generator for generating body values.
   * @return A body stub that returns the body produced by the generator.
   * @throws NullPointerException if the provided generator is null.
   */
  static BodyStub gen(final Gen gen) {
    Supplier supplier = Objects.requireNonNull(gen)
                                       .sample();
    return n -> reqBody -> uri -> headers -> supplier.get();
  }

  /**
   * Creates a body stub that always returns the given body as the HTTP response body.
   *
   * @param body The body to be returned.
   * @return A body stub that returns the specified body.
   * @throws NullPointerException if the provided body is null.
   */
  static BodyStub cons(final String body) {
    Objects.requireNonNull(body);
    return n -> reqBody -> uri -> headers -> body;
  }


  /**
   * Creates a body stub that always returns the given body as the HTTP response body after the specified delay.
   *
   * @param body  The body to be returned.
   * @param delay The delay before returning the body.
   * @return A body stub that returns the specified body after the delay.
   * @throws NullPointerException if the provided body or delay is null.
   */
  static BodyStub consAfter(final String body,
                            final Duration delay
                           ) {
    Objects.requireNonNull(body);
    Objects.requireNonNull(delay);
    return n -> reqBody -> uri -> headers -> {
      try {
        Thread.sleep(delay.toMillis());
      } catch (InterruptedException e) {
        Thread.currentThread()
              .interrupt();
      }
      return body;
    };
  }
}