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 | } |