如何編寫單元測試-基於Spring

單元測試

首先單元測試真的算是一種「髒活累活」，可是我我的感受仍是有必要，至少本人最近開始寫單元測試後仍是能發現一些「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就是用來建立數據庫表的，就不列出來的，下面須要注意的地方：

1. H2 DataBase有些語法是不支持的，例如 insert ignore，還有相似在建立數據庫時候"CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP"
2. 若是一些sql是Mapper不提供的，那麼可使用jdbcTemplate來執行
3. 每次單元測試執行完後，建議刪除當前單元測試方法上，加上 @Transactional 註解，保證每次執行完後數據回滾
4. 若是當前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的時候被指出來要好不少吧。並且寫單元測試也算是一種休息吧！