快捷導(dǎo)航

SpringBoot對數(shù)據(jù)訪問層進(jìn)行單元測試的方法詳解

更新時間：2023年08月14日 10:54:12 作者：Young丶

我們公司作為一個面向銀行、金融機(jī)構(gòu)的TO B類企業(yè),頻繁遇到各個甲方爸爸提出的國產(chǎn)化數(shù)據(jù)庫的改造需求,包括OceanBase, TiDB,geldenDB等等,本文就介紹一種快高效、可復(fù)用的解決方案——對數(shù)據(jù)訪問層做單元測試,需要的朋友可以參考下

Mybatis測試依賴

我們項(xiàng)目采用的 SpringBoot + Mybatis 作為開發(fā)框架，大家第一想到可以用 SpringBoot 自帶的測試注解 @SpringBootTest 進(jìn)行測試。但是使用該注解有一個最大的弊端就是需要啟動整個容器，注入全部的bean，那么一次測試就相當(dāng)于啟動一次應(yīng)用，我們的應(yīng)用啟動一次就要花費(fèi)近70~80秒，黃花菜都涼了。

那么有沒有更好的辦法，有必要注入全量的bean嗎？是不是只要注入數(shù)據(jù)訪問層相關(guān)的bean即可，其實(shí)官方的 mybatis 就給了我們這樣的的解決方案。

引入依賴 mybatis-spring-boot-starter-test

<dependency>
  <groupId>org.mybatis.spring.boot</groupId>
  <artifactId>mybatis-spring-boot-starter-test</artifactId>
  <version>2.3.0</version>
  <scope>test</scope>
</dependency>

Mapper接口

@Mapper
public interface CityMapper {
    @Select("SELECT * FROM CITY WHERE state = #{state}")
    City findByState(@Param("state") String state);
}

使用junit5的測試類

// 使用junit5
@MybatisTest
// 使用真實(shí)的數(shù)據(jù)源進(jìn)行測試
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)
public class CityMapperTest {
    @Autowired
    private CityMapper cityMapper;
    @Test
    public void findByStateTest() {
        City city = cityMapper.findByState("CA");
        assertThat(city.getName()).isEqualTo("San Francisco");
        assertThat(city.getState()).isEqualTo("CA");
        assertThat(city.getCountry()).isEqualTo("US");
    }
}

自定義一個啟動類

@MybatisTest 在默認(rèn)情況下將會探測到帶有 @SpringBootApplication 的類。因此，由于 bean 定義的一些方法，可能會發(fā)生一些意想不到的錯誤，或者一些不必要的組件被裝入 ApplicationContext 。為了避免這種情況，我們可以在與測試類相同的包中創(chuàng)建帶有 @SpringBootApplication 的類。

package sample.mybatis.mapper;
import org.springframework.boot.autoconfigure.SpringBootApplication;
@SpringBootApplication
class MapperTestApplication {
}

運(yùn)行結(jié)果

但是， 上面是使用原生mybatis的測試方式，而我們項(xiàng)目用的是基于mybatis封裝的開源框架 tkMapper ， github 地址是 https://github.com/abel533/Mapper ，并不適用啊，無解，只能去看下 mybatis-spring-boot-starter-test 的原理。