JMockit Mock 私有方法和私有属性

calendar Feb 3, 2014   ---   · 2 min read · Unit Test JUnit jMockit  ·
Share on: linkedin copy
前面说过 JMockit 因身处前线，所以简直无不可，本节例子演示 JMockit 怎么 Mock 私有方法和私有属性，示例虽然是静态方法和属性，但因采用的是反射手法，所以这种 Deencapsulation 的 Mock 手段同样适用于公有的方法或属性，无论是否静态。

本文所用 JMockit 版本为 1.6, 可能网上所搜索的方法与此有所不同，请注意 JMockit 版本差异。仍需重复一下，运行 JMockit 的例子 classpath 上必须让 jmockit.jar 在 junit.jar 之前，或用 javaagent 参数来加载 jmockit.jar，并且 junit 要 4.8 及以上版本.

1. Mock 私有方法(非静态类似)
 1package cc.unmi;
 2
 3public class MyService {
 4    public static String fetchData(String name){
 5        System.out.println("call MyService.fetchData");
 6        return fetchDataFromDB(name);
 7    }
 8
 9    private static String fetchDataFromDB(String name){
10        throw new RuntimeException("Not implemented yet!");
11    }
12}
fetchDataFromDB 是私有静态方法，正常测试的话肯定不过
 1package cc.unmi;
 2
 3import mockit.Deencapsulation;
 4import mockit.Expectations;
 5
 6import org.junit.Assert;
 7import org.junit.Test;
 8
 9public class MyServiceTest {
10
11    @Test
12    public void testFetchData() {
13
14        new Expectations(MyService.class) {
15            {
16                Deencapsulation.invoke(MyService.class, "fetchDataFromDB", "Unmi");
17                result = "http://unmi.cc";
18            }
19        };
20
21        String actual = MyService.fetchData("Unmi");
22        Assert.assertEquals("http://unmi.cc", actual);
23    }
24}
这行
Deencapsulation.invoke(MyService.class, "fetchDataFromDB", "Unmi");
使用了反射的方式拦截了 MyService 的 fetchDataFromDB 方法，并非强设了它的返回值为  “http://unmi.cc”。关键是调用了 Deencapsulation.invoke() 方法，来看到这个类里还有哪些方法：
看到上图，我们必须要产生一些想法的，特别是 invoke 和 setField，以及它们的第一个参数可以是 Class，也可以是 Object，不难获知的是我们能够借助于它来 Mocket 私有方法和属性，不论它们是静态还是非静态的。

上面的测试用例可以见绿，输出为：
call MyService.fetchData
如果把测试代码中的调用处改为
String actual = MyService.fetchData("Unmis");
就要见红了，失败：
java.lang.RuntimeException: Not implemented yet!
at cc.unmi.MyService.fetchDataFromDB(MyService.java:10)
at cc.unmi.MyService.fetchData(MyService.java:6)
at cc.unmi.MyServiceTest.testFetchData(MyServiceTest.java:21)
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
at java.lang.reflect.Method.invoke(Method.java:606)
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
at java.lang.reflect.Method.invoke(Method.java:606)
................
对了，就是反射。受上面的点拨，怎么 Mock 属性已了然于心了，再行累述纯为拉长篇幅，不妨

2. Mock 私有属性(静态亦类似)
 1package cc.unmi;
 2
 3public class MyService {
 4
 5    private String url = "";
 6
 7    public String fetchData(String name){
 8        System.out.println("call MyService.fetchData");
 9        return this.url;
10    }
11}
欲 Mock 的是 url 属性，从 fetchData 方法的 return url 可以检测到 Mock 后的值
 1package cc.unmi;
 2
 3import mockit.Deencapsulation;
 4import mockit.Expectations;
 5
 6import org.junit.Assert;
 7import org.junit.Test;
 8
 9public class MyServiceTest {
10
11    @Test
12    public void testFetchData() {
13        final MyService service = new MyService();
14        Deencapsulation.setField(service, "url", "http://unmi.cc");
15
16        String actual = service.fetchData("Unmi");
17        Assert.assertEquals("http://unmi.cc", actual);
18    }
19}
断言成功，也用不着设置 result 值了, 而且这时候也用不着 new Expectations()  了。

Deencapsulation 是后来引入的，在 JMockit 0.999.19 中可以在  Expectations 中直接调用 invoke, setField 方法来改变运行中的值，Expectations 继承自 Invocations，原来的 Invocations 中定义了 invoke, setField, getFiled 等方法，新版的 JMockit 从  Invocations 中移除了那些方法。

记得当初我们用 @MockClass, @Mock, 再 Mockit.setUpMock() 三步曲来进行 Mock 操作，如今看来光 new Expectations() 这一招便可走遍天下。 永久链接 https://yanbin.blog/jmockit-mock-private-methods-fields/, 来自 隔叶黄莺 Yanbin's Blog
[版权声明] 本文采用 署名-非商业性使用-相同方式共享 4.0 国际 (CC BY-NC-SA 4.0) 进行许可。