autofixture.generators.vavr.VavrValidationGenerator Maven / Gradle / Ivy

Go to download
package autofixture.generators.vavr;

import autofixture.interfaces.FixtureContract;
import autofixture.interfaces.InstanceGenerator;
import autofixture.interfaces.InstanceType;
import io.vavr.control.Validation;

import java.util.HashMap;
import java.util.Map;

import static io.vavr.control.Validation.invalid;
import static io.vavr.control.Validation.valid;

public class VavrValidationGenerator implements InstanceGenerator {

  private Map lastValidationByType = new HashMap<>();

  @Override
  public  boolean appliesTo(final InstanceType instanceType) {
    return instanceType.isRawTypeAssignableFrom(Validation.class);
  }

  @Override
  public  T next(final InstanceType instanceType, final FixtureContract fixture) {
    if(lastValidationByType.containsKey(instanceType)) {
      Validation validation = lastValidationByType.get(instanceType);
      if(validation.isValid()) {
        validation = invalid(fromFirstGenericOf(instanceType, fixture));
      } else {
        validation = valid(fromSecondGenericOf(instanceType, fixture));
      }
      lastValidationByType.put(instanceType, validation);
      return (T) validation;
    } else {
      Validation validation = valid(fromSecondGenericOf(instanceType, fixture));
      lastValidationByType.put(instanceType, validation);
      return (T) validation;
    }

  }

  @Override
  public  T nextEmpty(final InstanceType instanceType, final FixtureContract fixture) {
    return (T) Validation.valid(fixture.createDummy(instanceType.getNestedGenericType2()));
  }

  @Override
  public void setOmittingAutoProperties(final boolean isOn) {

  }

  private  Object fromSecondGenericOf(final InstanceType instanceType, final FixtureContract fixture) {
    return fixture.create(instanceType.getNestedGenericType2());
  }

  private  Object fromFirstGenericOf(final InstanceType instanceType, final FixtureContract fixture) {
    return fixture.create(instanceType.getNestedGenericType1());
  }
}