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

作者: 邹小创 | 来源:发表于2016-07-10 11:54 被阅读2871次

    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里面。

    获取最新文章或想加入安卓单元测试交流群,请关注下方公众号

    相关文章

      网友评论

      • 天才木木:为什么要讨厌继承?
      • fe2a5bcdf6ea:跟@Mock绝配!看来我也得写一篇rule了グッ!(๑•̀ㅂ•́)و✧
      • 键盘男:跟@Mock绝配!看来我也得写一篇rule了グッ!(๑•̀ㅂ•́)و✧
        邹小创:@键盘男kkmike999 写写写!

      本文标题:安卓单元测试(八):Junit Rule的使用

      本文链接:https://www.haomeiwen.com/subject/sqgrjttx.html