Make sure only record separators we can handle are used

git-svn-id: https://svn.apache.org/repos/asf/commons/proper/csv/trunk@1609548 13f79535-47bb-0310-9956-ffa450edef68

Author: britter

src/main/java/org/apache/commons/csv/CSVFormat.java

@@ -701,6 +701,12 @@ private void validate() throws IllegalArgumentException {
             throw new IllegalArgumentException("No quotes mode set but no escape character is set");
         }
 
+        if(recordSeparator != null
+                && !(CRLF.equals(recordSeparator)
+                    || String.valueOf(CR).equals(recordSeparator)
+                    || String.valueOf(LF).equals(recordSeparator))) {
+            throw new IllegalArgumentException("Record separator can only by CR, LF or CRLF");
+        }
     }
 
     /**
@@ -927,6 +933,8 @@ public CSVFormat withQuotePolicy(final Quote quotePolicy) {
      *            the record separator to use for output.
      *
      * @return A new CSVFormat that is equal to this but with the the specified output record separator
+     * @throws IllegalArgumentException
+     *              if recordSeparator is neither CR nor LF
      */
     public CSVFormat withRecordSeparator(final char recordSeparator) {
         return withRecordSeparator(String.valueOf(recordSeparator));
@@ -941,6 +949,8 @@ public CSVFormat withRecordSeparator(final char recordSeparator) {
      *            the record separator to use for output.
      *
      * @return A new CSVFormat that is equal to this but with the the specified output record separator
+     * @throws IllegalArgumentException
+     *              if recordSeparator is none of CR, LF or CRLF
      */
     public CSVFormat withRecordSeparator(final String recordSeparator) {
         return new CSVFormat(delimiter, quoteChar, quotePolicy, commentStart, escape,

src/test/java/org/apache/commons/csv/CSVFormatTest.java

@@ -19,6 +19,7 @@
 
 import static org.apache.commons.csv.CSVFormat.RFC4180;
 import static org.apache.commons.csv.Constants.CR;
+import static org.apache.commons.csv.Constants.CRLF;
 import static org.apache.commons.csv.Constants.LF;
 import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
@@ -118,7 +119,7 @@ public void testEqualsEscape() {
     @Test
     public void testEqualsHeader() {
         final CSVFormat right = CSVFormat.newFormat('\'')
-                .withRecordSeparator('*')
+                .withRecordSeparator(CR)
                 .withCommentStart('#')
                 .withEscape('+')
                 .withHeader("One", "Two", "Three")
@@ -183,23 +184,23 @@ public void testEqualsQuotePolicy() {
     @Test
     public void testEqualsRecordSeparator() {
         final CSVFormat right = CSVFormat.newFormat('\'')
-                .withRecordSeparator('*')
+                .withRecordSeparator(CR)
                 .withCommentStart('#')
                 .withEscape('+')
                 .withIgnoreEmptyLines(true)
                 .withIgnoreSurroundingSpaces(true)
                 .withQuoteChar('"')
                 .withQuotePolicy(Quote.ALL);
         final CSVFormat left = right
-                .withRecordSeparator('!');
+                .withRecordSeparator(LF);
 
         assertNotEquals(right, left);
     }
 
     @Test
     public void testEqualsNullString() {
         final CSVFormat right = CSVFormat.newFormat('\'')
-                .withRecordSeparator('*')
+                .withRecordSeparator(CR)
                 .withCommentStart('#')
                 .withEscape('+')
                 .withIgnoreEmptyLines(true)
@@ -216,7 +217,7 @@ public void testEqualsNullString() {
     @Test
     public void testEqualsSkipHeaderRecord() {
         final CSVFormat right = CSVFormat.newFormat('\'')
-                .withRecordSeparator('*')
+                .withRecordSeparator(CR)
                 .withCommentStart('#')
                 .withEscape('+')
                 .withIgnoreEmptyLines(true)
@@ -409,8 +410,25 @@ public void testWithQuotePolicy() throws Exception {
     }
 
     @Test
-    public void testWithRecordSeparator() throws Exception {
-        final CSVFormat formatWithRecordSeparator = CSVFormat.DEFAULT.withRecordSeparator('!');
-        assertEquals("!", formatWithRecordSeparator.getRecordSeparator());
+    public void testWithRecordSeparatorCR() throws Exception {
+        final CSVFormat formatWithRecordSeparator = CSVFormat.DEFAULT.withRecordSeparator(CR);
+        assertEquals(String.valueOf(CR), formatWithRecordSeparator.getRecordSeparator());
+    }
+
+    @Test
+    public void testWithRecordSeparatorLF() throws Exception {
+        final CSVFormat formatWithRecordSeparator = CSVFormat.DEFAULT.withRecordSeparator(LF);
+        assertEquals(String.valueOf(LF), formatWithRecordSeparator.getRecordSeparator());
+    }
+
+    @Test
+    public void testWithRecordSeparatorCRLF() throws Exception {
+        final CSVFormat formatWithRecordSeparator = CSVFormat.DEFAULT.withRecordSeparator(CRLF);
+        assertEquals(CRLF, formatWithRecordSeparator.getRecordSeparator());
+    }
+
+    @Test(expected = IllegalArgumentException.class)
+    public void testWithRecordSeparatorIllegal() throws Exception {
+        CSVFormat.DEFAULT.withRecordSeparator('!');
     }
 }