转载自嘟嘟独立博客 本文链接地址: Spring Boot干货系列：（十二）Spring Boot使用单元测试
这次来介绍下Spring Boot中对单元测试的整合使用，本篇会通过以下4点来介绍，基本满足日常需求
Service层单元测试
Controller层单元测试
新断言assertThat使用
单元测试的回滚
Spring Boot中引入单元测试很简单，依赖如下：
1 2 3 4 5 <dependency > <groupId > org.springframework.boot</groupId > <artifactId > spring-boot-starter-test</artifactId > <scope > test</scope > </dependency >
本篇实例Spring Boot版本为1.5.9.RELEASE，引入spring-boot-starter-test后，有如下几个库：
JUnit — The de-facto standard for unit testing Java applications.
Spring Test & Spring Boot Test — Utilities and integration test support for Spring Boot applications.
AssertJ — A fluent assertion library.
Hamcrest — A library of matcher objects (also known as constraints or predicates).
Mockito — A Java mocking framework.
JSONassert — An assertion library for JSON.
JsonPath — XPath for JSON.
Spring Boot中单元测试类写在在src/test/java目录下，你可以手动创建具体测试类，如果是IDEA，则可以通过IDEA自动创建测试类，如下图，也可以通过快捷键⇧⌘T(MAC)或者Ctrl+Shift+T(Window)来创建，如下：
自动生成测试类如下：
然后再编写创建好的测试类，具体代码如下：
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 package com.dudu.service;import com.dudu.domain.LearnResource;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.context.SpringBootTest;import org.springframework.test.context.junit4.SpringRunner;import static org.hamcrest.CoreMatchers.*;@RunWith (SpringRunner.class ) @SpringBootTest public class LearnServiceTest @Autowired private LearnService learnService; @Test public void getLearn () LearnResource learnResource=learnService.selectByKey(1001L ); Assert.assertThat(learnResource.getAuthor(),is("嘟嘟MD独立博客" )); } }
上面就是最简单的单元测试写法，顶部只要@RunWith(SpringRunner.class)和SpringBootTest即可，想要执行的时候，鼠标放在对应的方法，右键选择run该方法即可。
测试用例中我使用了assertThat断言，下文中会介绍，也推荐大家使用该断言。
上面只是针对Service层做测试，但是有时候需要对Controller层（API）做测试，这时候就得用到MockMvc了，你可以不必启动工程就能测试这些接口。
MockMvc实现了对Http请求的模拟，能够直接使用网络的形式，转换到Controller的调用，这样可以使得测试速度快、不依赖网络环境，而且提供了一套验证的工具，这样可以使得请求的验证统一而且很方便。
Controller类：
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 package com.dudu.controller;@Controller @RequestMapping ("/learn" )public class LearnController extends AbstractController @Autowired private LearnService learnService; private Logger logger = LoggerFactory.getLogger(this .getClass()); @RequestMapping ("" ) public String learn (Model model) model.addAttribute("ctx" , getContextPath()+"/" ); return "learn-resource" ; } @RequestMapping (value = "/queryLeanList" ,method = RequestMethod.POST) @ResponseBody public AjaxObject queryLearnList (Page<LeanQueryLeanListReq> page) List<LearnResource> learnList=learnService.queryLearnResouceList(page); PageInfo<LearnResource> pageInfo =new PageInfo<LearnResource>(learnList); return AjaxObject.ok().put("page" , pageInfo); } @RequestMapping (value = "/add" ,method = RequestMethod.POST) @ResponseBody public AjaxObject addLearn (@RequestBody LearnResource learn) learnService.save(learn); return AjaxObject.ok(); } @RequestMapping (value = "/update" ,method = RequestMethod.POST) @ResponseBody public AjaxObject updateLearn (@RequestBody LearnResource learn) learnService.updateNotNull(learn); return AjaxObject.ok(); } @RequestMapping (value="/delete" ,method = RequestMethod.POST) @ResponseBody public AjaxObject deleteLearn (@RequestBody Long[] ids) learnService.deleteBatch(ids); return AjaxObject.ok(); } @RequestMapping (value="/resource/{id}" ,method = RequestMethod.GET) @ResponseBody public LearnResource qryLearn (@PathVariable(value = "id" ) Long id) LearnResource lean= learnService.selectByKey(id); return lean; } }
这里我们也自动创建一个Controller的测试类，具体代码如下：
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 package com.dudu.controller;import com.dudu.domain.User;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.http.MediaType;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.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.web.context.WebApplicationContext;@RunWith (SpringRunner.class ) @SpringBootTest public class LearnControllerTest @Autowired private WebApplicationContext wac; private MockMvc mvc; private MockHttpSession session; @Before public void setupMockMvc () mvc = MockMvcBuilders.webAppContextSetup(wac).build(); session = new MockHttpSession(); User user =new User("root" ,"root" ); session.setAttribute("user" ,user); } @Test public void addLearn () throws Exception String json="{\"author\":\"HAHAHAA\",\"title\":\"Spring\",\"url\":\"http://tengj.top/\"}" ; mvc.perform(MockMvcRequestBuilders.post("/learn/add" ) .accept(MediaType.APPLICATION_JSON_UTF8) .content(json.getBytes()) .session(session) ) .andExpect(MockMvcResultMatchers.status().isOk()) .andDo(MockMvcResultHandlers.print()); } @Test public void qryLearn () throws Exception mvc.perform(MockMvcRequestBuilders.get("/learn/resource/1001" ) .contentType(MediaType.APPLICATION_JSON_UTF8) .accept(MediaType.APPLICATION_JSON_UTF8) .session(session) ) .andExpect(MockMvcResultMatchers.status().isOk()) .andExpect(MockMvcResultMatchers.jsonPath("$.author" ).value("嘟嘟MD独立博客" )) .andExpect(MockMvcResultMatchers.jsonPath("$.title" ).value("Spring Boot干货系列" )) .andDo(MockMvcResultHandlers.print()); } @Test public void updateLearn () throws Exception String json="{\"author\":\"测试修改\",\"id\":1031,\"title\":\"Spring Boot干货系列\",\"url\":\"http://tengj.top/\"}" ; mvc.perform(MockMvcRequestBuilders.post("/learn/update" ) .accept(MediaType.APPLICATION_JSON_UTF8) .content(json.getBytes()) .session(session) ) .andExpect(MockMvcResultMatchers.status().isOk()) .andDo(MockMvcResultHandlers.print()); } @Test public void deleteLearn () throws Exception String json="[1031]" ; mvc.perform(MockMvcRequestBuilders.post("/learn/delete" ) .accept(MediaType.APPLICATION_JSON_UTF8) .content(json.getBytes()) .session(session) ) .andExpect(MockMvcResultMatchers.status().isOk()) .andDo(MockMvcResultHandlers.print()); } }
上面实现了基本的增删改查的测试用例，使用MockMvc的时候需要先用MockMvcBuilders使用构建MockMvc对象，如下
1 2 3 4 5 6 7 @Before public void setupMockMvc () mvc = MockMvcBuilders.webAppContextSetup(wac).build(); session = new MockHttpSession(); User user =new User("root" ,"root" ); session.setAttribute("user" ,user); }
因为拦截器那边会判断是否登录，所以这里我注入了一个用户，你也可以直接修改拦截器取消验证用户登录，先测试完再开启。
这里拿一个例子来介绍一下MockMvc简单的方法
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 @Test public void qryLearn () throws Exception mvc.perform(MockMvcRequestBuilders.get("/learn/resource/1001" ) .contentType(MediaType.APPLICATION_JSON_UTF8) .accept(MediaType.APPLICATION_JSON_UTF8) .session(session) ) .andExpect(MockMvcResultMatchers.status().isOk()) .andExpect(MockMvcResultMatchers.jsonPath("$.author" ).value("嘟嘟MD独立博客" )) .andExpect(MockMvcResultMatchers.jsonPath("$.title" ).value("Spring Boot干货系列" )) .andDo(MockMvcResultHandlers.print()); }
mockMvc.perform执行一个请求MockMvcRequestBuilders.get(“/user/1”)构造一个请求，Post请求就用.post方法contentType(MediaType.APPLICATION_JSON_UTF8)代表发送端发送的数据格式是application/json;charset=UTF-8accept(MediaType.APPLICATION_JSON_UTF8)代表客户端希望接受的数据类型为application/json;charset=UTF-8session(session)注入一个session，这样拦截器才可以通过ResultActions.andExpect添加执行完成后的断言ResultActions.andExpect(MockMvcResultMatchers.status().isOk())方法看请求的状态响应码是否为200如果不是则抛异常，测试不通过andExpect(MockMvcResultMatchers.jsonPath(“$.author”).value(“嘟嘟MD独立博客”))这里jsonPath用来获取author字段比对是否为嘟嘟MD独立博客,不是就测试不通过ResultActions.andDo添加一个结果处理器，表示要对结果做点什么事情，比如此处使用MockMvcResultHandlers.print()输出整个响应结果信息
本例子测试如下：
mockMvc 更多例子可以本篇下方参考查看
JUnit 4.4 结合 Hamcrest 提供了一个全新的断言语法——assertThat。程序员可以只使用 assertThat 一个断言语句，结合 Hamcrest 提供的匹配符，就可以表达全部的测试思想，我们引入的版本是Junit4.12所以支持assertThat。
清单 1 assertThat 基本语法
1 assertThat( [value], [matcher statement] );
value 是接下来想要测试的变量值；
matcher statement 是使用 Hamcrest 匹配符来表达的对前面变量所期望的值的声明，如果 value 值与 matcher statement 所表达的期望值相符，则测试成功，否则测试失败。
优点 1：以前 JUnit 提供了很多的 assertion 语句，如：assertEquals，assertNotSame，assertFalse，assertTrue，assertNotNull，assertNull 等，现在有了 JUnit 4.4，一条 assertThat 即可以替代所有的 assertion 语句，这样可以在所有的单元测试中只使用一个断言方法，使得编写测试用例变得简单，代码风格变得统一，测试代码也更容易维护。
优点 2：assertThat 使用了 Hamcrest 的 Matcher 匹配符，用户可以使用匹配符规定的匹配准则精确的指定一些想设定满足的条件，具有很强的易读性，而且使用起来更加灵活。如清单 2 所示：
清单 2 使用匹配符 Matcher 和不使用之间的比较
1 2 3 4 5 6 assertThat(s, anyOf(containsString("developer" ), containsString("Works" )));
优点 3：assertThat 不再像 assertEquals 那样，使用比较难懂的“谓宾主”语法模式（如：assertEquals(3, x);），相反，assertThat 使用了类似于“主谓宾”的易读语法模式（如：assertThat(x,is(3));），使得代码更加直观、易读。
优点 4：可以将这些 Matcher 匹配符联合起来灵活使用，达到更多目的。如清单 3 所示：1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 字符相关匹配符 assertThat(testedValue, equalTo(expectedValue)); assertThat(testedString, equalToIgnoringCase(expectedString)); assertThat(testedString, equalToIgnoringWhiteSpace(expectedString); assertThat(testedString, containsString(subString) ); assertThat(testedString, endsWith(suffix)); assertThat(testedString, startsWith(prefix)); 一般匹配符 assertThat(object,nullValue()); assertThat(object,notNullValue()); assertThat(testedString, is(equalTo(expectedValue))); assertThat(testedValue, is(expectedValue)); assertThat(testedObject, is(Cheddar.class )) ; assertThat(testedString, not(expectedString)); assertThat(testedNumber, allOf( greaterThan(8 ), lessThan(16 ) ) ); assertThat(testedNumber, anyOf( greaterThan(16 ), lessThan(8 ) ) ); 数值相关匹配符 assertThat(testedDouble, closeTo( 20.0 , 0.5 )); assertThat(testedNumber, greaterThan(16.0 )); assertThat(testedNumber, lessThan (16.0 )); assertThat(testedNumber, greaterThanOrEqualTo (16.0 )); assertThat(testedNumber, lessThanOrEqualTo (16.0 )); 集合相关匹配符 assertThat(mapObject, hasEntry("key" , "value" ) ); assertThat(iterableObject, hasItem (element)); assertThat(mapObject, hasKey ("key" )); assertThat(mapObject, hasValue(value));
单元个测试的时候如果不想造成垃圾数据，可以开启事物功能，记在方法或者类头部添加@Transactional注解即可,如下：
1 2 3 4 5 6 7 8 9 @Test @Transactional public void add () LearnResource bean = new LearnResource(); bean.setAuthor("测试回滚" ); bean.setTitle("回滚用例" ); bean.setUrl("http://tengj.top" ); learnService.save(bean); }
这样测试完数据就会回滚了，不会造成垃圾数据。如果你想关闭回滚，只要加上@Rollback(false)注解即可。@Rollback表示事务执行完回滚，支持传入一个参数value，默认true即回滚，false不回滚。
如果你使用的数据库是Mysql，有时候会发现加了注解@Transactional 也不会回滚，那么你就要查看一下你的默认引擎是不是InnoDB，如果不是就要改成InnoDB。
MyISAM与InnoDB是mysql目前比较常用的两个数据库存储引擎，MyISAM与InnoDB的主要的不同点在于性能和事务控制上。这里简单的介绍一下两者间的区别和转换方法：
MyISAM：MyISAM是MySQL5.5之前版本默认的数据库存储引擎。MYISAM提供高速存储和检索，以及全文搜索能力，适合数据仓库等查询频繁的应用。但不支持事务、也不支持外键。MyISAM格式的一个重要缺陷就是不能在表损坏后恢复数据。
InnoDB：InnoDB是MySQL5.5版本的默认数据库存储引擎，不过InnoDB已被Oracle收购，MySQL自行开发的新存储引擎Falcon将在MySQL6.0版本引进。InnoDB具有提交、回滚和崩溃恢复能力的事务安全。但是比起MyISAM存储引擎，InnoDB写的处理效率差一些并且会占用更多的磁盘空间以保留数据和索引。尽管如此，但是InnoDB包括了对事务处理和外来键的支持，这两点都是MyISAM引擎所没有的。
MyISAM适合：(1)做很多count 的计算；(2)插入不频繁，查询非常频繁；(3)没有事务。
InnoDB适合：(1)可靠性要求比较高，或者要求事务；(2)表更新和查询都相当的频繁，并且表锁定的机会比较大的情况。(4)性能较好的服务器，比如单独的数据库服务器，像阿里云的关系型数据库RDS就推荐使用InnoDB引擎。
查看MySQL当前默认的存储引擎:
1 mysql> show variables like '%storage_engine%';
你要看user表用了什么引擎(在显示结果里参数engine后面的就表示该表当前用的存储引擎):
1 mysql> show create table user;
将user表修为InnoDB存储引擎(也可以此命令将InnoDB换为MyISAM)：
1 mysql> ALTER TABLE user ENGINE=INNODB;
如果要更改整个数据库表的存储引擎，一般要一个表一个表的修改，比较繁琐，可以采用先把数据库导出，得到SQL，把MyISAM全部替换为INNODB，再导入数据库的方式。
到此为止，Spring Boot整合单元测试就基本完结，关于MockMvc以及assertThat的用法大家可以继续深入研究。后续会整合Swagger UI这个API文档工具，即提供API文档又提供测试接口界面，相当好用。
想要查看更多Spring Boot干货教程,可前往：Spring Boot干货系列总纲
.png)
