fixed issue #416

uniVocity · Oct 12, 2020 · 91e2c7f · 91e2c7f
1 parent e09114c
commit 91e2c7f
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 0 deletions.
diff --git a/src/main/java/com/univocity/parsers/common/DefaultContext.java b/src/main/java/com/univocity/parsers/common/DefaultContext.java
@@ -54,6  54,9 @@ public String[] headers() {
 	}
 
 	public String[] selectedHeaders() {
 		if(headers == null) {
 			headers();
 		}
 		int[] extractedFieldIndexes = extractedFieldIndexes();
 		if (extractedFieldIndexes != null) {
 			String[] extractedFields = new String[extractedFieldIndexes.length];

diff --git a/src/test/java/com/univocity/parsers/issues/github/Github_415.java b/src/test/java/com/univocity/parsers/issues/github/Github_415.java
@@ -41,6  41,7 @@ public void detectedFormatTest() {
 		StringReader stringReader = new StringReader(lines);
 
 		CsvParserSettings parserSettings = new CsvParserSettings();
 		parserSettings.getFormat().setLineSeparator("\n");
 		parserSettings.setDelimiterDetectionEnabled(true, '|', '\t');
 
 		CsvParser csvParser = new CsvParser(parserSettings);

diff --git a/src/test/java/com/univocity/parsers/issues/github/Github_416.java b/src/test/java/com/univocity/parsers/issues/github/Github_416.java
@@ -0,0  1,30 @@
 package com.univocity.parsers.issues.github;
 
 import com.univocity.parsers.csv.*;
 import org.testng.annotations.*;
 
 import java.io.*;
 
 import static org.testng.Assert.*;
 
 
 /**
  * From: https://github.com/univocity/univocity-parsers/issues/416
  *
  * @author Univocity Software Pty Ltd - <a href="mailto:[email protected]">[email protected]</a>
  */
 public class Github_416 {
 	@Test
 	public void testSelectedHeaders() {
 		CsvParserSettings s = new CsvParserSettings();
 		s.getFormat().setLineSeparator("\n");
 		s.setHeaderExtractionEnabled(true);
 		s.selectFields("a", "c", "e");
 
 		CsvParser p = new CsvParser(s);
 		p.beginParsing(new StringReader("a,b,c,d,e\n1,2,3,4,5"));
 		assertEquals(p.getContext().selectedHeaders(), new String[]{"a", "c", "e"});
 		assertEquals(p.getContext().parsedHeaders(), new String[]{"a", "b", "c", "d", "e"});
 		assertEquals(p.parseNext(), new String[]{"1", "3", "5"});
 	}
 }