如何编写单元测试-基于Spring

发表于:2018-10-17 11:44

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

 作者:atheva    来源:博客园

分享:
  单元测试
  首先单元测试真的算是一种“脏活累活”,但是我个人感觉还是有必要,至少本人最近开始写单元测试后还是能发现一些“bug”的。
  如何写单元测试
  单元测试的要求...网上很多。下面来分享一下我是如何写单元测试。首先我们项目一般都是MVC分层的,而单元测试主要是在Dao层和Service层上进行编写。从项目结构上来说,Service层是依赖Dao层的,但是从单元测试角度,对某个Service进行单元的时候,他所有依赖的类都应该进行Mock。而Dao层单元测试就比较简单了(下面Dao层就以Jdbc为例子),只依赖数据库中的数据。
  DAO层单元测试。
  dao层单元测试肯定要是连数据库的,但是不需要本地起一个Mysql,可以使用H2内存数据库来做单测就足够了。
  实战
  首先给一个通用模板吧,就是所有DAO层单元测试都会用到的,注:ORM使用Mybatis
  @RunWith(SpringJUnit4ClassRunner.class)public class XXXDaoTest {
  @MapperScan({"com.xxx.xxx.mapper"})
  @Configuration
  public static class MybatisScanConfiguration {
  @Bean
  public XXXDao xxxDao() {
  return new XXXDaoImpl();
  }
  @Bean
  public DataSource h2DataSource() {
  EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder();
  EmbeddedDatabase database = builder.setType(EmbeddedDatabaseType.H2)
  .addScript("classpath:/xxx/init_table.sql") //启动时初始化建表语句
  .build();
  return database;
  }
  @Bean
  public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
  final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
  sessionFactory.setDataSource(dataSource);
  PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
  //加载所有的sql mapper文件
  Resource[] mapperLocations = resolver.getResources("classpath:com/xxx/xxx/XXXXMapper.xml");
  sessionFactory.setMapperLocations(mapperLocations);
  return sessionFactory.getObject();
  }
  @Bean
  public PlatformTransactionManager platformTransactionManager(DataSource dataSource) {
  return new DataSourceTransactionManager(dataSource);
  }
  @Bean
  public JdbcTemplate jdbcTemplate(DataSource dataSource) {
  return new JdbcTemplate(dataSource);
  }
  }
  @Autowired
  XXXDao xxxDao
  @Test
  @Transactional
  public void xxxx() {
  }
  }
  其中的一些需要稍微修改的,我使用“XXX”或者“xxx”,如果配过Spring的肯定知道怎么弄。其中init_table.sql就是用来创建数据库表的,就不列出来的,下面需要注意的地方:
  H2 DataBase有些语法是不支持的,例如 insert ignore,还有类似在创建数据库时候"CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP"
  如果一些sql是Mapper不提供的,那么可以使用jdbcTemplate来执行
  每次单元测试执行完后,建议删除当前单元测试方法上,加上 @Transactional 注解,保证每次执行完后数据回滚
  如果当前dao层存在其他依赖,其实是可以mock的(可以参考后面Service层的单元测试)
  通用模板写好以后,就可以按照不同情况来写单测了,想想是不是很开心啊
  @Test
  @Transactional
  public void testUpdatexxx() {
  Long testUserId = 1L;
  //1.当数据不存在的时候,返回0
  int i = xxxDao.updateXxx(testUserId,1);
  Assert.assertTrue(i == 0);
  // 2.如果存在记录,状态是已经完成 返回0
  Domain domain = createBy(testUserId, 0);
  //插入一条数据
  mapper.insertSelective(domain);
  Assert.assertTrue(xxxDao.updateXxx(testUserId,1) == 0);
  //3.如果存在记录,状态是未完成 返回1
  ....
  }
  Service层单元测试
  说实话,Service层的单测是最麻烦的,特别是那种超级大方法。Service层的单元测试就是主要Mock其他依赖Bean来完成各种逻辑的判断。
  实战
  还是先给一个模板
  @RunWith(SpringRunner.class)
  @ContextConfiguration(classes = NeedTestService.Config.class)
  public class NeedTestServiceTest {
  @TestConfiguration
  public static class Config {
  //这里直接创建一个待测试的类
  @Bean
  public NeedTestService needTestService() {
  return new NeedTestService();
  }
  }
  /*
  * Mock掉依赖的Bean
  */
  @MockBean
  private X1Dao x1Dao;
  @MockBean
  private X1Service x1Service;
  //这里是Autowired需要测试的类
  @Autowired
  private NeedTestService needTestService;
  }
  通用模板在上面,当然,遇到恶心的类,Mock十几个Bean也是有可能的。 下面就开始开心的写单元测试的了,例如
  @Test
  public void testCalculateUserClaims1() {
  Long testUserId = 1L;
  Date transTime = DateUtil.getStartDate(new Date());
  given(x1Dao.insertInitIgnore(testUserId, transTime)).willReturn(1);
  List<X2> list = Lists.newArrayList();
  list.addAll(createTenderCell(1L, "100.00", 10));
  list.addAll(createTenderCell(2L, "200.00", 5));
  Collections.shuffle(list);
  given(x1Dao.selectxxx(testUserId, transTime)).willReturn(list);
  ... 把改情况下 各种依赖的方法调用返回mock出来,
  //这里直接调用被测试的方法
  Result result = needTestService.needTestMethod(...)
  Assert.assertTrue(result != null);
  //... 省略掉各种 Assert.
  }
  注:一般单元测试可以分为:生成数据,mock数据,假设校验
  当然,这仅仅是一种情况,还有各种情况需要您进行测试,good luck!
  小结
  我也是最近才开始写单元测试的,虽然很无脑,但是我个人还是觉得有必要的,一些很低级的错误至少能找出来(如果你稍微认真的写,而不是敷衍)。这样总比在Code Review的时候被指出来要好很多吧。而且写单元测试也算是一种休息吧!

   上文内容不用于商业目的,如涉及知识产权问题,请权利人联系博为峰小编(021-64471599-8017),我们将立即处理。
《2023软件测试行业现状调查报告》独家发布~

关注51Testing

联系我们

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

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

沪ICP备05003035号

沪公网安备 31010102002173号