1 package org
.argeo
.util
;
3 import java
.io
.ByteArrayInputStream
;
4 import java
.io
.ByteArrayOutputStream
;
5 import java
.util
.ArrayList
;
6 import java
.util
.Arrays
;
9 /** {@link CsvWriter} tests. */
10 public class CsvWriterTest
{
11 public void testWrite() throws Exception
{
12 ByteArrayOutputStream out
= new ByteArrayOutputStream();
13 final CsvWriter csvWriter
= new CsvWriter(out
);
15 String
[] header
= { "Header1", "Header 2", "Header,3", "Header\n4", "Header\"5\"" };
16 String
[] line1
= { "Value1", "Value 2", "Value,3", "Value\n4", "Value\"5\"" };
17 csvWriter
.writeLine(Arrays
.asList(header
));
18 csvWriter
.writeLine(Arrays
.asList(line1
));
20 String reference
= "Header1,Header 2,\"Header,3\",\"Header\n4\",\"Header\"\"5\"\"\"\n"
21 + "Value1,Value 2,\"Value,3\",\"Value\n4\",\"Value\"\"5\"\"\"\n";
22 String written
= new String(out
.toByteArray());
23 assert reference
.equals(written
);
25 System
.out
.println(written
);
27 final List
<String
> allTokens
= new ArrayList
<String
>();
28 CsvParser csvParser
= new CsvParser() {
29 protected void processLine(Integer lineNumber
, List
<String
> header
, List
<String
> tokens
) {
31 allTokens
.addAll(header
);
32 allTokens
.addAll(tokens
);
35 ByteArrayInputStream in
= new ByteArrayInputStream(written
.getBytes());
38 List
<String
> allTokensRef
= new ArrayList
<String
>();
39 allTokensRef
.addAll(Arrays
.asList(header
));
40 allTokensRef
.addAll(Arrays
.asList(line1
));
42 assert allTokensRef
.size() == allTokens
.size();
43 for (int i
= 0; i
< allTokensRef
.size(); i
++)
44 assert allTokensRef
.get(i
).equals(allTokens
.get(i
));