SpringBoot Junit Demo

package com.yunen.camera.test;

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.mock.web.MockHttpSession;
import org.springframework.test.annotation.Rollback;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.ResultActions;
import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.result.MockMvcResultHandlers;
import org.springframework.test.web.servlet.result.MockMvcResultMatchers;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;

import com.YunenCameraManagerApplication;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.serializer.SerializerFeature;

import lombok.extern.slf4j.Slf4j;

/**
 * 相机预置位测试类
 * 
 * @version V1.0
 * @author songxiaotong
 * @date 2018年9月17日 下午2:08:14
 * @Description
 */
@Slf4j
@Rollback(false)
// 让 JUnit 运行 Spring 的测试环境， 获得 Spring 环境的上下文的支持
@RunWith(SpringRunner.class)
// 获取启动类，加载配置，确定装载 Spring 程序的装载方法，它回去寻找 主配置启动类（被 @SpringBootApplication 注解的）
@SpringBootTest(classes = YunenCameraManagerApplication.class)
public class CameraPreSetControllerTest {
    /**
     * WebApplicationContext
     */
    @Autowired
    private WebApplicationContext wac;

    private MockMvc mockMvc;

    /**
     * 初始化mvc请求对象
     * 
     * @throws Exception
     */
    @Before
    public void setUp() throws Exception {
        this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).build();
    }

    /**
     * 新增
     * 
     * @author songxiaotong
     */
    @Test
    public void create() {
        try {
            MockHttpSession session = new MockHttpSession();

            JSONObject param = new JSONObject();
            param.put("cameraId", "110");
            param.put("name", "预置位名称");
            param.put("zoom", 1222.1);
            param.put("pan", 2222.2);
            param.put("tilt", 3222.3);

            MockHttpServletRequestBuilder requestBuilder = MockMvcRequestBuilders
                    .post("/yunen/camera_pre_set/create");
            requestBuilder.contentType(MediaType.APPLICATION_JSON);

            requestBuilder.content(JSONObject.toJSONString(param,
                    SerializerFeature.WriteMapNullValue));
            // requestBuilder.content(param.toString());

            requestBuilder.session(session);

            // 执行一个请求；
            ResultActions result = mockMvc.perform(requestBuilder);

            // 添加执行完成后的断言
            result.andExpect(MockMvcResultMatchers.status().isOk());

            // 添加一个结果处理器，表示要对结果做点什么事情，比如此处使用MockMvcResultHandlers.print()输出整个响应结果信息。
            result.andDo(MockMvcResultHandlers.print());

            // 表示执行完成后返回相应的结果。
            MvcResult mvcResult = result.andReturn();

            log.info(mvcResult.getResponse().getContentAsString());
        } catch (Exception e) {
            log.error(e.getMessage());
        }
    }

    /**
     * 更新
     * 
     * @author songxiaotong
     */
    @Test
    public void update() {
        String url = "/yunen/camera_pre_set/update";
        JSONObject param = new JSONObject();
        param.put("id", 1);
        param.put("cameraId", "110");
        param.put("name", "更新后的预置位名称");
        param.put("zoom", 1222.1);
        param.put("pan", 2222.2);
        param.put("tilt", 3222.3);
        // 调用测试模拟方法
        requestMock(url, param);
    }

    /**
     * http请求模拟
     * 
     * @param url 请求地址
     * @param param 请求参数
     * @author songxiaotong
     */
    private void requestMock(String url, JSONObject param) {
        try {
            MockHttpSession session = new MockHttpSession();

            MockHttpServletRequestBuilder requestBuilder = MockMvcRequestBuilders
                    .post(url);
            requestBuilder.contentType(MediaType.APPLICATION_JSON);

            requestBuilder.content(JSONObject.toJSONString(param,
                    SerializerFeature.WriteMapNullValue));
            // requestBuilder.content(param.toString());

            requestBuilder.session(session);

            // 执行一个请求；
            ResultActions result = mockMvc.perform(requestBuilder);

            // 添加执行完成后的断言
            result.andExpect(MockMvcResultMatchers.status().isOk());

            // 添加一个结果处理器，表示要对结果做点什么事情，比如此处使用MockMvcResultHandlers.print()输出整个响应结果信息。
            result.andDo(MockMvcResultHandlers.print());

            // 表示执行完成后返回相应的结果。
            MvcResult mvcResult = result.andReturn();

            log.info(mvcResult.getResponse().getContentAsString());
        } catch (Exception e) {
            log.error(e.getMessage());
        }
    }
}