io.ebean.WhenBeanReturn Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ebean-mocker Show documentation

Ability to set a Mockito mock EbeanServer as 'default' EbeanServer

There is a newer version: 13.7.0

package io.ebean;

/**
 * Created by rob on 16/07/15.
 */
public class WhenBeanReturn {

  Class beanType;

  Object id;

  T bean;

  public WhenBeanReturn(Class beanType) {
    this(beanType, null);
  }

  public WhenBeanReturn(Class beanType, Object id) {
    this.beanType = beanType;
    this.id = id;
  }

  public void thenReturn(T bean) {
    this.bean = bean;
  }

  T val() {
    return bean;
  }

  /**
   * Return true if matched by beanType and there is no specific id set.
   */
  boolean isMatch(Class beanType) {
    return beanType.equals(this.beanType) && this.id == null;
  }

  /**
   * Return true if matched by beanType and id value.
   */
  boolean isMatch(Class beanType, Object id) {

    return beanType.equals(this.beanType) && idMatch(id);
  }

  boolean idMatch(Object id) {
    if (id == null) return this.id == null;
    return id.equals(this.id);
  }
}