1+ /*
2+ * Licensed to the Apache Software Foundation (ASF) under one or more
3+ * contributor license agreements. See the NOTICE file distributed with
4+ * this work for additional information regarding copyright ownership.
5+ * The ASF licenses this file to You under the Apache License, Version 2.0
6+ * (the "License"); you may not use this file except in compliance with
7+ * the License. You may obtain a copy of the License at
8+ *
9+ * http://www.apache.org/licenses/LICENSE-2.0
10+ *
11+ * Unless required by applicable law or agreed to in writing, software
12+ * distributed under the License is distributed on an "AS IS" BASIS,
13+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14+ * See the License for the specific language governing permissions and
15+ * limitations under the License.
16+ */
17+
18+ package org .apache .commons .csv .issues ;
19+
20+ import static org .junit .jupiter .api .Assertions .assertArrayEquals ;
21+ import static org .junit .jupiter .api .Assertions .assertEquals ;
22+
23+ import java .io .InputStreamReader ;
24+ import java .io .StringReader ;
25+ import java .io .StringWriter ;
26+ import java .util .ArrayList ;
27+ import java .util .Iterator ;
28+ import java .util .List ;
29+
30+ import org .apache .commons .csv .CSVFormat ;
31+ import org .apache .commons .csv .CSVParser ;
32+ import org .apache .commons .csv .CSVPrinter ;
33+ import org .apache .commons .csv .CSVRecord ;
34+ import org .junit .jupiter .api .Test ;
35+
36+ // psql (14.5 (Homebrew))
37+ //
38+ // create table COMMONS_CSV_PSQL_TEST (ID INTEGER, COL1 VARCHAR, COL2 VARCHAR, COL3 VARCHAR, COL4 VARCHAR);
39+ // insert into COMMONS_CSV_PSQL_TEST select 1, 'abc', 'test line 1' || chr(10) || 'test line 2', null, '';
40+ // insert into COMMONS_CSV_PSQL_TEST select 2, 'xyz', '\b:' || chr(8) || ' \n:' || chr(10) || ' \r:' || chr(13), 'a', 'b';
41+ // insert into COMMONS_CSV_PSQL_TEST values (3, 'a', 'b,c,d', '"quoted"', 'e');
42+ // copy COMMONS_CSV_PSQL_TEST TO '/tmp/psql.csv' WITH (FORMAT CSV);
43+ // copy COMMONS_CSV_PSQL_TEST TO '/tmp/psql.tsv';
44+ //
45+ // cat /tmp/psql.csv
46+ // 1,abc,"test line 1
47+ // test line 2",,""
48+ // 2,xyz,"\b:^H \n:
49+ // \r:^M",a,b
50+ // 3,a,"b,c,d","""quoted""",e
51+ //
52+ // cat /tmp/psql.tsv
53+ // 1 abc test line 1\ntest line 2 \N
54+ // 2 xyz \\b:\b \\n:\n \\r:\r a b
55+ // 3 a b,c,d "quoted" e
56+ //
57+ public class JiraCsv290Test {
58+ private void testHelper (String filename , CSVFormat format ) throws Exception {
59+ List <List <String >> content = new ArrayList <>();
60+ try (CSVParser csvParser = CSVParser .parse (new InputStreamReader (
61+ this .getClass ().getResourceAsStream ("/org/apache/commons/csv/CSV-290/" + filename )), format )) {
62+ for (CSVRecord csvRecord : csvParser ) {
63+ List <String > row = new ArrayList <>();
64+ content .add (row );
65+ for (int i = 0 ; i < csvRecord .size (); i ++) {
66+ row .add (csvRecord .get (i ));
67+ }
68+ }
69+ }
70+ assertEquals (3 , content .size ());
71+
72+ assertEquals ("1" , content .get (0 ).get (0 ));
73+ assertEquals ("abc" , content .get (0 ).get (1 ));
74+ assertEquals ("test line 1\n test line 2" , content .get (0 ).get (2 )); // new line
75+ assertEquals (null , content .get (0 ).get (3 )); // null
76+ assertEquals ("" , content .get (0 ).get (4 ));
77+
78+ assertEquals ("2" , content .get (1 ).get (0 ));
79+ assertEquals ("\\ b:\b \\ n:\n \\ r:\r " , content .get (1 ).get (2 )); // \b, \n, \r
80+
81+ assertEquals ("3" , content .get (2 ).get (0 ));
82+ assertEquals ("b,c,d" , content .get (2 ).get (2 )); // value has comma
83+ assertEquals ("\" quoted\" " , content .get (2 ).get (3 )); // quoted
84+ }
85+
86+ @ Test
87+ public void testPostgresqlCsv () throws Exception {
88+ testHelper ("psql.csv" , CSVFormat .POSTGRESQL_CSV );
89+ }
90+
91+ @ Test
92+ public void testPostgresqlText () throws Exception {
93+ testHelper ("psql.tsv" , CSVFormat .POSTGRESQL_TEXT );
94+ }
95+
96+ @ Test
97+ public void testWriteThenRead () throws Exception {
98+ StringWriter sw = new StringWriter ();
99+
100+ CSVPrinter printer = new CSVPrinter (sw ,
101+ CSVFormat .POSTGRESQL_CSV .builder ().setHeader ().setSkipHeaderRecord (true ).build ());
102+
103+ printer .printRecord ("column1" , "column2" );
104+ printer .printRecord ("v11" , "v12" );
105+ printer .printRecord ("v21" , "v22" );
106+ printer .close ();
107+
108+ CSVParser parser = new CSVParser (new StringReader (sw .toString ()),
109+ CSVFormat .POSTGRESQL_CSV .builder ().setHeader ().setSkipHeaderRecord (true ).build ());
110+
111+ assertArrayEquals (new Object [] { "column1" , "column2" }, parser .getHeaderNames ().toArray ());
112+
113+ Iterator <CSVRecord > i = parser .iterator ();
114+ assertArrayEquals (new String [] { "v11" , "v12" }, i .next ().toList ().toArray ());
115+ assertArrayEquals (new String [] { "v21" , "v22" }, i .next ().toList ().toArray ());
116+ }
117+ }
0 commit comments