Spring Boot的测试，和普通项目的测试类同，可以继续使用我们熟悉的测试工具。当然，这里的测试，主要还是后台代码的测试。
![]()
主要需要注意的地方仅有三点：
1、依赖包的引入：pom.xml中仅依赖spring-boot-starter-test，它把相关的依赖全部引入。
2、在测试类上的注解，常用的注解有三个
@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes = Application.class)
@WebAppConfiguration
这三个注解，只要注意第二个注解括号内的Application.class就行，把它替换为项目的启动类即可。
我们前面spring-boot-sample-mysql工程的启动类为SpringBootSampleMysqlApplication，那么测试类上的注解就是
@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes = SpringBootSampleMysqlApplication.class)
@WebAppConfiguration
3、测试类的文件结构，保持src/test/java和src/main/java结构一直，即：包+文件夹。
如：com.example包service中类的测试，那么在src/test/java也是建立com.example包，再在包下建立文件夹service.
注：由于我们测试的启动类用的是项目的启动类，所以Spring Boot项目的测试配置文件任然用src/main/resources的。
下面贴两个个测试类代码
1、服务类的
- package com.example.service;
-
- import java.util.List;
-
- import org.junit.Assert;
- import org.junit.Test;
- import org.junit.runner.RunWith;
- import org.springframework.beans.factory.annotation.Autowired;
- import org.springframework.boot.test.SpringApplicationConfiguration;
- import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
- import org.springframework.test.context.web.WebAppConfiguration;
-
- import com.example.Application;
- import com.example.domain.TestPOJO;
- import com.example.dto.HotelDto;
-
- @RunWith(SpringJUnit4ClassRunner.class)
- @SpringApplicationConfiguration(classes = Application.class)
- @WebAppConfiguration
- public class TestServicesTest {
-
- @Autowired
- TestServices testServices;
-
- @Test
- public void testShow() {
- String expectedResult="hello world!";
- String result=testServices.show();
- Assert.assertTrue("数据一致", expectedResult.equals(result));
- Assert.assertFalse("数据不一致", !expectedResult.equals(result));
- }
-
- @Test
- public void testShowDao() {
- List<TestPOJO> testPOJOList=testServices.showDao(10);
- Assert.assertTrue("数据集不对", testPOJOList.size()==1);
- Assert.assertTrue("数据一致", testPOJOList.get(0).getName().equals("nice"));
- }
-
- @Test
- public void testFindByCountry() {
- List<HotelDto> testPOJOList=testServices.findByCountry("US");
- Assert.assertTrue("数据集不对", testPOJOList.size()==1);
- Assert.assertTrue("数据一致", testPOJOList.get(0).getCityName().equals("San Francisco"));
- }
-
- }
2、controller类的
- package com.example.controller;
-
- import java.util.List;
-
- import org.junit.Assert;
- 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.SpringApplicationConfiguration;
- import org.springframework.http.MediaType;
- 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.MvcResult;
- import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
- import org.springframework.test.web.servlet.setup.MockMvcBuilders;
- import org.springframework.web.context.WebApplicationContext;
-
- import com.example.Application;
- import com.example.domain.TestPOJO;
- import com.example.dto.HotelDto;
- import com.example.service.TestServices;
- import com.fasterxml.jackson.core.JsonProcessingException;
- import com.fasterxml.jackson.databind.ObjectMapper;
-
- @RunWith(SpringJUnit4ClassRunner.class)
- @SpringApplicationConfiguration(classes = Application.class)
- @WebAppConfiguration
- public class TestControllerTest {
-
- MockMvc mvc;
-
- @Autowired
- WebApplicationContext webApplicationConnect;
-
- @Autowired
- TestServices testServices;
-
- String expectedJson;
-
- @Before
- public void setUp() throws JsonProcessingException {
- mvc = MockMvcBuilders.webAppContextSetup(webApplicationConnect).build();
-
- }
-
- @Test
- public void testShow() throws Exception {
- String expectedResult = "hello world!";
- String uri = "/show";
- MvcResult mvcResult = mvc.perform(MockMvcRequestBuilders.get(uri).accept(MediaType.APPLICATION_JSON))
- .andReturn();
- int status = mvcResult.getResponse().getStatus();
- String content = mvcResult.getResponse().getContentAsString();
-
- Assert.assertTrue("错误，正确的返回值为200", status == 200);
- Assert.assertFalse("错误，正确的返回值为200", status != 200);
- Assert.assertTrue("数据一致", expectedResult.equals(content));
- Assert.assertFalse("数据不一致", !expectedResult.equals(content));
- }
-
- protected String Obj2Json(Object obj) throws JsonProcessingException {
- ObjectMapper mapper=new ObjectMapper();
- return mapper.writeValueAsString(obj);
- }
-
- @Test
- public void testShowDaoInt() throws Exception {
- List<TestPOJO> testPOJOList = testServices.showDao(10);
- expectedJson = Obj2Json(testPOJOList);
-
- String uri="/showDao?age=10";
- MvcResult mvcResult=mvc.perform(MockMvcRequestBuilders.get(uri).accept(MediaType.APPLICATION_JSON)).andReturn();
- int status=mvcResult.getResponse().getStatus();
- String content=mvcResult.getResponse().getContentAsString();
-
- Assert.assertTrue("错误，正确的返回值为200", status == 200);
- Assert.assertFalse("错误，正确的返回值为200", status != 200);
- Assert.assertTrue("数据一致", expectedJson.equals(content));
- Assert.assertFalse("数据不一致", !expectedJson.equals(content));
- }
-
- @Test
- public void testShowDaoString() throws Exception {
- List<HotelDto> hotelDtoList=testServices.findByCountry("US");
- expectedJson = Obj2Json(hotelDtoList);
-
- String uri="/country/US";
- MvcResult mvcResult=mvc.perform(MockMvcRequestBuilders.get(uri).accept(MediaType.APPLICATION_JSON)).andReturn();
- int status=mvcResult.getResponse().getStatus();
- String content=mvcResult.getResponse().getContentAsString();
-
- Assert.assertTrue("错误，正确的返回值为200", status == 200);
- Assert.assertFalse("错误，正确的返回值为200", status != 200);
- Assert.assertTrue("数据一致", expectedJson.equals(content));
- Assert.assertFalse("数据不一致", !expectedJson.equals(content));
- }
-
- }
controller类的，为了MockMvc，注入了WebApplicationContext。
原文地址http://www.bieryun.com/833.html
