1. 程式人生 > >SpringBoot Controller單元測試MockMvc

SpringBoot Controller單元測試MockMvc

直接看程式碼
package com.lemon.proudct.test;

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.http.MediaType;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.web.WebAppConfiguration;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.RequestBuilder;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;

import com.alibaba.fastjson.JSON;
import com.lemon.product.ProductApplication;

/**
*
* @author jiangqk
* @data 2018年1月16日 下午5:23:41
*/
@RunWith(SpringJUnit4ClassRunner.class)
@WebAppConfiguration
@SpringBootTest(classes = ProductApplication.class)//Application啟動類
public class MockTest {

@Autowired
private WebApplicationContext context;

private MockMvc mockMvc;

@Before
public void setUp() {
    //單個類,專案攔截器無效
//      mvc = MockMvcBuilders.standaloneSetup(new ProductController()).build();
        //專案攔截器有效
        mockMvc = MockMvcBuilders.webAppContextSetup(context).build();
    }

@Test
public void execute() throws  Exception {
    //呼叫介面,傳入新增的使用者引數
    RequestBuilder request = MockMvcRequestBuilders.post("/product/ec")
            .contentType(MediaType.APPLICATION_JSON_UTF8)  
            .content(JSON.toJSONString(null));

     MvcResult mvcResult = mockMvc.perform(request).andReturn() ;
     int status = mvcResult.getResponse().getStatus();  
     String content = mvcResult.getResponse().getContentAsString();
     Assert.assertTrue("正確", status == 200);  
     Assert.assertFalse("錯誤", status != 200);  
     System.out.println("返回結果:"+status);
     System.out.println(content);
}

}