/*
* Copyright 2014 the original author or authors.
*
* Licensed 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.springframework.xd.tuple.batch;
import static org.junit.Assert.assertEquals;
import java.math.BigDecimal;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.HashMap;
import org.junit.Before;
import org.junit.Test;
import org.springframework.batch.item.file.transform.DefaultFieldSet;
import org.springframework.batch.item.file.transform.FieldSet;
import org.springframework.xd.tuple.Tuple;
public class TupleFieldSetMapperTests {
private TupleFieldSetMapper mapper;
private static DateFormat formatter = new SimpleDateFormat("yyyy-MM-dd");
@Before
public void setUp() throws Exception {
mapper = new TupleFieldSetMapper();
}
@Test
public void testVanillaTupleCreation() throws Exception {
FieldSet fieldSet = new DefaultFieldSet(new String[] { "This is some dummy string", "true", "C" },
new String[] { "varString", "varBoolean", "varChar" });
Tuple result = mapper.mapFieldSet(fieldSet);
assertEquals("This is some dummy string", result.getString(0));
assertEquals(true, result.getBoolean(1));
assertEquals('C', result.getChar(2));
}
@Test
@SuppressWarnings("serial")
public void testValueConversionByName() throws Exception {
mapper.setTypes(new HashMap<String, FieldSetType>() {
{
put("field1", FieldSetType.BIG_DECIMAL);
put("field2", FieldSetType.BOOLEAN);
put("field3", FieldSetType.BYTE);
put("field4", FieldSetType.CHAR);
put("field5", FieldSetType.DATE);
put("field6", FieldSetType.DOUBLE);
put("field7", FieldSetType.FLOAT);
put("field8", FieldSetType.INT);
put("field9", FieldSetType.LONG);
put("field10", FieldSetType.SHORT);
put("field11", FieldSetType.STRING);
}
});
FieldSet fieldSet = new DefaultFieldSet(new String[] { "19.95", "true", "1", "M", "1977-10-22", "2.22", "9.99",
"5", "9", "10", "some dummy string", "2007-06-23" },
new String[] { "field1", "field2", "field3", "field4", "field5", "field6", "field7", "field8",
"field9", "field10", "field11", "field12" });
Tuple result = mapper.mapFieldSet(fieldSet);
assertEquals(new BigDecimal("19.95"), result.getValue("field1"));
assertEquals(true, result.getValue("field2"));
assertEquals((byte) 1, result.getValue("field3"));
assertEquals('M', result.getValue("field4"));
assertEquals(formatter.parse("1977-10-22"), result.getValue("field5"));
assertEquals(2.22, result.getValue("field6"));
assertEquals(9.99f, result.getValue("field7"));
assertEquals(5, result.getValue("field8"));
assertEquals(9l, result.getValue("field9"));
assertEquals((short) 10, result.getValue("field10"));
assertEquals("some dummy string", result.getValue("field11"));
assertEquals("2007-06-23", result.getValue("field12"));
}
@Test
public void testCustomDateFormatting() throws Exception {
SimpleDateFormat customFormatter = new SimpleDateFormat("MM/dd/yyyy hh:mm a");
mapper.setDateFormat(customFormatter);
FieldSet fieldSet = new DefaultFieldSet(new String[] { "10/22/1977 01:25 AM" }, new String[] { "dateField" });
Tuple result = mapper.mapFieldSet(fieldSet);
assertEquals(customFormatter.parse("10/22/1977 01:25 AM"), result.getDate(0));
}
}