springboot2 junit4测试controller

上来直接贴源码好了: 反正都是为了自己方便回来看 哈哈

import com.oohad.oohadshowmanager.mapper.AccountMapper;
import com.oohad.oohadshowmanager.model.Account;
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.mock.web.MockHttpSession;
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.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.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;

@SpringBootTest
@RunWith(SpringRunner.class)
public class ManagerControllerTest {

    @Autowired
    private ManagerController managerController;
    @Autowired
    private AccountMapper accountMapper;
    private MockMvc mockMvc;


    @Before
    public void setup(){
        mockMvc = MockMvcBuilders.standaloneSetup(managerController).build();
    }
    @Test
    public void activationBaiDu() throws Exception {
        // 如果需要设置seesion
        MockHttpSession session = new MockHttpSession();
        Account account = accountMapper.selectAccountById(1);
        session.setAttribute("user",account);
        // 请求头
        MultiValueMap<String, String> params = new LinkedMultiValueMap();
        params.add("fileId","32");
        params.add("userId","20");
        MockHttpServletRequestBuilder params1 = MockMvcRequestBuilders.post("/activationBaiDu").session(session).params(params);
        MvcResult mvcResult = mockMvc.perform(params1)
                .andExpect(MockMvcResultMatchers.status().isOk())
                .andDo(MockMvcResultHandlers.print())
                .andReturn();
        String contentAsString = mvcResult.getResponse().getContentAsString();
        System.out.println(contentAsString);
    }
}