使用h2数据库做单元测试

单纯的使用Mockito模拟接口返回数据，并不能满足我们测复杂sql的需要。

代码结构如下：

src/test/java
    --Test.java
src/test/resources
    --application.yml        ->根据就近原则，执行测试用例时，会使用test中的这个配置文件
    --schema.sql             ->h2数据库建表语句脚本
src/test/resources/sql
    --table1_init_data.sql
    --table2_init_data.sql   ->表初始化数据

yml配置：

spring:
  datasource:
    url: "jdbc:h2:mem:test_db;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=TRUE;MODE=MSSQLServer;USER=sa;PASSWORD=0000abc"
    driver-class-name: org.h2.Driver
    username: sa
    password: 0000abc
    schema: classpath:schema.sql     #执行建表语句
    data: classpath:sql/*.sql        #执行数据初始化

测试用例：

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@AutoConfigureMockMvc
public class Test {
    @Autowired
    MockMvc mockMvc;
    @Test
    public void test01(){     mockMvc.perform(post("/XXX").accept(MediaType.APPLICATION_JSON).contentType(MediaType.APPLICATION_JSON)
                .content(requsetJson.getBytes()))
                .andExpect(status().isOk())
                .andExpect(content().contentType(MediaType.APPLICATION_JSON))
                .andExpect(content().json(new JSONObject()
                        .put("code", "200")
                        .put("message", "success").toString()))
                .andDo(print()); //输出MvcResult到控制台
    }
}

这样就可以测试整个代码流程，如果某些测试需要用到特殊的数据，可以指定sql来初始化数据：

    @Test
    @Sql(value = {
            "/truncate_all.sql", // truncate脚本
            "/sql/xxx.sql", //数据初始化脚本
            "/sql/xxx.sql"
    })

使用h2数据库做单元测试

猜你喜欢