Spring 也提供了完善的测试框架,我们可以方便的测试Spring Web MVC应用程序。为了使用这个测试框架,我们需要添加它的依赖项。
compile group: 'org.springframework', name: 'spring-test', version: '4.3.6.RELEASE'
服务端测试
我们可以利用Spring提供的Mock对象来测试我们Spring程序的服务端行为。通过这些Mock对象,我们可以建立一个假的服务器,然后发送一些假的请求,来测试我们的程序。为了能简洁的编写测试代码,我们最好在代码中使用静态导入将MockMvcRequestBuilders.*
、MockMvcResultMatchers.*
和MockMvcBuilders.*
引入到代码中。
建立测试环境
建立Spring Web MVC的测试环境和普通的Spring 单元测试略有不同。我们需要使用@WebAppConfiguration注解测试类。Spring知道这是一个Web MVC测试之后,就会使用@ContextConfiguration注解中的配置文件来创建一个WebApplicationContext,然后我们可以将其注入到测试类中。然后要做的事情就是创建MockMvc对象,我们大部分测试都要通过该对象进行。
@RunWith(SpringRunner.class)
@WebAppConfiguration
@ContextConfiguration("file:src/main/webapp/WEB-INF/applicationContext.xml")
public class UserControllerTest {
@Autowired
private WebApplicationContext context;
private MockMvc mvc;
@Before
public void init() {
mvc = MockMvcBuilders.webAppContextSetup(context).build();
}
}
当然,如果只需要测试某个控制器,我们完全可以不加载完整的配置文件。这时候可以使用MockMvcBuilders.standaloneSetup来仅使用Spring默认配置配置某个控制器。
public class SimpleTests {
private MockMvc mockMvc;
@Before
public void setup() {
this.mockMvc = MockMvcBuilders.standaloneSetup(new AccountController()).build();
}
}
发起请求
这里假定代码中已经静态导入上面提到的一些类。
我们使用MockMvc的perform方法发起一个HTTP请求,这个请求可以是get、post等,然后我们还可以为请求设置accept等信息。
mockMvc.perform(post("/users/{id}", 42).accept(MediaType.ALL));
当然也可以发起文件上传请求。
mockMvc.perform(fileUpload("/upload").file("file", file.getBytes("UTF-8")));
我们可以直接在请求中包含参数。
mockMvc.perform(get("/users?user={foo}", "bar"));
也可以使用param方法传递参数,这种方式可以传递POST表单数据。
mockMvc.perform(post("/users").param("foo", "bar"));
如有需要,我们还可以为请求添加contextPath和servletPath。
mockMvc.perform(get("/myproject/contextpath/users").contextPath("/myproject").servletPath("/contextpath"))
期望结果
发起请求之后,我们需要验证请求是否正确处理。这时候需要在perform方法之后再调用andExpect方法。我们可以期望获得各种结果,最常用的就是获得各种响应码。下面的例子期望首页可以正常访问。当然status()方法也提供了其他了响应码方法来满足我们的需求。
mockMvc.perform(get("/index")).andExpect(status().isOk());
还可以期望结果的媒体类型。
mvc.perform(get("/users.xml"))
.andExpect(status().isOk())
.andExpect(content().contentType(MediaType.APPLICATION_XML));
有时候需要验证请求返回的模型,比如下面就断言结果会有错误。
mockMvc.perform(post("/updateInfo"))
.andExpect(status().isOk())
.andExpect(model().attributeHasErrors("user"));
某些情况下需要查看请求或响应的内容。我们可以调用Spring提供的print或log方法来打印信息或者记录日志。默认情况下print方法会将结果输出到System.out
,而log方法会将日志记录到调试级别的org.springframework.test.web.servlet.result
包下。
mockMvc.perform(post("/updateInfo"))
.andExpect(status().isOk())
.andDo(print())
.andExpect(model().attributeHasErrors("user"));
有时候需要详细检验返回结果。我们可以在所有期望方法的最后添加andReturn方法。该方法会返回一个MvcResult对象,我们可以调用该对象的各种get方法获取我们需要的信息。
MvcResult mvcResult = mockMvc.perform(post("/listUsers")).andExpect(status().isOk()).andReturn();
如果某些期望是所有方法都需要的,我们可以将它设置为共用的。但是一旦设置就无法更改。所以如果我们不需要某个共用期望的话就只能创建一个新的MockMvc对象了。
standaloneSetup(new UserController())
.alwaysExpect(status().isOk())
.alwaysExpect(content().contentType("application/json;charset=UTF-8"))
.build()
如果我们希望在单个控制器中添加过滤器的话,可以在建立MockMvc对象的时候指定过滤器。
mockMvc = standaloneSetup(new UserController()).addFilters(new CharacterEncodingFilter()).build();
spring-mvc-showcase是一个Spring官方开发的示例程序,包含了Spring Web MVC的例子和基本功能,也包含了所有的服务端测试代码。这也是一个很好的学习资源。
HtmlUnit集成
MockMvc虽然好用,但是毕竟是一个假的测试,它没有实际运行的服务器, 也不会进行实际的视图渲染、转发和重定向等操作。如果我们希望测试实际的HTML视图、JavaScript验证等功能,就需要使用HtmlUnit。
我们需要在项目中引用HtmlUnit的依赖。
compile group: 'net.sourceforge.htmlunit', name: 'htmlunit', version: '2.24'
然后初始化一个WebClient。
@Autowired
WebApplicationContext context;
WebClient webClient;
@Before
public void setup() {
webClient = MockMvcWebClientBuilder
.webAppContextSetup(context)
.build();
}
这样配置的话,默认所有localhost
下的请求就会自动通过MockMvc对象来访问,不需要实际HTTP连接,这方便我们本机测试。而其他域名会正常使用网络来连接,这可以让我们测试CDN等的状况。
然后我们可以使用WebClient来创建测试了。这里我直接贴Spring文档里的例子了。我们从例子中可以看到,WebClient的使用方法和使用普通的JavaScript操作DOM差不多。下面是创建请求的代码。
HtmlForm form = createMsgFormPage.getHtmlElementById("messageForm");
HtmlTextInput summaryInput = createMsgFormPage.getHtmlElementById("summary");
summaryInput.setValueAttribute("Spring Rocks");
HtmlTextArea textInput = createMsgFormPage.getHtmlElementById("text");
textInput.setText("In case you didn't know, Spring Rocks!");
HtmlSubmitInput submit = form.getOneHtmlElementByAttribute("input", "type", "submit");
HtmlPage newMessagePage = submit.click();
下面是执行验证的代码。这里的断言使用了AssertJ库。
assertThat(newMessagePage.getUrl().toString()).endsWith("/messages/123");
String id = newMessagePage.getHtmlElementById("id").getTextContent();
assertThat(id).isEqualTo("123");
String summary = newMessagePage.getHtmlElementById("summary").getTextContent();
assertThat(summary).isEqualTo("Spring Rocks");
String text = newMessagePage.getHtmlElementById("text").getTextContent();
assertThat(text).isEqualTo("In case you didn't know, Spring Rocks!");
从这里我们就可以看到直接使用HtmlUnit的缺点了,那就是代码笨重,不好看。Spring还提供了另外两个类库WebDriver和Geb来简化HtmlUnit的测试过程,详见Spring 参考文档 HtmlUnit集成
客户端的REST测试
如果需要客户端测试REST程序,Spring也提供了相关功能。直接来看Spring官方的例子。我们需要先创建一个RestTemplate对象,然后创建MockRestServiceServer并绑定到RestTemplate上。然后使用MockRestServiceServer的expect方法发起请求并测试结果。最后调用verify方法验证是否满足所有期望。这种方式不需要启动实际服务器,效率很高。
RestTemplate restTemplate = new RestTemplate();
MockRestServiceServer mockServer = MockRestServiceServer.bindTo(restTemplate).build();
mockServer.expect(requestTo("/greeting")).andRespond(withSuccess());
// 使用RestTemplate进行其他测试 ...
mockServer.verify();
客户端测试也可以和服务端测试结合起来。我们可以利用MockMvc对象来创建RestTemplate,这样就会使用服务端的逻辑来测试代码而不需要启动实际服务器。
MockMvc mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).build();
this.restTemplate = new RestTemplate(new MockMvcClientHttpRequestFactory(mockMvc));
// 使用RestTemplate进行其他测试 ...
mockServer.verify();