基于hadoop的单元测试mock技术分享

发表于:2012-12-25 11:31

字体: | 上一篇 | 下一篇 | 我要投稿

 作者:赵烈    来源:51Testing软件测试网采编

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

32/3<123>
《2023软件测试行业现状调查报告》独家发布~

精彩评论

  • cathleen_417
    2012-12-26 10:16:39

    表示看不太懂,因为没做过呢

关注51Testing

联系我们

快捷面板 站点地图 联系我们 广告服务 关于我们 站长统计 发展历程

法律顾问:上海兰迪律师事务所 项棋律师
版权所有 上海博为峰软件技术股份有限公司 Copyright©51testing.com 2003-2024
投诉及意见反馈:webmaster@51testing.com; 业务联系:service@51testing.com 021-64471599-8017

沪ICP备05003035号

沪公网安备 31010102002173号