上面是个基本代码框架,这次我挑了个硬骨头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);