package com.opencsv.bean;
   
   import com.opencsv.CSVReader;
   import org.junit.jupiter.api.Test;
   
   import java.io.StringReader;
   import java.util.ArrayList;
   import java.util.HashMap;
   import java.util.List;
  import java.util.Map;
  
  import static org.junit.jupiter.api.Assertions.*;
  
  public class CsvToBeanFilterTest {
  
     private static final String TEST_STRING =
           "FEATURE_NAME,STATE,USER_COUNT\n" +
                 "hello world,production,3228\n" +
                 "calc age,beta,74\n" +
                 "wash dishes,alpha,3";
  
     private static final String TEST_EMPTY_STRING =
             "FEATURE_NAME,STATE,USER_COUNT\n" +
                     "hello world,    ,3228\n" +
                     "calc age,,74\n" +
                     "wash dishes,    ,3";
  
     private CSVReader createReader() {
        StringReader reader = new StringReader(TEST_STRING);
        return new CSVReader(reader);
     }
  
     private MappingStrategy<Feature> createMappingStrategy() {
        HeaderColumnNameTranslateMappingStrategy<Feature> strategy = new HeaderColumnNameTranslateMappingStrategy<>();
        Map<String, String> columnMap = new HashMap<>();
        columnMap.put("FEATURE_NAME", "name");
        columnMap.put("STATE", "state");
        strategy.setColumnMapping(columnMap);
        strategy.setType(Feature.class);
        return strategy;
     }
  
     public static class Feature {
  
        @CsvBindByName(column = "FEATURE_NAME")
        private String name;
  
        @CsvBindByName(required = true)
        private String state;
  
        public void setName(String name) {
           this.name = name;
        }
  
        public void setState(String state) {
           this.state = state;
        }
  
        public String getName() {
           return name;
        }
  
        public String getState() {
           return state;
        }
     }
  
     // This would be better done with the BeanVerifier.
     private static class NonProductionFilter implements CsvToBeanFilter {
  
        @Override
        public boolean allowLine(String[] line) {
           String value = line[1];
           return !"production".equals(value);
        }
  
     }
  
     @Test
     public void testColumnNameTranslationWithLineFiltering() {
        CsvToBean<Feature> csvToBean = new CsvToBeanBuilder<Feature>(createReader())
                .withMappingStrategy(createMappingStrategy())
                .withFilter(new NonProductionFilter())
                .build();
        List<Feature> list = csvToBean.parse();
        assertEquals(2, list.size(), "Parsing resulted in the wrong number of items.");
        assertEquals("calc age", list.get(0).getName(), "The first item has the wrong name.");
        assertEquals("beta", list.get(0).getState(), "The first item has the wrong state.");
        assertEquals("wash dishes", list.get(1).getName(), "The second item has the wrong name.");
        assertEquals("alpha", list.get(1).getState(), "The second item has the wrong state.");
     }
  
     @Test
     public void testColumnNameTranslationWithLineFilteringAndEmptyState() {
        CsvToBean<Feature> csvToBean = new CsvToBeanBuilder<Feature>(new StringReader(TEST_EMPTY_STRING))
                .withMappingStrategy(createMappingStrategy())
                .withFilter(new NonProductionFilter())
                .build();
        List<Feature> list = csvToBean.parse();
       assertEquals("    ", list.get(0).getState());
       assertTrue(list.get(1).getState().isEmpty());
       assertEquals("    ", list.get(2).getState());
    }
 
 
    @Test
    public void testFilterWithParallelParsing() {
       MappingStrategy<Feature> strategy = new HeaderColumnNameMappingStrategy<>();
       strategy.setType(Feature.class);
       List<Feature> list = new CsvToBeanBuilder<Feature>(new StringReader(TEST_STRING))
               .withMappingStrategy(strategy)
               .withFilter(new NonProductionFilter())
               .build().parse();
       assertEquals(2, list.size(), "Parsing resulted in the wrong number of items.");
       assertEquals("calc age", list.get(0).getName(), "The first item has the wrong name.");
       assertEquals("beta", list.get(0).getState(), "The first item has the wrong state.");
       assertEquals("wash dishes", list.get(1).getName(), "The second item has the wrong name.");
       assertEquals("alpha", list.get(1).getState(), "The second item has the wrong state.");
    }
 
    @Test
    public void testFilterWithIteratorParsing() {
       MappingStrategy<Feature> strategy = new HeaderColumnNameMappingStrategy<>();
       strategy.setType(Feature.class);
       CsvToBean<Feature> ctb = new CsvToBeanBuilder<Feature>(new StringReader(TEST_STRING))
               .withMappingStrategy(strategy)
               .withFilter(new NonProductionFilter())
               .build();
       List<Feature> list = new ArrayList<>(2);
       for(Feature f : ctb) { list.add(f); }
       assertEquals(2, list.size(), "Parsing resulted in the wrong number of items.");
       assertEquals("calc age", list.get(0).getName(), "The first item has the wrong name.");
       assertEquals("beta", list.get(0).getState(), "The first item has the wrong state.");
       assertEquals("wash dishes", list.get(1).getName(), "The second item has the wrong name.");
       assertEquals("alpha", list.get(1).getState(), "The second item has the wrong state.");
    }
 }