MRUnit - MapReduce单元测试框架


Apache
跨平台
Java

软件简介

MRUnit是由Couldera公司开发的专门针对
Hadoop中编写MapReduce单元测试的框架,基本原理是JUnit4和
EasyMock。MR就是Map和Reduce的缩写。MRUnit框架非常精简,其核心的单元测试依赖于JUnit。而且MRUnit实现了一套
Mock对象来控制OutputCollector的操作,从而可以拦截OutputCollector的输出,和我们的期望结果进行比较,达到自动断言 的目的。

有了MRUnit,对MR程序做重构的时候,只要明确输入和输出,就可以写出单元测试,并且在放到群集校验前进行试验,从而节省时间和资源,也
能更快的定位到问题。而进行重构的话,只要写得足够详细的单元测试都是绿色的话,那么基本就可以保证在群集运行的结果也是正常的。

MRUnit不在Apache标准的Hadoop的发行版中,而是在Couldera公司的增强版本中hadoop-
0.20.1+133.tar.gz的contrib\mrunit\hadoop-0.20.1+169.56-mrunit.jar,已经贴在附件
中。只要把它和Junit4的jar添加到Hadoop程序项目的classpath中,就可以使用MRUnit了。

MRUnit包含四种
Driver:MapDriver,ReduceDriver,MapReduceDriver,PipelineMapReduceDriver。可以
根据自己的需要选择合适的Driver。

给出一个Reduce的很简单例子,Reduce的逻辑就是把Value中的各个值相加。

public class ExtractKeywordTest {

private Reducer reducer;

private ReduceDriver reduceDriver;

@Before

public void setUp() throws Exception {

reducer = new ExtractKeywordAcookie.Reduce();

reduceDriver = new ReduceDriver(reducer);

}

@Test

public void testReduce() {

List values = new ArrayList();

values.add(new Text(1.0_0.1));

values.add(new Text(2.0_0.2));

values.add(new Text(3.0_0.3));

reduceDriver.withInput(new Text(20100106_00_IBM), values)

.withOutput(new Text(20100106_00_IBM_6.00_0.60_), null)

.runTest();

}