RESOLVED - issue BATCH-183: Unit test for JMX notifications / control

spring-batch-samples/src/main/java/org/springframework/batch/sample/launch/DefaultJobLoader.java

@@ -0,0 +1,96 @@
+/**
+ * 
+ */
+package org.springframework.batch.sample.launch;
+
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.Map;
+
+import org.springframework.batch.core.Job;
+import org.springframework.batch.core.configuration.ListableJobRegistry;
+import org.springframework.batch.core.repository.NoSuchJobException;
+import org.springframework.beans.BeanWrapperImpl;
+import org.springframework.beans.BeansException;
+import org.springframework.beans.PropertyAccessorUtils;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ApplicationContextAware;
+import org.springframework.context.support.ClassPathXmlApplicationContext;
+import org.springframework.util.Assert;
+
+public class DefaultJobLoader implements JobLoader,
+		ApplicationContextAware {
+
+	private ListableJobRegistry registry;
+	private ApplicationContext applicationContext;
+	private Map configurations = new HashMap();
+
+	public void setApplicationContext(ApplicationContext applicationContext)
+			throws BeansException {
+		this.applicationContext = applicationContext;
+	}
+
+	public void setRegistry(ListableJobRegistry registry) {
+		this.registry = registry;
+	}
+
+	public Map getConfigurations() {
+		Map result = new HashMap(configurations);
+		for (Iterator iterator = registry.getJobNames().iterator(); iterator
+				.hasNext();) {
+			try {
+				Job configuration = (Job) registry.getJob((String) iterator.next());
+				String name = configuration.getName();
+				if (!configurations.containsKey(name)) {
+					result.put(name, "<unknown path>: " + configuration);
+				}
+			}
+			catch (NoSuchJobException e) {
+				throw new IllegalStateException("Registry could not locate its own job (NoSuchJobException).");
+			}
+		}
+		return result;
+	}
+
+	public void loadResource(String path) {
+		ClassPathXmlApplicationContext context = new ClassPathXmlApplicationContext(
+				new String[] { path }, applicationContext);
+		String[] names = context.getBeanNamesForType(Job.class);
+		for (int i = 0; i < names.length; i++) {
+			String name = names[i];
+			configurations.put(name, path);
+		}
+	}
+	
+	public Object getJobConfiguration(String name) {
+		try {
+			return registry.getJob(name);
+		} catch (NoSuchJobException e) {
+			return null;
+		}
+	}
+
+	public Object getProperty(String path) {
+		int index = PropertyAccessorUtils.getFirstNestedPropertySeparatorIndex(path);
+		BeanWrapperImpl wrapper = createBeanWrapper(path, index);
+		String key = path.substring(index+1);
+		return wrapper.getPropertyValue(key);
+	}
+
+	public void setProperty(String path, String value) {
+		int index = PropertyAccessorUtils.getFirstNestedPropertySeparatorIndex(path);
+		BeanWrapperImpl wrapper = createBeanWrapper(path, index);
+		String key = path.substring(index+1);
+		wrapper.setPropertyValue(key, value);
+	}
+
+	private BeanWrapperImpl createBeanWrapper(String path, int index) {
+		Assert.state(index>0, "Path must be nested, e.g. bean.value");
+		String name = path.substring(0,index);
+		Object bean = getJobConfiguration(name);
+		Assert.notNull(bean, "No JobConfiguration exists with name="+name);
+		BeanWrapperImpl wrapper = new BeanWrapperImpl(bean);
+		return wrapper;
+	}
+
+}

spring-batch-samples/src/main/java/org/springframework/batch/sample/launch/ExportedJobLoader.java

@@ -0,0 +1,37 @@
+/*
+ * 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.launch;
+
+import java.util.Map;
+
+
+/**
+ * @author Dave Syer
+ * 
+ */
+public interface ExportedJobLoader {
+	
+	void loadResource(String path);
+	
+	Map getConfigurations();
+	
+	String getJob(String path);
+	
+	String getProperty(String path);
+
+	void setProperty(String path, String value);
+
+}

spring-batch-samples/src/main/java/org/springframework/batch/sample/launch/JobLoader.java

@@ -0,0 +1,37 @@
+/*
+ * 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.launch;
+
+import java.util.Map;
+
+
+/**
+ * @author Dave Syer
+ * 
+ */
+public interface JobLoader {
+	
+	void loadResource(String path);
+	
+	Map getConfigurations();
+	
+	Object getJobConfiguration(String path);
+	
+	Object getProperty(String path);
+
+	void setProperty(String path, String value);
+
+}

spring-batch-samples/src/main/java/org/springframework/batch/sample/launch/TaskExecutorLauncher.java

@@ -15,6 +15,9 @@
  */
 package org.springframework.batch.sample.launch;
 
+import java.util.ArrayList;
+import java.util.List;
+
 import org.springframework.batch.core.Job;
 import org.springframework.batch.core.configuration.JobRegistry;
 import org.springframework.batch.core.repository.DuplicateJobException;
@@ -39,6 +42,8 @@ public class TaskExecutorLauncher implements ResourceLoaderAware {
 
 	private ApplicationContext parentContext = null;
 
+	private static List errors = new ArrayList();
+
 	/**
 	 * Public setter for the {@link JobRegistry}.
 	 * @param registry the registry to set
@@ -54,6 +59,14 @@ public class TaskExecutorLauncher implements ResourceLoaderAware {
 	public void setResourceLoader(ResourceLoader resourceLoader) {
 		this.resourceLoader = resourceLoader;
 	}
+	
+	/**
+	 * Public getter for the errors.
+	 * @return the errors
+	 */
+	public static List getErrors() {
+		return errors;
+	}
 
 	private void register(String[] paths) throws DuplicateJobException {
 		for (int i = 0; i < paths.length; i++) {
@@ -70,10 +83,17 @@ public class TaskExecutorLauncher implements ResourceLoaderAware {
 	public static void main(String[] args) throws Exception {
 
 		final TaskExecutorLauncher launcher = new TaskExecutorLauncher();
+		errors.clear();
 
 		new Thread(new Runnable() {
 			public void run() {
-				launcher.run();
+				try {
+					launcher.run();
+				}
+				catch (RuntimeException e) {
+					errors.add(e);
+					throw e;
+				}
 			};
 		}).start();