easymock中提供对于类的mock功能,我们可以方便的mock这个类的某些方法,指定预期的行为以便测试这个类的调用者。这种场景下被mock的类在测试案例中扮演的是次要测试对象或者说依赖的角色,主要测试对象是这个mock类的调用者。但是有时候我们需要将这个测试类作为主要测试对象,我们希望这个类中的部分(通常是大部分)方法保持原有的正常行为,只有个别方法被我们mock掉以便测试。
1. 使用方法
我们先来看看这个partial class mocking 是如何工作的:
public class Service {
public void execute() {
actualMethod();
needMockMethod();
}
void actualMethod() {
System.out.println("call actualMethod()");
}
public void needMockMethod() {
System.out.println("call needMockMethod()");
}
}
我们给出了一个非常简单的类,我们将要测试execute()方法,期望能测试到actualMethod()这个方法的正常行为,然后需要mock掉needMockMethod().
public class PartialClassMockTest extends Assert {
@Test
public void testPartialMock() {
Service service = EasyMock.createMockBuilder(Service.class).addMockedMethod("needMockMethod").createMock();
service.needMockMethod();
EasyMock.expectLastCall();
EasyMock.replay(service);
service.execute();
EasyMock.verify(service);
}
}
上面的测试案例运行通过,输出为"call actualMethod()",没有"call needMockMethod()",说明我们设置的mock生效了。我们创建的mock类的确是只有部分我们制定的方法是mock的,其他都是正常行为。
再来看看为什么我们要需要partial class mocking 这个功能?为什么需要mock掉其中的一个方法?
我们来看看下面这个更加真实的例子:
public class Service {
public String execute2() {
return getConfiguration();
}
public String getConfiguration() {
return Configuration.getUsername();
}
}
public class Configuration {
public static String getUsername() {
//ignore the code to get configuration from file or database
return "username";
}
}
这里例子中,需要测试的 execute2()方法需要调用getConfiguration()方法,而getConfiguration()方法则调用了Configuration的静态方法来获取配置信息。我们假设读取配置的代码比较复杂不能直接在单元测试环境下运行,因此通过情况下这里的execute2()方法就会因为这个getConfiguration()而造成无法测试。因此我们可以考虑通过partial class mocking的功能来mock掉getConfiguration()方法从而使得我们的测试案例可以覆盖到execute2()方法
@Test
public void testStaticMethod() {
Service service = EasyMock.createMockBuilder(Service.class).addMockedMethod("getConfiguration").createMock();
EasyMock.expect(service.getConfiguration()).andReturn("abc");
EasyMock.replay(service);
assertEquals("abc", service.execute2());
EasyMock.verify(service);
}
这个测试案例可以正常通过,我们通过partial class mocking成功的避开了getConfiguration()这个绊脚石。
当然这里的实例代码本身就有点问题,应该采用DI的方法将configuration注入进来,而不是在内部通过静态方法来获取。因此一个建议是在使用partial class mocking功能前,先看看是不是可以通过重构来显改进测试类。只有当我们有足够充分的不得已的理由时,才使用partial class mocking这种变通(或者说取巧)的方式来解决问题。
2. 限制
上面两个例子中,我们仔细看看会发现,被mock的方法都是public的。我们试着将方法修改为protected和default,partial class mocking依然生效。但是修改为private之后,则抛出异常:
java.lang.IllegalArgumentException: Method not found (or private): needMockMethod
at org.easymock.internal.MockBuilder.addMockedMethod(MockBuilder.java:75)
at net.sourcesky.study.easymock.tutorial.PartialClassMockTest.testPartialMock(PartialClassMockTest.java:52)
或者将mock的方法继续保持public,但是加上final,则抛出以下异常:
java.lang.IllegalStateException: no last call on a mock available
at org.easymock.EasyMock.getControlForLastCall(EasyMock.java:521)
at org.easymock.EasyMock.expectLastCall(EasyMock.java:512)
at net.sourcesky.study.easymock.tutorial.PartialClassMockTest.testPartialMock(PartialClassMockTest.java:54)
我们回到之前的章节,class mocking里面讲述了class mocking的一些限制:private方法和final方法是不能mock的。partial class mocking下这些限制依然存在。因此,为了开启partial class mocking,我们不得不稍微破坏一下类的封装原则,对于原本应该是private的方法,修改为protected或者default。
不得不再次申明,partial class mocking不是一个足够好的解决方案,它只适合在不得已的情况下使用,不要太依赖这个特性。重构代码改善代码才是王道。
3. 疑问
另外class mocking中还讲到,对于类的equals(), toString()和hashCode()这三个方法,class mocking下是easymock为这三个方法内建了easymock的实现,因此也不能mock。而partial class mocking,这三个方法同样不能mock,但是easymock不再为它们内建实现,而是使用它们正常的功能。
关于这点还是有一点疑问,我在easymock的官方文档中看到以下描述
Remark: EasyMock provides a default behavior for Object's methods (equals, hashCode, toString). However, for a partial mock, if these methods are not mocked explicitly, they will have their normal behavior instead of EasyMock default's one.
言下之意,似乎equals, hashCode, toString这三个方法还是可以显式mock的。但是我测试了一下:
public class Service {
public String execute3() {
actualMethod();
return toString();
}
@Override
public String toString() {
return "defaultToString()";
}
}
@Test
public void testToStringMethod() {
Service service = EasyMock.createMockBuilder(Service.class).addMockedMethod("toString").createMock();
EasyMock.expect(service.toString()).andReturn("abc");
EasyMock.replay(service);
assertEquals("abc", service.execute3());
EasyMock.verify(service);
}
toString()方法的mock没能生效,抛出异常:
java.lang.IllegalStateException: no last call on a mock available
at org.easymock.EasyMock.getControlForLastCall(EasyMock.java:521)
at org.easymock.EasyMock.expect(EasyMock.java:499)
at net.sourcesky.study.easymock.tutorial.PartialClassMockTest.testToStringMethod(PartialClassMockTest.java:74)
可以看到明显是EasyMock.expect(service.toString()).andReturn("abc"); 这里的record没有成功。