autofixture.generators.collections.BuiltInCollectionGenerator Maven / Gradle / Ivy

Go to download
package autofixture.generators.collections;

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

import java.util.Collection;

public class BuiltInCollectionGenerator implements InstanceGenerator {

  @Override
  public  boolean appliesTo(final InstanceType clazz) {
    return
        clazz.isAssignableTo(Collection.class)
            || Iterable.class.isAssignableFrom(clazz.getRawType());
  }

  @SuppressWarnings({"unchecked", "rawtypes"})
  @Override
  public  T next(final InstanceType type, final FixtureContract fixture) {
    //TODO make same as array generation
    final Collection collection = type.createCollection(fixture.getRepeatCount());
    final InstanceType nestedGenericType = type.getNestedGenericType1();
    collection.addAll(fixture.createMany(nestedGenericType));
    return (T) collection;
  }

  @Override
  public  T nextEmpty(final InstanceType instanceType, final FixtureContract fixture) {
    final Collection collection = instanceType.createCollection(0);
    return (T) collection;
  }

  @Override
  public void setOmittingAutoProperties(final boolean isOn) {
  }


}