JUnit4.12 源码分析(二)之TestRule

1. TestRule

• TestRule和@Before,@After,@BeforeClass,@AfterClass功能类似,但是更加强大;
• JUnit 识别TestRule的两种方式:
  • 方法级别:@Rule;
  • 类级别:@ClassRule;
• TestRule的实现类:
  • ErrorCollector
  • ExpectedException
  • ExternalResource
  • TemporaryFolder
  • TestName
  • TestWatcher
  • Timeout
  • Verifier

// org.juit.rules.TestRule 源码

public interface TestRule{

    // 只有一个抽象方法

    // 子类需要对Statement进行修改

    // 返回值:可能是原来的base,或者装饰之后的base,或者一个全新的base

    Statement apply(Statement base, Description description);

}

// org.junit.runners.model.Statement

public abstract class Statement{

    // 只有一个抽象方法

    public abstract void evalute() throws Throwable;

}

// org.junit.rules.ExpectedException

public class ExpectedException implements TestRule{

    ...(略)

    // apply 方法的实现

    public Statement apply(Statement base, Description description){

        return new ExpectedExceptionStatement(base);

    }

    // 对Statement的处理

    private class ExpectedExceptionStatement extends Statement{

        private final Statement next;

        public ExpectedExceptionStatement(Statement base){

            next = base;

        }

        @Override

        public void evalute() throws Throwable{

            try{

                next.evalute();

            }catch(Throwable e){

                handleException(e);

                return;

            }

            if(isAnyExceptionExpected()){

                failDueToMissingException();

            }

        }

    }

    ...(略)

}

1.1 测试示例

// 自定义Statement

public class MyStatement extends Statement{

    private final Statement myBase;

    public MyStatement(Statement base){

        this.myBase = base;

    }

    @Override

    public void evalute() throws Throwable{

        System.out.println("方法执行开始之前");

        try{

            myBase.evalute();

        }finally{

            System.out.println("方法执行之后");

        }

    }

}

// 自定义Rule

public class MyRule implements TestRule{

    @Override

    public Statement apply(Statement base, Description description){

        return new MyStatement(base);

    }

}

// 自定义Test

public class MyTest{

    @Rule

    public MyRule myRule = new MyRule();

    @Test

    public void testCase(){

        System.out.println("测试运行....");

    }

}

// 控制台输出:

方法执行开始之前

测试运行....

方法执行之后

参考资料:

JUnit之Rule的使用