JMockit简单使用(二)

       JMockit使用时，建议使用Expectations{}块来录制行为，这样mock对象在运行时必须 依据Expectations块中定义的顺序依次调用方法，不能多调用也不能少调用，可以省略掉Verifications块。一旦多调用或则少调用，那么测试就不会通过，这样可以清晰的把握单元测试每一步、每一次执行的过程。

       但是，测试的方法中如果存在线程的调用来执行录制行为，很多情况用Expectations{}块测试都不会通过，而使用NonStrictExpectations{}块来录制就不会有问题。

示例:

测试目标：

package org.zero.jmockit;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

class OtherVO {
private String value;

public String getValue() {
return value;
}

public void setValue(String value) {
this.value = value;
}
}

public class Work {
private OtherVO otherVO;
private ExecutorService executor = Executors.newFixedThreadPool(4);

public void task() {
executor.execute(new Runnable() {
public void run() {
otherVO.getValue();
}
});
}
}

测试文件：

package org.zero.jmockit;

import mockit.Deencapsulation;
import mockit.Expectations;
import mockit.Mocked;
import mockit.Tested;
import mockit.integration.junit4.JMockit;

import org.junit.Test;
import org.junit.runner.RunWith;

@RunWith(JMockit.class)
public class WorkTest {
@Tested
@Mocked
Work work;

@Test
public void testTask() {
final OtherVO otherVO = new OtherVO();
new Expectations(OtherVO.class) { // mock OtherVO的行为
{
otherVO.getValue();
result = "zero";
}
};
new Expectations() {
{
Deencapsulation.setField(work, "otherVO", otherVO);
}
};
// --- 通过以上方式可以将mock出来的otherVO设置到work中，当然可以采取其它简便方式

work.task();
}
}

运行测试，会发现测试不通过，会抛出一下异常：

mockit.internal.MissingInvocation: Missing 1 invocation to:
org.zero.jmockit.OtherVO#getValue()
on mock instance: org.zero.jmockit.OtherVO@3419866c
at org.eclipse.jdt.internal.junit4.runner.JUnit4TestReference.run(JUnit4TestReference.java:50)
at org.eclipse.jdt.internal.junit.runner.TestExecution.run(TestExecution.java:38)
at org.eclipse.jdt.internal.junit.runner.RemoteTestRunner.runTests(RemoteTestRunner.java:459)
at org.eclipse.jdt.internal.junit.runner.RemoteTestRunner.runTests(RemoteTestRunner.java:675)
at org.eclipse.jdt.internal.junit.runner.RemoteTestRunner.run(RemoteTestRunner.java:382)
at org.eclipse.jdt.internal.junit.runner.RemoteTestRunner.main(RemoteTestRunner.java:192)
Caused by: Missing invocations
at org.zero.jmockit.OtherVO.getValue(Work.java)
at org.zero.jmockit.WorkTest$1.<init>(WorkTest.java:23)
at org.zero.jmockit.WorkTest.testTask(WorkTest.java:21)
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
at java.lang.reflect.Method.invoke(Method.java:497)
... 6 more

明明调用了work.task(),task()里会调用getValue()方法，但是怎么会报Missing invocations错误(少调了OtherVO#getValue()方法)。为什么呢？原因在于线程的调用，executor.execute()只是通知线程去执行runnable，runnable中任务的执行需要JVM去调度，至于什么时候去执行，可能发生在整个testTask()结束后，而Expectations{}检测到录制行为在testTask()结束前依然没被回放，因此就会报错。

怎么解决呢？

       在work.task()调用后sleep()一段时间，以等runnable中任务被执行了再结束testTask()不就可以了么。这也是一个办法。但最好的解决方式是mock executor的execute()方法，让它不执行runnable.start(),而是执行runnable.run()，如下：

@RunWith(JMockit.class)
public class WorkTest {
@Tested
@Mocked
Work work;

private ExecutorService executor;

@Before
public void initThread() {
executor = new MockUp<ThreadPoolExecutor>() {
@Mock
public void execute(Runnable command) {
command.run();
}
}.getMockInstance();
}

@Test
public void testTask() {
final OtherVO otherVO = new OtherVO();
new Expectations(OtherVO.class) { // mock OtherVO的行为
{
otherVO.getValue();
result = "zero";
}
};
new Expectations() {
{
Deencapsulation.setField(work, "otherVO", otherVO);
}
};
// --- 通过以上方式可以将mock出来的otherVO设置到work中，当然可以采取其它简便方式

new Expectations() {
{
Deencapsulation.setField(work, "executor", executor);
// --- 将mock出来executor设置到work中
}
};

work.task();
}
}

另一种简单方式：

@RunWith(JMockit.class)
public class WorkTest {
@Tested
@Mocked
Work work;

@Test
public void testTask() {
final OtherVO otherVO = new OtherVO();
new Expectations(OtherVO.class) { // mock OtherVO的行为
{
otherVO.getValue();
result = "zero";
}
};
new Expectations() {
{
Deencapsulation.setField(work, "otherVO", otherVO);
}
};
// --- 通过以上方式可以将mock出来的otherVO设置到work中，当然可以采取其它简便方式

new MockUp<ThreadPoolExecutor>() {
@Mock
public void execute(Runnable command) {
command.run();
}
};

work.task();
}
}