com.noenv.wiremongo.mapping.distinct.Distinct Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertx-wiremongo Show documentation

Lightweight mongo mocking for Vert.x

There is a newer version: 4.5.10

package com.noenv.wiremongo.mapping.distinct;

import com.noenv.wiremongo.command.Command;
import com.noenv.wiremongo.command.distinct.DistinctCommand;
import com.noenv.wiremongo.mapping.collection.WithCollection;
import com.noenv.wiremongo.matching.JsonMatcher;
import com.noenv.wiremongo.matching.Matcher;
import io.vertx.core.json.JsonArray;
import io.vertx.core.json.JsonObject;
import io.vertx.ext.mongo.DistinctOptions;

import static com.noenv.wiremongo.matching.EqualsMatcher.equalTo;

public class Distinct extends WithCollection {

  private Matcher options;
  private Matcher fieldName;
  private Matcher resultClassname;

  public Distinct() {
    super("distinct");
  }

  public Distinct(JsonObject json) {
    super(json);
    fieldName = Matcher.create(json.getJsonObject("fieldName"));
    resultClassname = Matcher.create(json.getJsonObject("resultClassname"));
    options = Matcher.create(json.getJsonObject("options"), j -> new DistinctOptions((JsonObject) j), DistinctOptions::toJson);
  }

  @Override
  public Distinct returns(final JsonArray response) {
    return stub(c -> null == response ? null : response.copy());
  }

  @Override
  protected JsonArray parseResponse(Object jsonValue) {
    return (JsonArray) jsonValue;
  }

  @Override
  public boolean matches(Command cmd) {
    if (!super.matches(cmd)) {
      return false;
    }
    if (!(cmd instanceof DistinctCommand)) {
      return false;
    }
    DistinctCommand c = (DistinctCommand) cmd;
    return (fieldName == null || fieldName.matches(c.getFieldName()))
      && (resultClassname == null || resultClassname.matches(c.getResultClassname()))
      && (options == null || options.matches(c.getOptions()));
  }

  public Distinct withFieldName(String fieldName) {
    return withFieldName(equalTo(fieldName));
  }

  public Distinct withFieldName(Matcher fieldName) {
    this.fieldName = fieldName;
    return self();
  }

  public Distinct withResultClassname(String resultClassname) {
    return withResultClassname(equalTo(resultClassname));
  }

  public Distinct withResultClassname(Matcher resultClassname) {
    this.resultClassname = resultClassname;
    return self();
  }

  public Distinct withOptions(DistinctOptions options) {
    return withOptions(JsonMatcher.equalToJson(options.toJson(), DistinctOptions::toJson));
  }

  public Distinct withOptions(Matcher options) {
    this.options = options;
    return self();
  }
}