???关注微信公众号：【芋艿的后端小屋】有福利：

1. RocketMQ / MyCAT / Sharding-JDBC 所有源码分析文章列表
2. RocketMQ / MyCAT / Sharding-JDBC 中文注释源码 GitHub 地址
3. 您对于源码的疑问每条留言都将得到认真回复。甚至不知道如何读源码也可以请教噢。
4. 新的源码解析文章实时收到通知。每周更新一篇左右。

---

• 1. 为什么做单元测试
• 2. 选型
  • 2.1 JUnit
  • 2.2 Mockito
  • 2.3 PowerMock
• 3. 实践
  • 3.1 Service 层
  • 3.2 Controller 层
  • 3.3 Dao 层
• 4. 建议

本文主要面向Java工程师，见谅

1. 为什么做单元测试

• 提升重复测试效率
• 提升代码质量
• 提升部署可靠性

2. 选型

2.1 JUnit

单元测试套件

为什么不使用 TestNG ？在整合 TestNG + Mockito + PowerMock + Spring 时，Spring 容器初始化失败， @Autowired 的属性无法注入，查找资料，基本无解，只好退而求其次，选择 JUnit 。从功能强大程度，易用性上，个人较为推荐 TestNG 。

2.2 Mockito

Mock 工具

为什么不使用 JMock 、EasyMock ？因为 API 上更加轻量级，用起来爽爽的。

2.3 PowerMock

作为 Mockito 的补充，增加对 private、static、final 的支持。

为什么 Mockito 不自己支持？Mockito-And-Private-Methods 作者有自己的考虑，不过对于使用者，增加了很多整合的痛苦。

3. 实践

3.1 Service 层

方案一：

• 依赖的 Dao 层代码，使用 mock 进行模拟
• 依赖的 Service 层代码，使用 mock 进行模拟

方案二：

• 依赖的 Dao 层代码，使用内嵌数据库，例如说 h2database
• 依赖的 Service 层代码，使用 mock 进行模拟

个人较为推荐方案二。从可靠性角度考虑，相对真实的模拟了 Dao 层，例如数据插入后，可以进行查询，判断Service是否正确设置值，是否有属性漏设置会设置错；当然，该方案在维护上会相对麻烦一些，各有利弊。

3.2 Controller 层

方案一：

• 依赖的 Service 层代码，使用 mock 进行模拟

方案二：

• 依赖的 Service 层【读】类逻辑，执行真实逻辑。读取依赖的数据，使用内嵌数据库插入。
• 依赖的 Service 层【写】类逻辑，使用 mock 进行模拟。和 Controller 类的逻辑或者返回有关系的数据，使用内嵌数据库插入。

个人较为推荐方案二。原因同 Service 层单元测试。

另外，大部门互联网公司 MVC 框架选用的是 SpringMVC ，可以使用 SpringMVC Test 框架。使用方式如下：

MockMvcBuilders.standaloneSetup(XXXController).build()复制代码

3.3 Dao 层

使用内嵌的数据实现进行测试。

• MySQL ：h2database
• MongoDB ：fongo
• Redis ：embedded-redis

4. 建议

1. 加 QQ：7685413，进行互相交流。?。
2. 当使用 TestNG + Mockito + PowerMock ，测试类使用 IObjectFactory 而不要去继承 PowerMockTestCase 。Java 无法多继承，如果此时你有必须继承的类就十分棘手。

@ObjectFactory  
public IObjectFactory getObjectFactory() {  
    return new PowerMockObjectFactory();  
}复制代码

1. 当使用 PowerMockTestCase 每个测试单元执行完后，会清理掉对方法的 mock。因此，如果有公用的 mock 方法，需要使用下 @BeforeMethod（TestNG）、@Before（JUnit）里在进行一次mock。
2. 当使用 Mockito + PowerMock ，对测试类进行测试时，有需要对测试类的一些私有方法、或者静态方法进行 mock 时，不要使用 @Spy，而要使用PowerMokito.spy()进行 mock 。很酸爽。