Add testcases for CSVRecord with get(Enum) and toString.

Author: dota17

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

@@ -41,6 +41,23 @@ private enum EnumFixture {
         UNKNOWN_COLUMN
     }
 
+    public enum EnumHeader {
+        FIRST("first"),
+        SECOND("second"),
+        THIRD("third");
+
+        private String number;
+
+        EnumHeader(String number) {
+            this.number = number;
+        }
+
+        @Override
+        public String toString() {
+            return number;
+        }
+    }
+
     private String[] values;
     private CSVRecord record, recordWithHeader;
     private Map<String, Integer> headerMap;
@@ -164,7 +181,7 @@ public void testPutInMap() {
         final Map<String, String> map = new ConcurrentHashMap<>();
         this.recordWithHeader.putIn(map);
         this.validateMap(map, false);
-        // Test that we can compile with assigment to the same map as the param.
+        // Test that we can compile with assignment to the same map as the param.
         final TreeMap<String, String> map2 = recordWithHeader.putIn(new TreeMap<String, String>());
         this.validateMap(map2, false);
     }
@@ -222,4 +239,18 @@ private void validateMap(final Map<String, String> map, final boolean allowsNull
         assertEquals(null, map.get("fourth"));
     }
 
+    @Test
+    public void testToString() {
+        assertNotNull(recordWithHeader.toString());
+        assertTrue(recordWithHeader.toString().contains("comment="));
+        assertTrue(recordWithHeader.toString().contains("recordNumber="));
+        assertTrue(recordWithHeader.toString().contains("values="));
+    }
+
+    @Test
+    public void testGetWithEnum() {
+        assertEquals(recordWithHeader.get("first"), recordWithHeader.get(EnumHeader.FIRST));
+        assertEquals(recordWithHeader.get("second"), recordWithHeader.get(EnumHeader.SECOND));
+        assertThrows(IllegalArgumentException.class, () -> recordWithHeader.get(EnumFixture.UNKNOWN_COLUMN));
+    }
 }