autofixture.generators.OptionalsGenerator Maven / Gradle / Ivy

Go to download
package autofixture.generators;

import autofixture.exceptions.ObjectCreationException;
import autofixture.interfaces.FixtureContract;
import autofixture.interfaces.InstanceGenerator;
import autofixture.interfaces.InstanceType;

import java.util.Optional;
import java.util.OptionalDouble;
import java.util.OptionalInt;
import java.util.OptionalLong;

/**
 * Created by astral on 02.10.14.
 */
public class OptionalsGenerator implements InstanceGenerator {
  @Override
  public  boolean appliesTo(final InstanceType instanceType) {
    return instanceType.isRawTypeAssignableFrom(Optional.class)
        || instanceType.isRawTypeAssignableFrom(OptionalInt.class)
        || instanceType.isRawTypeAssignableFrom(OptionalDouble.class)
        || instanceType.isRawTypeAssignableFrom(OptionalLong.class);

  }

  @SuppressWarnings("unchecked")
  @Override
  public  T next(final InstanceType instanceType, final FixtureContract fixture) {
    if (instanceType.isRawTypeAssignableFrom(Optional.class)) {
      return (T) Optional.of(fixture.create(instanceType.getNestedGenericType1()));
    } else if (instanceType.isRawTypeAssignableFrom(OptionalInt.class)) {
      return (T) OptionalInt.of(fixture.create(Integer.class));
    } else if (instanceType.isRawTypeAssignableFrom(OptionalLong.class)) {
      return (T) OptionalLong.of(fixture.create(Long.class));
    } else if (instanceType.isRawTypeAssignableFrom(OptionalDouble.class)) {
      return (T) OptionalDouble.of(fixture.create(Double.class));
    } else {
      throw new ObjectCreationException(instanceType);
    }
  }

  @Override
  public  T nextEmpty(final InstanceType instanceType, final FixtureContract fixture) {
    return (T) Optional.empty();
  }

  @Override
  public void setOmittingAutoProperties(final boolean isOn) {

  }
}