View Javadoc

1   /*
2    * Copyright 2006-2010 the original author or authors.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *      http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  package org.springframework.batch.test;
17  
18  import java.util.concurrent.Callable;
19  
20  import org.springframework.batch.core.StepExecution;
21  import org.springframework.batch.core.scope.StepScope;
22  import org.springframework.batch.core.scope.context.StepSynchronizationManager;
23  
24  /**
25   * Utility class for creating and manipulating {@link StepScope} in unit tests.
26   * This is useful when you want to use the Spring test support and inject
27   * dependencies into your test case that happen to be step scoped in the
28   * application context.
29   * 
30   * @author Dave Syer
31   * 
32   */
33  public class StepScopeTestUtils {
34  
35  	public static <T> T doInStepScope(StepExecution stepExecution, Callable<T> callable) throws Exception {
36  		try {
37  			StepSynchronizationManager.register(stepExecution);
38  			return callable.call();
39  		}
40  		finally {
41  			StepSynchronizationManager.close();
42  		}
43  	}
44  
45  }