Add simple integration test for restart using JobLauncher

spring-batch-core/src/test/java/org/springframework/batch/core/launch/JobLauncherIntegrationTests.java

@@ -0,0 +1,76 @@
+package org.springframework.batch.core.launch;
+
+import static org.junit.Assert.assertEquals;
+
+import java.util.Calendar;
+
+import javax.sql.DataSource;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.batch.core.Job;
+import org.springframework.batch.core.JobExecution;
+import org.springframework.batch.core.JobInstance;
+import org.springframework.batch.core.JobParametersBuilder;
+import org.springframework.batch.core.repository.dao.JdbcJobInstanceDao;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.jdbc.core.simple.SimpleJdbcTemplate;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+@ContextConfiguration
+@RunWith(SpringJUnit4ClassRunner.class)
+public class JobLauncherIntegrationTests {
+
+	private SimpleJdbcTemplate jdbcTemplate;
+
+	@Autowired
+	private JobLauncher jobLauncher;
+
+	@Autowired
+	private Job job;
+
+	@Autowired
+	public void setDataSource(DataSource dataSource) {
+		jdbcTemplate = new SimpleJdbcTemplate(dataSource);
+	}
+	
+	@Test
+	public void testLaunchAndRelaunch() throws Exception {
+		
+		int before = jdbcTemplate.queryForInt("select count(*) from BATCH_JOB_INSTANCE");
+		
+		JobExecution jobExecution = launch(true,0);
+		launch(false, jobExecution.getJobId());
+		launch(false, jobExecution.getJobId());
+
+		int after = jdbcTemplate.queryForInt("select count(*) from BATCH_JOB_INSTANCE");
+		assertEquals(before+1, after);
+
+	}
+
+	private JobExecution launch(boolean start, long jobInstanceID) throws Exception {
+
+		if (start) {
+
+			Calendar c = Calendar.getInstance();
+			JobParametersBuilder builder = new JobParametersBuilder();
+			builder.addDate("TIMESTAMP", c.getTime());
+			return jobLauncher.run(job, builder.toJobParameters());
+
+		} else {
+
+			JdbcJobInstanceDao dao = new JdbcJobInstanceDao();
+			dao.setJdbcTemplate(jdbcTemplate);
+			JobInstance instance = dao.getJobInstance(jobInstanceID);
+			if (instance != null) {
+				return jobLauncher.run(job, instance.getJobParameters());
+			}
+
+			return null;
+
+		}
+
+	}
+
+}