java~springboot~h2數據庫在單元測試中的使用

單元測試有幾點要說的

事實上springboot框架是一個tdd框架，你在進行建立項目時它會同時建立一個單元測試項目，而我們的代碼用例可以在這個項目里完成，對於單元測試大叔有以下幾點需要說明一下：

1. 單元測試的用例之間不要有相互依賴
2. 單元測試數據來源為本地，不要訪問外部資源，外部數據庫也是不行的
3. 對於集成測試，每個控制器對應一個測試類，它們可以有統一的抽象基類，用來存儲公用的屬性，如數據對象，http對象等

引入相關依賴包

testCompile('com.h2database:h2')

下面對h2數據庫的配置

spring:
  profiles: integTest
  cloud.config.enabled: false
  h2:
    console:
      enabled: true
      path: /h2
  datasource:
    url: jdbc:h2:mem:testdb;MODE=MYSQL;DB_CLOSE_DELAY=-1;DATABASE_TO_UPPER=false
    driver-class-name: org.h2.Driver
    username: sa
    password: sa
    schema-username: sa
    schema-password: sa
    data-username: sa
    data-password: sa
    schema: classpath:db/*.sql
    data: classpath:data/*.sql
    initialization-mode: always
    platform: h2

下面直接寫單元測試即可，業務層不用修改，數據庫根據profile去選擇mysql還是h2

@RunWith(SpringRunner.class)
@ActiveProfiles("integTest")
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public abstract class ControllerTestBase {

  protected MockMvc mockMvc;

  @Autowired
  protected ObjectMapper objectMapper;

  @Autowired
  private WebApplicationContext webApplicationContext;

  @Before
  public void setUp() {
    mockMvc = MockMvcBuilders.webAppContextSetup(webApplicationContext).build();
  }


public class CustomerTagsControllerTest extends ControllerTestBase {
  @Autowired
  CustomerTagsService customerTagsService;

  @Test
  public void getCustomerTags() throws Exception {
    mockMvc.perform(
        get("/api/tags")
            .accept(MediaType.APPLICATION_JSON_UTF8))
        .andExpect(status().isOk())
        .andExpect(jsonPath("$[0].tagsDescription").value("未接"));
  }
}

對於h2在單元測試里的使用就說這么說，有不清楚的可以給大叔留言！