SpringBoot單元測試

針對不同的代碼層次, 有不同的測試注意事項.

我的理解是:

針對Mapper層或者是Dao層,由於需要跟數據庫交互, 寫完代碼的時候手工執行就好, 由於數據庫的數據的不穩定性,不適合用作大量回歸測試.

針對Service層的單元測試才是我們通常理解的那種單元測試或者說白盒測試.

針對Controller層的測試,如果Mock了Service層的行為,那這個算是白盒測試,如果沒有Mock Service層行為,而是直接調用,我覺得其實已經可以算是集成測試了,這個偏向黑盒測試的范疇,就是我們通常說的API測試.

 

1. 針對Mapper層或者Dao層

如果是有依賴其他Bean對象,可以先寫一個專門用來裝配Bean的類,並用注解TestConfiguration標注.

當然,這一步也不是必須的,這個只是為了方便在單元測試類中使用@Autowired引入對象.可以在單元測試類中,使用Product product = new Product()代替.

import org.springframework.boot.test.context.TestConfiguration;
import org.springframework.context.annotation.Bean;

import com.windy.mall.product.bean.Product;

@TestConfiguration
public class TestBeanConfiguration {

    @Bean
    public Product createProduct(){
        return new Product();
    }
    
}

 

單元測試類中必須要聲明:

(1) @RunWith(Spring.class),表示要在Spring環境中做測試, 於是就可以使用@Autowired等注解了,

(2) @SpringBootTest(classes=TestBeanConfiguration.class),表示可以使用SpringBoot環境了,這樣可以用@Autowired注解,把@Mapper注釋過的Mapper對象引入.為了引入外部依賴的Bean對象,需要指定classes=TestBeanConfiguration.class.

注意: 之所以要指定classes=TestBeanConfiguration.class,這時因為Product這個類,並沒有使用@Component等注解注入IOC容器.

可選:

(3) @Transactional注解來開啟事務

(4) @Rollback(true)表示開啟回滾數據功能,默認為True.

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.annotation.Rollback;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.transaction.annotation.Transactional;

import com.windy.mall.product.bean.Product;
import com.windy.mall.product.bean.TestBeanConfiguration;

@RunWith(SpringRunner.class)
@SpringBootTest(classes=TestBeanConfiguration.class)
@Transactional
@Rollback(true)
public class ProductMapperTest {

    @Autowired
    private ProductMapper mapper;
    
    @Autowired
    private Product product;
    
    @Before
    public void setUp(){
        product.setPname("持續交付");
        product.setType("書籍");
        product.setPrice(69d);
    }
    
    @Test
    public void testAdd() {
        Assert.assertEquals(Integer.valueOf(1), mapper.add(product));
    }

}

 

 

2. 針對Service層

由於Service層需要調用Mapper層或者是Dao層,通常需要Mock Mapper對象.這里需要使用到@MockBean注解,Mock Mapper層的行為.

package com.windy.mall.product.service;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.BDDMockito;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.test.context.junit4.SpringRunner;

import com.windy.mall.product.bean.Product;
import com.windy.mall.product.bean.TestBeanConfiguration;
import com.windy.mall.product.mapper.ProductMapper;

@RunWith(SpringRunner.class)
@SpringBootTest(classes = TestBeanConfiguration.class)
public class ProductServiceTest {

    @MockBean
    private ProductMapper mapper;

    @Autowired
    private Product product;

    @Autowired
    private ProductService service;

    @Before
    public void setUp() {
        BDDMockito.given(mapper.add(product)).willReturn(1);
        BDDMockito.given(mapper.add(null)).willReturn(0);
    }

    @Test
    public void testAddService() {
        Assert.assertEquals(Integer.valueOf(1), service.add(product));
        Assert.assertEquals(Integer.valueOf(0), service.add(null));
    }

}

 

3.針對Controller

有兩種方法可以做單元測試:

(1) 這里需要用到web環境(webEnvironment=WebEnvironment.RANDOM_PORT).需要用到TestRestTemplate這個對象作為客戶方,調用Controller(API)

import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
import org.springframework.boot.test.web.client.TestRestTemplate;
import org.springframework.test.context.junit4.SpringRunner;

@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment=WebEnvironment.RANDOM_PORT)
public class ProductControllerTest {

    @Autowired
    private TestRestTemplate template;
    
    @Test
    public void test() {
        String body = template.getForObject("/ms/product/1", String.class);
        Assert.assertEquals("success", body);
    }

}

 

(2) 如果是使用MockMVC對象來測試,需要開啟@AutoConfigureMockMvc注解

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.result.MockMvcResultMatchers;

@RunWith(SpringRunner.class)
@SpringBootTest
@AutoConfigureMockMvc
public class ProductControllerTest2 {

    @Autowired
    private MockMvc mvc;

    @Test
    public void test() throws Exception {
        mvc.perform(MockMvcRequestBuilders.get("/ms/product/1"))
                .andExpect(MockMvcResultMatchers.status().isOk())
                .andExpect(MockMvcResultMatchers.content().string("success"));
    }

}