SPRING-Mock方式进行单元测试

编写测试Controller：

package com.wartao.test.mock.controller;


import java.util.ArrayList;
import java.util.Collections;
import java.util.List;


import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;


/**
 * @author ignore1992
 *
 */
@RestController
@RequestMapping(value = "/simple")
public class SimpleController
{
/**
* 存放元素的集合
*/
private List<String> demos = Collections.synchronizedList(new ArrayList());

@RequestMapping(value = "/{value}", method = RequestMethod.PUT)
public String addSimple(@PathVariable String value)throws Exception
{
demos.add(value);
return "success";
}

@RequestMapping(value = "/{index}", method = RequestMethod.GET)
public String getSimple(@PathVariable int index)throws Exception
{
return demos.get(index);
}

@RequestMapping(value = "/{key}", method = RequestMethod.DELETE)
public String deleteSimple(String key)throws Exception
{
demos.remove(key);
return "success";
}

}

编写单元测试类：

package com.wartao.test.mock.test;


import static org.hamcrest.Matchers.equalTo;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.put;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.delete;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;


import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.mock.web.MockServletContext;
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.RequestBuilder;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;


import com.wartao.test.mock.controller.SimpleController;




/**
 * @author ignore1992
 *
 */
@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(classes = MockServletContext.class)
@WebAppConfiguration
public class TestMock
{
private MockMvc mvc;

@Before
public void setUp()throws Exception

{

                //创建Mock实例

                mvc = MockMvcBuilders.standaloneSetup(new SimpleController()).build();

}

@Test
public void testSimpleController()throws Exception
{
//测试SimpleController
RequestBuilder request = null;

//使用put方法需要手工导入静态包
request = put("/simple/test");
//添加元素
mvc.perform(request).andExpect(status().isOk()).andExpect(content().string(equalTo("success")));

//获取一个index=0的元素
request = get("/simple/0");
mvc.perform(request).andExpect(status().isOk()).andExpect(content().string(equalTo("test")));

//删除值为test的元素
request = delete("/simple/test");
mvc.perform(request).andExpect(status().isOk()).andExpect(content().string(equalTo("success")));
}

}

注：

1、在单元测试类要使用put、get、delete等等静态方法，需要手工导入对应的静态类

2、Controller类中的方法要写上对应的请求方式，例如method=RequestMethod.PUT