RESOLVED - issue BATCH-1050: SimpleJobExplorer doesn't retrieve StepExecutions of running JobExecutions using MapStepExecutionDao

Added step executions explicitly in the Dao.

spring-batch-core/src/test/java/org/springframework/batch/core/JobExecutionTests.java

@@ -21,6 +21,7 @@ import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertNull;
 import static org.junit.Assert.assertTrue;
 
+import java.util.Arrays;
 import java.util.Date;
 import java.util.List;
 
@@ -171,6 +172,13 @@ public class JobExecutionTests {
 		assertEquals(1, execution.getStepExecutions().size());
 	}
 
+	@Test
+	public void testSetStepExecutions() throws Exception {
+		assertEquals(0, execution.getStepExecutions().size());
+		execution.addStepExecutions(Arrays.asList(new StepExecution("step", execution)));
+		assertEquals(1, execution.getStepExecutions().size());
+	}
+
 	@Test
 	public void testStop() throws Exception {
 		StepExecution stepExecution = execution.createStepExecution("step");

spring-batch-core/src/test/java/org/springframework/batch/core/exlore/support/MapJobExplorerIntegrationTests.java

@@ -0,0 +1,82 @@
+/*
+ * 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.core.exlore.support;
+
+import static org.junit.Assert.assertEquals;
+
+import java.util.Set;
+
+import org.junit.Test;
+import org.springframework.batch.core.JobExecution;
+import org.springframework.batch.core.JobParametersBuilder;
+import org.springframework.batch.core.StepContribution;
+import org.springframework.batch.core.explore.JobExplorer;
+import org.springframework.batch.core.explore.support.MapJobExplorerFactoryBean;
+import org.springframework.batch.core.job.SimpleJob;
+import org.springframework.batch.core.launch.support.SimpleJobLauncher;
+import org.springframework.batch.core.repository.JobRepository;
+import org.springframework.batch.core.repository.support.MapJobRepositoryFactoryBean;
+import org.springframework.batch.core.scope.context.ChunkContext;
+import org.springframework.batch.core.step.tasklet.Tasklet;
+import org.springframework.batch.core.step.tasklet.TaskletStep;
+import org.springframework.batch.repeat.RepeatStatus;
+import org.springframework.batch.support.transaction.ResourcelessTransactionManager;
+import org.springframework.core.task.SimpleAsyncTaskExecutor;
+
+/**
+ * @author Dave Syer
+ * 
+ */
+public class MapJobExplorerIntegrationTests {
+
+	private boolean block = true;
+
+	@Test
+	public void testRunningJobExecution() throws Exception {
+		SimpleJobLauncher jobLauncher = new SimpleJobLauncher();
+		MapJobRepositoryFactoryBean repositoryFactory = new MapJobRepositoryFactoryBean();
+		ResourcelessTransactionManager transactionManager = new ResourcelessTransactionManager();
+		repositoryFactory.setTransactionManager(transactionManager);
+		repositoryFactory.afterPropertiesSet();
+		JobRepository jobRepository = (JobRepository) repositoryFactory.getObject();
+		jobLauncher.setJobRepository(jobRepository);
+		jobLauncher.setTaskExecutor(new SimpleAsyncTaskExecutor());
+		jobLauncher.afterPropertiesSet();
+		SimpleJob job = new SimpleJob("job");
+		TaskletStep step = new TaskletStep("step");
+		step.setTasklet(new Tasklet() {
+			public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
+				while (block) {
+					Thread.sleep(100L);
+				}
+				return RepeatStatus.FINISHED;
+			}
+		});
+		step.setTransactionManager(transactionManager);
+		step.setJobRepository(jobRepository);
+		step.afterPropertiesSet();
+		job.addStep(step);
+		job.setJobRepository(jobRepository);
+		job.afterPropertiesSet();
+		jobLauncher.run(job, new JobParametersBuilder().addString("test", getClass().getName()).toJobParameters());
+		Thread.sleep(500L);
+		JobExplorer explorer = (JobExplorer) new MapJobExplorerFactoryBean().getObject();
+		Set<JobExecution> executions = explorer.findRunningJobExecutions("job");
+		assertEquals(1, executions.size());
+		assertEquals(1, executions.iterator().next().getStepExecutions().size());
+		block = false;
+	}
+}