|
1 |
| package com.agical.rmock.core.strategy.impl; |
|
2 |
| |
|
3 |
| import com.agical.rmock.core.strategy.TestStep; |
|
4 |
| |
|
5 |
| public class FinallyStep implements TestStep { |
|
6 |
| private TestStep mainStep; |
|
7 |
| private TestStep protectedStep; |
|
8 |
| |
|
9 |
558
| public void setMainStep(TestStep mainStep) {
|
|
10 |
558
| this.mainStep = mainStep;
|
|
11 |
| } |
|
12 |
| |
|
13 |
558
| public void setFinallyStep(TestStep protectedStep) {
|
|
14 |
558
| this.protectedStep = protectedStep;
|
|
15 |
| } |
|
16 |
| |
|
17 |
557
| public void run() throws Throwable {
|
|
18 |
557
| try {
|
|
19 |
557
| mainStep.run();
|
|
20 |
| } |
|
21 |
| finally { |
|
22 |
557
| if(protectedStep != null) {
|
|
23 |
557
| protectedStep.run();
|
|
24 |
| } |
|
25 |
| } |
|
26 |
| } |
|
27 |
| } |
