上面是个基本代码框架,这次我挑了个硬骨头main_process方法来演示一下如何使用mock,也是该类的核心方法。为什么说是硬骨头,因为:
1、首先是个private方法
2、调main_process之前,框架自动调用setup方法根据上下文进行各种初始化,然后才启用map。为了不依赖外部环境,就必须mock上下文,mock setup过程。
3、在setup方法中,会初始化HTable对象,而这又需要连上对应zookeeper和hbase环境。所以又必须mock HTable类。
4、在setup方法中,还需要获取Configuration,这个又需要mock。
5、在setup方法中,还有该死的Counter,我之前尝试new,发现不对,构造函数是protected。所以在前面都mock的前提下,你又必须mock Counter类。
6、在main_process方法中,这里需要从HTable对象中拿到数据。因为前面mock了HTable类,所以这里还需要mock get方法,不然数据是拿不到的。
可能有人问,为什么不直接mock好Context对象,一次性搞定。实际上,我试过,但是Context对象的构造也是非常头疼的,而且很多类是无法简单mock,搞了半天后,我放弃了,还是用最笨的方式。后面我会再想办法尝试尝试。
再看看测试代码
下面一条条看如何mock,上测试代码。
1、Private方法我是偷懒了,直接在源代码中注入了public int mainProcess方法,该方法只一句调用main_process,这样我能间接的获取该方法的调用。通过PowerMock是能够mock私有方法的,不过很多人不推荐,担心有隐患。
2、在使用powermock之前,需要在测试类前加入
@RunWith(PowerMockRunner.class) @PrepareForTest({ ShopScanMap.class }) public class TestMockito { @Test public void test01() throws Exception {…} |
对setup方法涉及到的几个基本外部依赖类要进行mock
Mapper.Context context = mock(Mapper.Context.class); Counter counter = mock(Counter.class); Configuration config = mock(Configuration.class); |
3、对HTable进行mock,然后初始化shop_info对象。接着对构造函数进行mock,在thenReturn中期望new之后返回的是mock后的HTable对象。PowerMockito的方法还是很好理解的,whenNew表示调用new方法时,withArguments表示构造函数的参数,thenReturn表示期望返回的值。
HTable shop_info = PowerMockito.mock(HTable.class); PowerMockito.whenNew(HTable.class) .withArguments(config, "sp_shopinfo".getBytes()) .thenReturn(shop_info); |
4、因为setup方法中调用了getConfiguration来获取配置,所以必须mock,才能消除对外部配置的依赖
when(context.getConfiguration()).thenReturn(config); |
5、原因同getConfiguration。不过不一样的是,这里需要带上模拟的参数。至于参数的值,我是从被测试代码中分析出的数据。
when(context.getCounter("Statistics", "sp_bmw_shops")).thenReturn(counter); |