cucumber-in-springboot_cucumber-type-registry.java

package com.arun.cucumber.hello.bdd;

import static java.util.Locale.ENGLISH;

import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializationFeature;
import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
import cucumber.api.TypeRegistryConfigurer;
import io.cucumber.cucumberexpressions.ParameterByTypeTransformer;
import io.cucumber.datatable.TableCellByTypeTransformer;
import io.cucumber.datatable.TableEntryByTypeTransformer;
import java.lang.reflect.Type;
import java.util.Locale;
import java.util.Map;
import org.springframework.beans.factory.annotation.Configurable;

/**
 * Class used to convert DataTable to Java Object using Jackson (Json library).
 * <p>
 * <b>NOTE:</b> This class assumes that you will use field names in Java POJO as Cucumber DataTable column headers.
 * </p>
 */
@Configurable
public class CucumberTypeRegistryConfigurer implements TypeRegistryConfigurer {

  private ObjectMapper mapper;

  public CucumberTypeRegistryConfigurer() {
    mapper = new ObjectMapper();

    // To serialize and deserialize java.time.LocalDate, LocalDateTime etc.
    mapper.registerModule(new JavaTimeModule());
    mapper.disable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS);

    mapper.enable(DeserializationFeature.READ_UNKNOWN_ENUM_VALUES_AS_NULL);
  }

  @Override
  public Locale locale() {
    return ENGLISH;
  }

  @Override
  public void configureTypeRegistry(cucumber.api.TypeRegistry typeRegistry) {
    Transformer transformer = new Transformer();
    typeRegistry.setDefaultDataTableCellTransformer(transformer);
    typeRegistry.setDefaultDataTableEntryTransformer(transformer);
    typeRegistry.setDefaultParameterTransformer(transformer);
  }

  private class Transformer implements ParameterByTypeTransformer, TableEntryByTypeTransformer,
      TableCellByTypeTransformer {

    @Override
    public Object transform(String s, Type type) {
      return mapper.convertValue(s, mapper.constructType(type));
    }

    @Override
    public <T> T transform(Map<String, String> map, Class<T> aClass,
        TableCellByTypeTransformer tableCellByTypeTransformer) {
      
      return mapper.convertValue(map, aClass);
    }

    @Override
    public <T> T transform(String s, Class<T> aClass) {
      return mapper.convertValue(s, aClass);
    }
  }

}