/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
    * with the License.  You may obtain a copy of the License at
    *
   *   https://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  
  package org.apache.commons.csv.issues;
  
  import static org.junit.jupiter.api.Assertions.assertArrayEquals;
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertNull;
  
  import java.io.InputStreamReader;
  import java.io.StringReader;
  import java.io.StringWriter;
  import java.util.ArrayList;
  import java.util.Iterator;
  import java.util.List;
  import java.util.stream.Collectors;
  
  import org.apache.commons.csv.CSVFormat;
  import org.apache.commons.csv.CSVParser;
  import org.apache.commons.csv.CSVPrinter;
  import org.apache.commons.csv.CSVRecord;
  import org.junit.jupiter.api.Test;
  
  // psql (14.5 (Homebrew))
  //
  // create table COMMONS_CSV_PSQL_TEST (ID INTEGER, COL1 VARCHAR, COL2 VARCHAR, COL3 VARCHAR, COL4 VARCHAR);
  // insert into COMMONS_CSV_PSQL_TEST select 1, 'abc', 'test line 1' || chr(10) || 'test line 2', null, '';
  // insert into COMMONS_CSV_PSQL_TEST select 2, 'xyz', '\b:' || chr(8) || ' \t:' || chr(9) || ' \n:' || chr(10) || ' \r:' || chr(13), 'a', 'b';
  // insert into COMMONS_CSV_PSQL_TEST values (3, 'a', 'b,c,d', '"quoted"', 'e');
  // copy COMMONS_CSV_PSQL_TEST TO '/tmp/psql.csv' WITH (FORMAT CSV);
  // copy COMMONS_CSV_PSQL_TEST TO '/tmp/psql.tsv';
  //
  // cat /tmp/psql.csv
  // 1,abc,"test line 1
  // test line 2",,""
  // 2,xyz,"\b:^H \t:         \n:
  //  \r:^M",a,b
  // 3,a,"b,c,d","""quoted""",e
  //
  // cat /tmp/psql.tsv
  // 1    abc    test line 1\ntest line 2                  \N
  // 2    xyz    \\b:\b \\t:\t \\n:\n \\r:\r    a          b
  // 3    a      b,c,d                         "quoted"    e
  //
  class JiraCsv290Test {
  
      private void testHelper(final String fileName, final CSVFormat format) throws Exception {
          List<List<String>> content = new ArrayList<>();
          try (CSVParser csvParser = CSVParser.parse(new InputStreamReader(this.getClass().getResourceAsStream("/org/apache/commons/csv/CSV-290/" + fileName)),
                  format)) {
              content = csvParser.stream().collect(Collectors.mapping(CSVRecord::toList, Collectors.toList()));
          }
  
          assertEquals(3, content.size());
  
          assertEquals("1", content.get(0).get(0));
          assertEquals("abc", content.get(0).get(1));
          assertEquals("test line 1\ntest line 2", content.get(0).get(2)); // new line
          assertNull(content.get(0).get(3)); // null
          assertEquals("", content.get(0).get(4));
  
          assertEquals("2", content.get(1).get(0));
          assertEquals("\\b:\b \\t:\t \\n:\n \\r:\r", content.get(1).get(2)); // \b, \t, \n, \r
  
          assertEquals("3", content.get(2).get(0));
          assertEquals("b,c,d", content.get(2).get(2)); // value has comma
          assertEquals("\"quoted\"", content.get(2).get(3)); // quoted
      }
  
      @Test
      void testPostgresqlCsv() throws Exception {
          testHelper("psql.csv", CSVFormat.POSTGRESQL_CSV);
      }
  
      @Test
      void testPostgresqlText() throws Exception {
          testHelper("psql.tsv", CSVFormat.POSTGRESQL_TEXT);
      }
  
      @Test
      void testWriteThenRead() throws Exception {
          final StringWriter sw = new StringWriter();
          final CSVFormat format = CSVFormat.POSTGRESQL_CSV.builder().setHeader().setSkipHeaderRecord(true).get();
         try (CSVPrinter printer = new CSVPrinter(sw, format)) {
             printer.printRecord("column1", "column2");
             printer.printRecord("v11", "v12");
             printer.printRecord("v21", "v22");
             printer.close();
             try (CSVParser parser = CSVParser.builder().setReader(new StringReader(sw.toString())).setFormat(format).get()) {
                 assertArrayEquals(new Object[] { "column1", "column2" }, parser.getHeaderNames().toArray());
                 final Iterator<CSVRecord> i = parser.iterator();
                 assertArrayEquals(new String[] { "v11", "v12" }, i.next().toList().toArray());
                 assertArrayEquals(new String[] { "v21", "v22" }, i.next().toList().toArray());
             }
         }
     }
 }