autofixture.generators.collections.BuiltInMapGenerator 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.Map;

public class BuiltInMapGenerator implements InstanceGenerator {

  @Override
  public  boolean appliesTo(final InstanceType clazz) {
    return clazz.isAssignableTo(Map.class);
  }

  @SuppressWarnings({"unchecked", "rawtypes"})
  @Override
  public  T next(final InstanceType type, final FixtureContract fixture) {
    //TODO make same as array generation
    final Map map = type.createEmptyMap();
    fillWithNestedInstances(map, type, fixture);
    return (T) map;
  }

  @Override
  public  T nextEmpty(final InstanceType instanceType, final FixtureContract fixture) {
    //TODO make same as array generation
    final Map map = instanceType.createEmptyMap();
    return (T) map;

  }

  public  void fillWithNestedInstances(final Map map, final InstanceType type, final FixtureContract fixture) {
    final InstanceType nestedGenericType1 = type.getNestedGenericType1();
    final InstanceType nestedGenericType2 = type.getNestedGenericType2();
    for(int i = 0 ; i < fixture.getRepeatCount() ; ++i) {
      map.put(fixture.create(nestedGenericType1), fixture.create(nestedGenericType2));
    }
  }

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


}