OptionalIntSubject.java

/*
 * Copyright (c) 2016 Google, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.google.common.truth;

import java.util.OptionalInt;
import javax.annotation.Nullable;

/**
 * Propositions for Java 8 {@link OptionalInt} subjects.
 *
 * @author Ben Douglass
 */
public final class OptionalIntSubject extends Subject<OptionalIntSubject, OptionalInt> {
  OptionalIntSubject(FailureStrategy failureStrategy, @Nullable OptionalInt subject) {
    super(failureStrategy, subject);
  }

  /** Fails if the {@link OptionalInt} is empty or the subject is null. */
  public void isPresent() {
    if (actual() == null || !actual().isPresent()) {
      failWithoutActual("is present");
    }
  }

  /** Fails if the {@link OptionalInt} is present or the subject is null. */
  public void isEmpty() {
    if (actual() == null || actual().isPresent()) {
      fail("is empty");
    }
  }

  /**
   * Fails if the {@link OptionalInt} does not have the given value or the subject is null. More
   * sophisticated comparisons can be done using {@link #hasValueThat()}.
   */
  public void hasValue(int expected) {
    if (actual() == null || !actual().isPresent()) {
      fail("has value", expected);
    } else {
      int actual = actual().getAsInt();
      if (actual != expected) {
        fail("has value", expected);
      }
    }
  }

  /**
   * Prepares for a check regarding the value contained within the {@link OptionalInt}. Fails
   * immediately if the subject is empty.
   */
  public IntegerSubject hasValueThat() {
    if (actual() == null || !actual().isPresent()) {
      failWithoutActual("is present");
      return ignoreCheck().that(0);
    } else {
      return check().that(actual().getAsInt());
    }
  }

  private static final SubjectFactory<OptionalIntSubject, OptionalInt> FACTORY =
      new SubjectFactory<OptionalIntSubject, OptionalInt>() {
        @Override
        public OptionalIntSubject getSubject(FailureStrategy fs, OptionalInt target) {
          return new OptionalIntSubject(fs, target);
        }
      };

  public static SubjectFactory<OptionalIntSubject, OptionalInt> optionalInts() {
    return FACTORY;
  }
}