首页 > 代码库 > springboot 添加测试模块

springboot 添加测试模块

@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes = Application.class)
@WebAppConfiguration
@ActiveProfiles("dev,common,system")
public class BaseTest {
 @Autowired
    public WebApplicationContext context;

    public MockMvc mvc;


  @Before
    public void setUp() throws Exception {
        this.mvc = MockMvcBuilders.webAppContextSetup(this.context).build();
    }

    public static MockHttpServletResponse getResponse(MockMvc mvc, MockHttpServletRequestBuilder builder,
            HttpHeaders httpHeaders, String content) throws Exception {
        builder.contentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
        if (httpHeaders != null) {
            builder.headers(httpHeaders);
        }
        if (!StringUtils.isEmpty(content)) {
            builder.content(content);
        }
        return getResponse(mvc, builder);
    }

    public static MockHttpServletResponse getResponse(MockMvc mvc, MockHttpServletRequestBuilder builder)
            throws Exception {
        ResultActions resultActions = mvc.perform(builder);
        resultActions.andDo(MockMvcResultHandlers.print());
        MvcResult result = resultActions.andReturn();
        resultActions.andExpect(status().isOk());
        MockHttpServletResponse response = result.getResponse();
        return response;
    }
}

 

springboot 添加测试模块