spring boot 测试
一、集成测试自动配置
代码清单4-1 用SpringJUnit4ClassRunner对Spring应用程序进行集成测试
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes=AddressBookConfiguration.class) // 加载应用程序上下文
public class AddressServiceTests {
@Autowired
private AddressService addressService; // 注入地址服务
@Test
public void testService() { // 测试地址服务
Address address = addressService.findByLastName("Sheman");
assertEquals("P", address.getFirstName());
assertEquals("Sherman", address.getLastName());
assertEquals("42 Wallaby Way", address.getAddressLine1());
assertEquals("Sydney", address.getCity());
assertEquals("New South Wales", address.getState());
assertEquals("2000", address.getPostCode());
}
}
解释:
SpringJUnit4ClassRunner
:是一个 JUnit 类运行器,会为 JUnit 测试加载 Spring 应用程 序上下文,并为测试类自动织入所需的 Bean。- 可以把
@ContextConfiguration
替换为 Spring Boot 的@SpringApplicationConfiguration
, 这样不仅会加载应用程序上下文,还会开启日志、加载外部属性(application.properties 或 application.yml),以及其他 Spring Boot 特性。 大多数情况下,为 Spring Boot 应用程序编写测试时应该用 @SpringApplicationConfiguration 代替 @ContextConfiguration。
二、测试 Web 应用程序
要恰当地测试一个 Web 应用程序,需要投入一些实际的 HTTP 请求,确认它能正确地处理 那些请求。Spring Boot 开发者有两个可选的方案能实现这类测试:
- Spring Mock MVC:能在一个近似真实的模拟 Servlet 容器里测试控制器,而不用实际启动 应用服务器。
- Web 集成测试:在嵌入式 Servlet 容器(比如 Tomcat 或 Jetty)里启动应用程序,在真正的应 用服务器里执行测试。
1. 模拟 Spring MVC
要在测试里设置 Mock MVC,可以使用 MockMvcBuilders,该类提供了两个静态方法:
- standaloneSetup():构建一个 Mock MVC,提供一个或多个手工创建并配置的控制器。
- webAppContextSetup():使用 Spring 应用程序上下文来构建 Mock MVC,该上下文里 可以包含一个或多个配置好的控制器。
两个方法区别:
- standaloneSetup():手工初始化并注入要测试的控制器,
- webAppContextSetup():基于一个 WebApplicationContext 的实例,通常由 Spring 加载。
前者同单元测试更加接近,你可能只想让它专注于单一控制器的测试,而后者让 Spring 加载控制 器及其依赖,以便进行完整的集成测试。
/* 代码清单 4-2 为集成测试控制器创建 Mock MVC */
@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes = ReadingListApplication.class) // 开启 Web 上下文
@WebAppConfiguration
public class MockMvcWebTests {
@Autowired
private WebApplicationContext webContext; // 注入 WebApplicationContext
private MockMvc mockMvc;
@Before
public void setupMockMvc() {
mockMvc = MockMvcBuilders
.webAppContextSetup(webContext) // 设置 MockMvc
.build();
}
}
/* 向 /readingList 发起一个 GET 请求 */
@Test
public void homePage() throws Exception {
mockMvc.perform(get("/readingList"))
.andExpect(status().isOk())
.andExpect(view().name("readingList"))
.andExpect(model().attributeExists("books"))
.andExpect(model().attribute("books", is(empty())));
}
/* 向/readingList发起一个POST请求 */
@Test
public void postBook() throws Exception {
mockMvc.perform(post("/readingList")
.contentType(MediaType.APPLICATION_FORM_URLENCODED)
.param("title", "BOOK TITLE")
.param("author", "BOOK AUTHOR")
.param("isbn", "1234567890")
.param("description", "DESCRIPTION"))
.andExpect(status().is3xxRedirection())
.andExpect(header().string("Location", "/readingList"));
/* 验证刚刚的 POST 请求 */
// 配置期望的图书
Book expectedBook = new Book();
expectedBook.setId(1L);
expectedBook.setReader("craig");
expectedBook.setTitle("BOOK TITLE");
expectedBook.setAuthor("BOOK AUTHOR");
expectedBook.setIsbn("1234567890");
expectedBook.setDescription("DESCRIPTION");
// 执行 GET 请求
mockMvc.perform(get("/readingList"))
.andExpect(status().isOk())
.andExpect(view().name("readingList"))
.andExpect(model().attributeExists("books"))
.andExpect(model().attribute("books", hasSize(1)))
.andExpect(model().attribute("books",
contains(samePropertyValuesAs(expectedBook))));
}
2. 测试 Web 安全
使用 Spring Security
-
添加依赖
<dependency> <groupId>org.springframework.security</groupId> <artifactId>spring-security-test</artifactId> <scope>test</scope> </dependency>
-
在创建 MockMvc 实例时运用 Spring Security 的配置器
@Before public void setupMockMvc() { mockMvc = MockMvcBuilders .webAppContextSetup(webContext) .apply(springSecurity()) .build(); }
-
使用(具体的安全配置取决于你如何配置 Spring Security(或者 Spring Boot 如何自动配置 Spring Security)。)
场景代码:
1) 请求未经身份验证
/* 请求未经身份验证,重定向回登录界面 */ @Test public void homePage_unauthenticatedUser() throws Exception { mockMvc.perform(get("/")) .andExpect(status().is3xxRedirection()) .andExpect(header().string("Location", "http://localhost/login")); }
2) 请求经过身份验证 Spring Security 提供了两个注解:
- @WithMockUser:用给定的值创建了一个 UserDetails 对象,指定用户名、密码和授权。
- @WithUserDetails:使用事先配置好的 UserDetailsService 来加载 UserDetails 对象, 根据给定的用户名查找 并返回一个 Reader 对象。
/* 经过身份验证的请求,使用 @WithMockUser */ @Test @WithMockUser(username="craig", password="password", roles="READER") public void homePage_authenticatedUser() throws Exception { ... }
/* 经过身份验证的请求,使用 @WithUserDetails */ @Test @WithUserDetails("craig") public void homePage_authenticatedUser() throws Exception { Reader expectedReader = new Reader(); expectedReader.setUsername("craig"); expectedReader.setPassword("password"); expectedReader.setFullname("Craig Walls"); mockMvc.perform(get("/")) .andExpect(status().isOk()) .andExpect(view().name("readingList")) .andExpect(model().attribute("reader", samePropertyValuesAs(expectedReader))) .andExpect(model().attribute("books", hasSize(0))) }
此处没有启动 Servlet 容器来运行这些测试, Spring 的 Mock MVC 取代了实际的 Servlet 容器。它比直接调用控制器方法要好,但它并没有真的在 Web 浏 览器里执行应用程序,验证呈现出的视图。
三、测试运行中的应用程序
Spring Boot 支持用嵌入式 Servlet 容器来启动应用程序。
Spring Boot 的 @WebIntegrationTest 注解就是这么做的。 在测试类上添加 @WebIntegrationTest 注解,可以声明你不仅希望 Spring Boot 为测试创建应用程序上下文,还要启 动一个嵌入式的 Servlet 容器。一旦应用程序运行在嵌入式容器里,你就可以发起真实的 HTTP 请 求,断言结果了。
/* 代码清单 4-5 在服务器里启动应用程序,以 Spring 的 RestTemplate 对应用程序发起 HTTP 请求 */
@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes=ReadingListApplication.class)
@WebIntegrationTest
public class SimpleWebTest {
@Test(expected=HttpClientErrorException.class)
public void pageNotFound() {
try {
RestTemplate rest = new RestTemplate();
rest.getForObject(
"http://localhost:8080/bogusPage", String.class);
fail("Should result in HTTP 404");
} catch (HttpClientErrorException e) {
assertEquals(HttpStatus.NOT_FOUND, e.getStatusCode());
throw e;
}
}
}
1. 用随机端口启动服务器
@WebIntegrationTest(value={"server.port=0"})
或者 @WebIntegrationTest("server.port=0")
或者 @WebIntegrationTest(randomPort=true)
使用端口
@Value("${local.server.port}")
private int port;
rest.getForObject(
"http://localhost:{port}/bogusPage", String.class, port);
2. 使用 Selenium 测试 HTML 页面
-
添加 org.seleniumhq.selenium 依赖
-
代码里使用
1> 配置
/* 在 Spring Boot 里使用 Selenium 测试的模板 */ @RunWith(SpringJUnit4ClassRunner.class) @SpringApplicationConfiguration(classes=ReadingListApplication.class) @WebIntegrationTest(randomPort=true) public class ServerWebTests { private static FirefoxDriver browser; @Value("${local.server.port}") private int port;
<span class="hljs-comment">//配置Firefox驱动</span> <span class="hljs-meta">@BeforeClass</span> <span class="hljs-keyword">public</span> <span class="hljs-keyword">static</span> <span class="hljs-keyword">void</span> <span class="hljs-title function_">openBrowser</span><span class="hljs-params">()</span> { browser = <span class="hljs-keyword">new</span> <span class="hljs-title class_">FirefoxDriver</span>(); browser.manage().timeouts() .implicitlyWait(<span class="hljs-number">10</span>, TimeUnit.SECONDS); } <span class="hljs-comment">//关闭浏览器</span> <span class="hljs-meta">@AfterClass</span> <span class="hljs-keyword">public</span> <span class="hljs-keyword">static</span> <span class="hljs-keyword">void</span> <span class="hljs-title function_">closeBrowser</span><span class="hljs-params">()</span> { browser.quit(); }
}
2> 测试
用Selenium测试阅读列表应用程序 @Test public void addBookToEmptyList() { String baseUrl = "http://localhost:" + port; browser.get(baseUrl); assertEquals("You have no books in your book list", browser.findElementByTagName("div").getText());
<span class="hljs-comment">//填充并发送表单 </span> browser.findElementByName(<span class="hljs-string">"title"</span>).sendKeys(<span class="hljs-string">"BOOK TITLE"</span>); browser.findElementByName(<span class="hljs-string">"author"</span>).sendKeys(<span class="hljs-string">"BOOK AUTHOR"</span>); browser.findElementByName(<span class="hljs-string">"isbn"</span>).sendKeys(<span class="hljs-string">"1234567890"</span>); browser.findElementByName(<span class="hljs-string">"description"</span>).sendKeys(<span class="hljs-string">"DESCRIPTION"</span>); browser.findElementByTagName(<span class="hljs-string">"form"</span>).submit(); <span class="hljs-comment">//判断列表中是否包含新书</span> <span class="hljs-type">WebElement</span> <span class="hljs-variable">dl</span> <span class="hljs-operator">=</span> browser.findElementByCssSelector(<span class="hljs-string">"dt.bookHeadline"</span>); assertEquals(<span class="hljs-string">"BOOK TITLE by BOOK AUTHOR (ISBN: 1234567890)"</span>, dl.getText()); <span class="hljs-type">WebElement</span> <span class="hljs-variable">dt</span> <span class="hljs-operator">=</span> browser.findElementByCssSelector(<span class="hljs-string">"dd.bookDescription"</span>); assertEquals(<span class="hljs-string">"DESCRIPTION"</span>, dt.getText());
}
备注:书上版本比较老,下面补充下新版本的测试方法。 例子摘自segmentfault_chenatu 的文章 (仅适用 spring-boot 1.4 版本以后的写法):
直接调用接口函数进行测试:
@RunWith(SpringRunner.class)
@SpringBootTest
public class ApiTest {
@Autowired
MessageApi messageApi;
...
测试 controller:
@RunWith(SpringRunner.class)
@SpringBootTest
@AutoConfigureMockMvc
public class ControllerTest {
@Autowired
private MockMvc mockMvc;
<span class="hljs-meta">@Test</span>
<span class="hljs-keyword">public</span> <span class="hljs-keyword">void</span> <span class="hljs-title function_">testControllerMethods</span><span class="hljs-params">()</span> {
<span class="hljs-type">MvcResult</span> <span class="hljs-variable">result</span> <span class="hljs-operator">=</span> mockMvc.perform(get(<span class="hljs-string">"/get-receive-message-abstracts"</span>).param(<span class="hljs-string">"siteId"</span>, <span class="hljs-string">"webtrn"</span>).param(<span class="hljs-string">"uid"</span>, <span class="hljs-string">"lucy"</span>)
.param(<span class="hljs-string">"limit"</span>, <span class="hljs-string">"100"</span>)).andExpect(status().isOk())
.andExpect(jsonPath(<span class="hljs-string">"$"</span>, hasSize(<span class="hljs-number">10</span>))).andExpect(jsonPath(<span class="hljs-string">"$[9].title"</span>, is(<span class="hljs-string">"hello0"</span>))).andReturn();
}</code></pre>
mockito 使用参考: