com.noenv.wiremongo.mapping.update.WithUpdate 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.update;

import com.mongodb.MongoCommandException;
import com.mongodb.ServerAddress;
import com.noenv.wiremongo.command.Command;
import com.noenv.wiremongo.command.update.WithUpdateCommand;
import com.noenv.wiremongo.mapping.WithQuery;
import com.noenv.wiremongo.matching.Matcher;
import io.vertx.core.json.JsonObject;
import org.bson.BsonDocument;
import org.bson.BsonElement;
import org.bson.BsonInt32;
import org.bson.BsonString;

import java.util.Arrays;

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

public abstract class WithUpdate, C extends WithUpdate> extends WithQuery {

  private Matcher update;

  public WithUpdate(String method) {
    super(method);
  }

  public WithUpdate(JsonObject json) {
    super(json);
    update = Matcher.create(json.getJsonObject("update"));
  }

  @Override
  public boolean matches(Command cmd) {
    if (!super.matches(cmd)) {
      return false;
    }
    if (!(cmd instanceof WithUpdateCommand)) {
      return false;
    }
    WithUpdateCommand c = (WithUpdateCommand) cmd;
    return update == null || update.matches(c.getUpdate());
  }

  public C withUpdate(V update) {
    return withUpdate(equalTo(update));
  }

  public C withUpdate(Matcher update) {
    this.update = update;
    return self();
  }

  @Override
  public C returnsDuplicateKeyError() {
    BsonDocument bsonDocument = new BsonDocument(
      Arrays.asList(
        new BsonElement("code", new BsonInt32(11000)),
        new BsonElement("codeName", new BsonString("DuplicateKey")),
        new BsonElement("errmsg", new BsonString("E11000 duplicate key error"))
      )
    );
    return returnsError(new MongoCommandException(bsonDocument, new ServerAddress()));
  }
}