单纯的使用Mockito模拟接口返回数据,并不能满足我们测复杂sql的需要。
所以,使用h2数据库是个不错的选择。
代码结构如下:
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"
})
|