RESOLVED - issue BATCH-76: ItemWriter for SQL batch updates

http://jira.springframework.org/browse/BATCH-76

spring-batch-samples/src/test/java/org/springframework/batch/sample/BatchSqlUpdateJobFunctionalTests.java

@@ -0,0 +1,10 @@
+package org.springframework.batch.sample;
+
+/**
+ * Test for BatchSqlUpdateJob - checks that customer credit has been updated to expected value.
+ * 
+ * @author Robert Kasanicky
+ */
+public class BatchSqlUpdateJobFunctionalTests extends AbstractCustomerCreditIncreaseTests {
+
+}

spring-batch-samples/src/test/java/org/springframework/batch/sample/item/writer/BatchSqlCustomerCreditIncreaseWriterTests.java

@@ -0,0 +1,101 @@
+/*
+ * Copyright 2006-2007 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.batch.sample.item.writer;
+
+import java.math.BigDecimal;
+
+import junit.framework.TestCase;
+
+import org.easymock.MockControl;
+import org.springframework.batch.item.ItemWriter;
+import org.springframework.batch.sample.domain.CustomerCredit;
+
+/**
+ * @author Dave Syer
+ * 
+ */
+public class BatchSqlCustomerCreditIncreaseWriterTests extends TestCase {
+
+	private BatchSqlCustomerCreditIncreaseWriter writer = new BatchSqlCustomerCreditIncreaseWriter();
+
+	private ItemWriter delegate;
+
+	private MockControl control = MockControl.createControl(ItemWriter.class);
+
+	private CustomerCredit customerCredit;
+
+	/*
+	 * (non-Javadoc)
+	 * @see junit.framework.TestCase#setUp()
+	 */
+	protected void setUp() throws Exception {
+		delegate = (ItemWriter) control.getMock();
+		writer.setDelegate(delegate);
+		customerCredit = new CustomerCredit();
+		customerCredit.setId(13);
+		customerCredit.setCredit(new BigDecimal(1000));
+		customerCredit.setName("foo");
+	}
+	
+	public void testAfterPropertiesSet() throws Exception {
+		try {
+			writer.afterPropertiesSet();
+			fail("Expected IllegalStateException");
+		} catch (IllegalStateException e) {
+			// expected: wrong class
+			String message = e.getMessage();
+			assertTrue("Message does not contain 'instance'"+message, message.indexOf("instance")>=0);
+		}
+	}
+
+	/**
+	 * Test method for
+	 * {@link org.springframework.batch.sample.item.writer.BatchSqlCustomerCreditIncreaseWriter#write(java.lang.Object)}.
+	 * @throws Exception
+	 */
+	public void testWrite() throws Exception {
+		delegate.write(customerCredit);
+		control.setVoidCallable();
+		control.replay();
+		writer.write(customerCredit);
+		control.verify();
+	}
+
+	/**
+	 * Test method for
+	 * {@link org.springframework.batch.sample.item.writer.BatchSqlCustomerCreditIncreaseWriter#clear()}.
+	 */
+	public void testClear() {
+		delegate.clear();
+		control.setVoidCallable();
+		control.replay();
+		writer.clear();
+		control.verify();
+	}
+
+	/**
+	 * Test method for
+	 * {@link org.springframework.batch.sample.item.writer.BatchSqlCustomerCreditIncreaseWriter#flush()}.
+	 */
+	public void testFlush() {
+		delegate.flush();
+		control.setVoidCallable();
+		control.replay();
+		writer.flush();
+		control.verify();
+	}
+
+}

spring-batch-samples/src/test/java/org/springframework/batch/sample/item/writer/CustomerCreditUpdatePreparedStatementSetterTests.java

@@ -0,0 +1,65 @@
+/*
+ * Copyright 2006-2007 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.batch.sample.item.writer;
+
+import java.math.BigDecimal;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
+
+import junit.framework.TestCase;
+
+import org.easymock.MockControl;
+import org.springframework.batch.sample.domain.CustomerCredit;
+
+/**
+ * @author Dave Syer
+ *
+ */
+public class CustomerCreditUpdatePreparedStatementSetterTests extends TestCase {
+	
+	private CustomerCreditUpdatePreparedStatementSetter setter = new CustomerCreditUpdatePreparedStatementSetter();
+
+	private CustomerCredit credit;
+
+	private PreparedStatement ps;
+
+	private MockControl control = MockControl.createControl(PreparedStatement.class);
+
+	/* (non-Javadoc)
+	 * @see junit.framework.TestCase#setUp()
+	 */
+	protected void setUp() throws Exception {
+		ps = (PreparedStatement) control.getMock();
+		credit = new CustomerCredit();
+		credit.setId(13);
+		credit.setCredit(new BigDecimal(12000));
+		credit.setName("foo");
+	}
+	/**
+	 * Test method for {@link org.springframework.batch.sample.item.writer.CustomerCreditUpdatePreparedStatementSetter#setValues(java.lang.Object, java.sql.PreparedStatement)}.
+	 * @throws SQLException 
+	 */
+	public void testSetValues() throws SQLException {
+		ps.setBigDecimal(1, credit.getCredit());
+		control.setVoidCallable();
+		ps.setLong(2, credit.getId());
+		control.setVoidCallable();
+		control.replay();
+		setter.setValues(credit, ps);
+		control.verify();
+	}
+
+}

spring-batch-samples/src/test/java/org/springframework/batch/sample/mapping/CustomerCreditRowMapperTests.java

@@ -11,11 +11,16 @@ import org.springframework.jdbc.core.RowMapper;
 
 public class CustomerCreditRowMapperTests extends AbstractRowMapperTests {
 
+	/**
+	 * 
+	 */
+	private static final int ID = 12;
 	private static final String CUSTOMER = "Jozef Mak";
 	private static final BigDecimal CREDIT = new BigDecimal(0.1);
 
 	protected Object expectedDomainObject() {
 		CustomerCredit credit = new CustomerCredit();
+		credit.setId(ID);
 		credit.setCredit(CREDIT);
 		credit.setName(CUSTOMER);
 		return credit;
@@ -26,6 +31,8 @@ public class CustomerCreditRowMapperTests extends AbstractRowMapperTests {
 	}
 
 	protected void setUpResultSetMock(ResultSet rs, MockControl rsControl) throws SQLException {
+		rs.getInt(CustomerCreditRowMapper.ID_COLUMN);
+		rsControl.setReturnValue(ID);
 		rs.getString(CustomerCreditRowMapper.NAME_COLUMN);
 		rsControl.setReturnValue(CUSTOMER);
 		rs.getBigDecimal(CustomerCreditRowMapper.CREDIT_COLUMN);