maven依赖
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>
目录结构
Controller
package com.example.demo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;
import java.util.HashMap;
import java.util.Map;
/**
* @description:
* @author: Yejunwei
* @date: Created in 14:33 2017/10/19
*/
@RestController
@RequestMapping("test")
public class BaseController {
private static Logger logger = LoggerFactory.getLogger(BaseController.class);
@RequestMapping(value = "get",method = RequestMethod.GET)
public Map getMap(){
Map map = new HashMap(1);
map.put("test", "test");
return map;
}
}
对应的测试类
package com.example.demo;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.internal.Classes;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.SpringBootConfiguration;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.http.MediaType;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.context.web.WebAppConfiguration;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MockMvcBuilder;
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.result.MockMvcResultMatchers;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;
import javax.servlet.ServletContext;
import static org.junit.Assert.*;
/**
* @description:
* @author: Yejunwei
* @date: Created in 14:36 2017/10/19
*/
@RunWith(SpringRunner.class)//
@SpringBootTest(classes = DemospringbootApplication.class)
@WebAppConfiguration
public class BaseControllerTest {
@Autowired
private WebApplicationContext context;
private MockMvc mockMvc;
@Before
public void setUp() throws Exception {
this.mockMvc = MockMvcBuilders.webAppContextSetup(context).build();
}
@Test
public void getMap() throws Exception {
MvcResult result = mockMvc.perform(MockMvcRequestBuilders.get("/test/get").contentType(MediaType.APPLICATION_JSON_UTF8))
.andExpect(MockMvcResultMatchers.status().isOk())
.andReturn();
int statusCode = result.getResponse().getStatus();
Assert.assertEquals(statusCode, 200);
}
}
运行即可