安卓单元测试(八):Junit Rule的使用

JUnit Rule是什么

一个JUnit Rule就是一个实现了TestRule的类,这些类的作用类似于@Before@After,是用来在每个测试方法的执行前后执行一些代码的一个方法。
如果你不清楚@Before@After这些Annotation的意思,Chances are你还不了解Junit的使用,建议先看这篇文章
那为什么不直接用这些annotation呢?这是因为它们都只能作用于一个类,如果同一个setup需要在两个类里面同时使用,那么你就要在两个测试类里面定义相同的@Before方法,然后里面写相同的代码,这就造成了代码重复。有的人说你可以用继承啊,首先我想说,我很讨厌继承这个东西,所以如果可以不用继承的话,我就不会用;再次我想说,如果你不讨厌继承的话,从现在开始,你也应该慢慢的讨厌它了。
此外,JUnit Rule还能做一些@Before这些Annotation做不到的事情,那就是他们可以动态的获取将要运行的测试类、测试方法的信息。这个在接下来的一个例子里面可以看到。

怎么用JUnit Rule?

使用框架自带的Rule

很多测试框架比如JUnit、Mockito自带给我们很多已经实现过好了的JUnit Rule,我们可以直接拿来用。比如TimeoutTemporaryFolder,等等。这些Rule的使用方法非常简单。定义一个这些类的public field,然后用@Rule修饰一下就好了。比如

public class ExampleTest {
    @Rule
    public Timeout timeout = new Timeout(1000);  //使用Timeout这个 Rule,

    @Test
    public void testMethod1() throws Exception {
        //your tests
    }

    @Test
    public void testMethod2() throws Exception {
        //your tests2
    }

    //other test methods
}

那么,对于上面这个ExampleTest的每一个测试方法。它们的运行时间都不能超过1秒钟,不然就会标志为失败。而它的实现方式就是在每个方法测试之前都会记录一下时间戳,然后开始倒计时,1秒钟之后如果方法还没有运行结束,就把结果标记为失败。
这里需要注意的一点是Rule需要是public field

实现自己的Rule

当然,如果只能用框架自带的Rule,那功能未免太受限了,JUnit Rule最强大的地方在于,我们可以自己写满足我们自己需要的Rule。所以现在的问题是怎么写这个Rule。简单来说,写一个Rule就是implement一个TestRule interface,实现一个叫apply()的方法。这个方法需要返回一个Statement对象。下面给一个例子,这个 Rule的作用是,在测试方法运行之前,记录测试方法所在的类名和方法名,然后在测试方法运行之后打印出来,至于怎么在测试方法运行前后做这些事情,下面例子中的注释里面说的很清楚。

public class MethodNameExample implements TestRule {
    @Override
    public Statement apply(final Statement base, final Description description) {
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                //想要在测试方法运行之前做一些事情,就在base.evaluate()之前做
                String className = description.getClassName();
                String methodName = description.getMethodName();

                base.evaluate();  //这其实就是运行测试方法

                //想要在测试方法运行之后做一些事情,就在base.evaluate()之后做
                System.out.println("Class name: "+className +", method name: "+methodName);
            }
        };
    }
}

这个Rule这样就算写好了,现在来试试,用这个 Rule的方法跟使用自带的Rule的用法是一样的,写一个public field,用@Rule修饰一下就好了。

public class ExampleUnitTest {
    @Rule
    public MethodNameExample methodNameExample = new MethodNameExample();
    @Test
    public void addition_isCorrect() throws Exception {
        assertEquals(4, 2 + 2);
    }

    @Test
    public void mulitiplication_isCorrect() throws Exception {
        assertEquals(4, 2 * 2);
    }
}

运行结果如下:

在右边的框框可以看到,把测试方法的方法名和所在的类名打印出来了。

上面的例子对于TestRule的实现应该说的比较清楚,但是看起来没多大用。下面给另外的一个例子,大家或许会觉得这个东西更有用一点。在安卓里面,我们经常在很多地方需要用到Context这个东西。我们的做法是将这个东西保存在一个类里面,作为静态变量存在:

public class ContextHolder {
    private static Context sContext;

    public static void set(Context context) {
        sContext = context;
    }

    public static Context get() {
        return sContext;
    }
}

然后在自定义的Application#onCreate()里面调一下ContextHolder.set()将这个context初始化。

如果你的当前项目是一个library,那么你在测试环境下是没有这个Application的,Robolectric会给你造一个Application,放在RuntimeEnvironment.application里面。所以在测试环境下你可以使用这个instance来将ContextHolder初始化。在这种情况下,你就可以用一个Rule来实现这样的效果,在用到Context的测试方法运行之前将ContextHolder初始化:

public class ContextRule implements TestRule {
    @Override
    public Statement apply(Statement base, Description description) {
        ContextHolder.set(RuntimeEnvironment.application);
        return base;
    }
}

这样,在运行一些用到context的测试方法之前,你就可以使用这个Rule来给Context赋值了。

其实,最常用的Rule之一就是结合@Mock之类的Annotation快速的创建mock,但是这点我想作为一篇单独的文章写一下。原因之一是因为它涉及到的东西不仅仅是Rule,还有其它的一些东西。更重要的原因是,我希望大家能知道这个东西,而不是被这篇文章淹没。请关注下一篇文章吧!

小结

JUnit Rule的介绍就到这里,应该说比较简单,却是非常有帮助。希望这篇文章能帮助到大家了解这个东西。
这篇文章的代码放在这个github repo里面。
如果你对安卓单元测试感兴趣,欢迎加入我们的交流群,因为群成员超过100人,没办法扫码加入,请关注公众号获取加入方法。