hope.kola.contract.api.QueryParameters Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of it-kola-contract Show documentation

null

There is a newer version: 1.1.2-RELEASE

package hope.kola.contract.api;

import static hope.kola.contract.api.KolaTestFunctions.$VString;

import com.squareup.javapoet.CodeBlock;
import groovy.lang.Closure;
import groovy.lang.DelegatesTo;
import java.util.LinkedHashSet;
import java.util.Set;
import java.util.function.Consumer;

public class QueryParameters {

  protected Set entries = new LinkedHashSet<>();

  public Set entries() {
    return entries;
  }

  public boolean hasEntries() {
    return !entries.isEmpty();
  }

  public void parameter(String name, @DelegatesTo(QueryParameter.class) Closure consumer) {
    final QueryParameter qp = new QueryParameter(name);
    consumer.setDelegate(qp);
    consumer.call();
    entries.add(qp);
  }

  /**
   * Populate dynamic value from the global context by the key
   *
   * @param name the query parameter name
   * @param key the key from the context
   */
  public void fromGlobal(String name, final String key) {
    final QueryParameter qp = new QueryParameter(name);
    qp.code(CodeBlock.builder().add("$L($S)", $VString, key));
    entries.add(qp);
  }

  public void parameter(final String name, final Consumer consumer) {
    final QueryParameter qp = new QueryParameter(name);
    consumer.accept(qp);
    entries.add(qp);
  }

  public void parameter(String name, String value) {
    final QueryParameter qp = new QueryParameter(name);
    qp.withString(value);
    entries.add(qp);
  }
}