Add CSVRecord#getInt(String) API and tests.
git-svn-id: https://svn.apache.org/repos/asf/commons/proper/csv/trunk@1509450 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
4bb75e3edb
commit
70485e44f2
|
@ -134,6 +134,23 @@ public class CSVRecord implements Serializable, Iterable<String> {
|
|||
return comment;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a value by name.
|
||||
*
|
||||
* @param name
|
||||
* the name of the column to be retrieved.
|
||||
* @return the column value
|
||||
* @throws IllegalStateException
|
||||
* if no header mapping was provided
|
||||
* @throws IllegalArgumentException
|
||||
* if the record is inconsistent
|
||||
* @see #isConsistent()
|
||||
*/
|
||||
public int getInt(String name) {
|
||||
String s = this.get(name);
|
||||
return s != null ? Integer.parseInt(s) : 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a value by name.
|
||||
*
|
||||
|
|
|
@ -0,0 +1,64 @@
|
|||
/*
|
||||
* 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
|
||||
*
|
||||
* http://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;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
|
||||
public class CSVRecordIntTest {
|
||||
|
||||
private CSVRecord record;
|
||||
|
||||
/**
|
||||
* @return
|
||||
* @throws IOException
|
||||
*/
|
||||
private CSVRecord createTestRecord() throws IOException {
|
||||
String csv = "A, B, C, D, E\n-1, 0, 1, " + Integer.MAX_VALUE + ", " + Integer.MIN_VALUE;
|
||||
CSVRecord record = CSVParser.parseString(csv, CSVFormat.DEFAULT.withHeader().withIgnoreSurroundingSpaces(true))
|
||||
.iterator().next();
|
||||
return record;
|
||||
}
|
||||
|
||||
@Before
|
||||
public void setUp() throws IOException {
|
||||
this.record = createTestRecord();
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testGetIntegerByMissingString() {
|
||||
Assert.assertEquals(null, Integer.valueOf(record.getInt("ABSENT")));
|
||||
}
|
||||
|
||||
@Test(expected = IllegalArgumentException.class)
|
||||
public void testGetIntegerByNullString() {
|
||||
Assert.assertEquals(null, Integer.valueOf(record.getInt(null)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetIntegerByString() {
|
||||
Assert.assertEquals(-1, record.getInt("A"));
|
||||
Assert.assertEquals(0, record.getInt("B"));
|
||||
Assert.assertEquals(1, record.getInt("C"));
|
||||
Assert.assertEquals(Integer.MAX_VALUE, record.getInt("D"));
|
||||
Assert.assertEquals(Integer.MIN_VALUE, record.getInt("E"));
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in New Issue