使用EasyMock扩展为Class提供Mock对象

使用EasyMock扩展为Class提供Mock对象

Author:kongxx

通常我么在测试一些类的时候需要依赖另外一些类或接口，而此时可能这些类或接口可能仅仅是方法签名而没有真正的实现。此时我们就可以使用一些第三方类库来解决这个问题，比如常用的EasyMock，JMock等。但是这两个类库默认情况下都只能Mock接口，而不能Mock类，那么我们应该怎样解决这个问题呢？其实EasyMock和JMock两个框架也提供了对类作Mock的解决办法，不过这个功能需要他们提供的一个扩展类库才能实现。我们可以在他们的官方网站上下载这两个扩展。对于EasyMock下载EasyMock Class Extension 2.0_Pre-Release，对于JMock下载 jMock/CGLIB extension binary JAR。
下面是一个小例子:
Worker.java一个需要测试的业务类,它依赖一个抽象的Configuration对象.

class package easymock;

public class Worker {

private Configuration configuration;

public Configuration getConfiguration() {
return configuration;
}

public void setConfiguration(Configuration configuration) {
this.configuration = configuration;
}

public boolean execute() {
String name = configuration.getName();
String type = configuration.getType();

if(name.equals("System") && type.equals("Snapshot")) {
// TODO

return true;
}
return false;
}
}

{
public static void main(String[] args)
{
System.out.println("Hello World!");
}
}

Configuration.java一个抽象类,会有一些子类来实现它的抽象方法.

package easymock;

import java.io.File;

public abstract class Configuration {

protected String name;
protected String type;

public abstract Configuration configurate(File file);

public abstract String getName();

public abstract String getType();
}

WorkerTest.java一个测试类,需要用到Configuration对象实例,这里Mock了一个Configuration对象用来为execute方法提供支持.

package easymock;

import junit.framework.TestCase;
import org.easymock.classextension.*;

public class WorkerTest extends TestCase {
private Worker worker;
protected void setUp() throws Exception {
super.setUp();

worker = new Worker();
}

protected void tearDown() throws Exception {
super.tearDown();
}

/*
* Test method for 'easymock.Worker.execute()'
*/
public void testExecute() {
Configuration conf = (Configuration)EasyMock.createMock(Configuration.class);
EasyMock.expect(conf.getName()).andReturn("System");
EasyMock.expect(conf.getType()).andReturn("Snapshot");

EasyMock.replay(conf);

worker.setConfiguration(conf);

// 执行带测试的方法
assertTrue(worker.execute());
}

}

运行JUnit测试，GreenBar。OK！