powermock+mockito+testng测试用例编写

lsekfe

1. 序
　　最近在搞测试用例，技术选型powermock+mockito+testng，比之前用的ulitils+easymock强大很多。
　　2. pom依赖

1. 　<dependency>
   
2. 　　    <groupId>org.powermock</groupId>
   
3. 　　    <artifactId>powermock-api-mockito2</artifactId>
   
4. 　　    <version>${powermock.version}</version>
   
5. 　　    <scope>test</scope>
   
6. 　　</dependency>
   
7. 　　<dependency>
   
8. 　　    <groupId>org.powermock</groupId>
   
9. 　　    <artifactId>powermock-module-junit4</artifactId>
   
10. 　　    <version>${powermock.version}</version>
    
11. 　　    <scope>test</scope>
    
12. 　　</dependency>
    
13. 　　<dependency>
    
14. 　　    <groupId>org.powermock</groupId>
    
15. 　　    <artifactId>powermock-module-testng</artifactId>
    
16. 　　    <version>${powermock.version}</version>
    
17. 　　    <scope>test</scope>
    
18. 　　</dependency>

复制代码

3. 示例

1. public class PowermockTest extends PowerMockTestCase {
   
2. 　　@Mock
   
3. 　　PowermockDao powermockDao;
   
4. 　　@InjectMocks
   
5. 　　PowermockService powermockService = new PowermockService();
   
6. 　　@Test
   
7. 　　public void testSave_正常插入() {
   
8. 　　final PowermockDto powermockDto = new PowermockDto();
   
9. 　　powermockDto.setName("滨江吴彦祖");
   
10. 　　// mock根据名称查询返回
    
11. 　　PowerMockito.when(powermockDao.getByName(Mockito.anyString()))
    
12. 　　            .thenReturn(null);
    
13. 　　// mock插入
    
14. 　　PowerMockito.doReturn(1)
    
15. 　　            .when(powermockDao)
    
16. 　　            .insert(ArgumentMatchers.argThat(new ArgumentMatcher<PowermockDto>() {
    
17. 　　        @Override
    
18. 　　public boolean matches(PowermockDto argument) {
    
19. 　　// 生成ID
    
20. 　　argument.setId(ThreadLocalRandom.current().nextLong(1));
    
21. 　　return true;
    
22. 　　}
    
23. 　　}));
    
24. 　　// 目标方法执行
    
25. 　　Assert.assertTrue(powermockService.save(powermockDto) > 0);
    
26. 　　// 回放目标方法执行，验证方法调用
    
27. 　　Mockito.verify(powermockDao).getByName("滨江吴彦祖");
    
28. 　　Mockito.verify(powermockDao).insert(powermockDto);
    
29. 　　}
    
30. 　　}

复制代码

4. 辅助类：

1. 　public class PowermockService {
   
2. 　　private PowermockDao powermockDao;
   
3. 　　public Integer save(PowermockDto powermockDto) {
   
4. 　　if (StringUtils.isBlank(powermockDto.getName())) {
   
5. 　　throw new IllegalArgumentException("名称不能为空");
   
6. 　　}
   
7. 　　final PowermockDto dbPowermockDto = powermockDao.getByName(powermockDto.getName());
   
8. 　　if (dbPowermockDto != null) {
   
9. 　　if (powermockDto.getId() == null
   
10. 　　                                || !powermockDto.getId().equals(dbPowermockDto.getId())) {
    
11. 　　throw new IllegalArgumentException(String.format("名称[%s]重复了", powermockDto.getName()));
    
12. 　　}
    
13. 　　}
    
14. 　　if (powermockDto.getId() == null) {
    
15. 　　return powermockDao.insert(powermockDto);
    
16. 　　} else {
    
17. 　　return powermockDao.update(powermockDto);
    
18. 　　}
    
19. 　　}
    
20. 　　}
    
21. 　　public interface PowermockDao {
    
22. 　　PowermockDto getByName(String name);
    
23. 　　Integer insert(PowermockDto powermockDto);
    
24. 　　Integer update(PowermockDto powermockDto);
    
25. 　　}
    
26. 　　public class PowermockDto {
    
27. 　　private Long id;
    
28. 　　private String name;
    
29. 　　public Long getId() {
    
30. 　　return id;
    
31. 　　}
    
32. 　　public void setId(Long id) {
    
33. 　　this.id = id;
    
34. 　　}
    
35. 　　public String getName() {
    
36. 　　return name;
    
37. 　　}
    
38. 　　public void setName(String name) {
    
39. 　　this.name = name;
    
40. 　　}
    
41. 　　@Override
    
42. 　　public boolean equals(Object o) {
    
43. 　　if (this == o)
    
44. 　　return true;
    
45. 　　if (!(o instanceof PowermockDto))
    
46. 　　return false;
    
47. 　　PowermockDto that = (PowermockDto) o;
    
48. 　　if (getId() != null ? !getId().equals(that.getId())
    
49. 　　                     : that.getId() != null)
    
50. 　　return false;
    
51. 　　return getName() != null ? getName().equals(that.getName())
    
52. 　　                         : that.getName() == null;
    
53. 　　}
    
54. 　　@Override
    
55. 　　public int hashCode() {
    
56. 　　int result = getId() != null ? getId().hashCode() : 0;
    
57. 　　result = 31 * result + (getName() != null ? getName().hashCode() : 0);
    
58. 　　return result;
    
59. 　　}
    
60. 　　}

复制代码

5. 使用经验
　　5.1 mock 静态方法
　　mockito库并不能mock静态方法，需要依赖powermock

　　第一步：给类添加注解

1. // 静态类优先加载，所以需要提前告诉powermock哪些静态类需要mock
   
2. 　　@PrepareForTest({ PasswordGenerator.class, UserSecurityContext.class })
   
3. 　　public class SupplierServiceImplTest extends PowerMockTestCase {
   
4. 　　}

复制代码

第二步：mock使用

1. @Test(expectedExceptions = BusinessException.class)
   
2. 　　public void testAddSupplierAccount_genIdentityNoError() {
   
3. 　　        // 告诉powermock，需要mock该类的所有静态方法
   
4. 　　PowerMockito.mockStatic(PasswordGenerator.class);
   
5. 　　    final SupplierAccountDto supplierAccountDto = new SupplierAccountDto();
   
6. 　　supplierAccountDto.setName("滨江吴彦祖");
   
7. 　　supplierAccountDto.setCompanyName("163");
   
8. 　　supplierAccountDto.setEmail("wyz@corp.netease.com");
   
9. 　　    final String randomPWd = "xxxxoooooo";
   
10. 　　PowerMockito.when(supplierDao.selectByEmail(anyString()))
    
11. 　　.thenReturn(new ArrayList<HaitaoSupplier>());
    
12. 　　    // 静态方法mock
    
13. 　　PowerMockito.when(PasswordGenerator.genPwd()).thenReturn(randomPWd);
    
14. 　　PowerMockito.when(pwEncoder.encode(anyString())).thenReturn(randomPWd);
    
15. 　　PowerMockito.when(identityNoGenerator.genIdentityNo()).thenReturn(-1L);
    
16. 　　supplierServiceImpl.addSupplierAccount(supplierAccountDto);
    
17. 　　verify(pwEncoder).encode(randomPWd);
    
18. 　　}

复制代码

5.2 mock void 方法

1. 　// void嘛，doNothing顾名思义
   
2. 　　PowerMockito.doNothing().when(casService).addSupplier(anyLong(), any(ServiceKey.class));

复制代码

5.3 修改mock方法的入参

1. PowerMockito.doReturn(1).when(mainSubSupplierApplyDao)
   
2. 　　.insert(argThat(new ArgumentMatcher<MainSubSupplierApply>() {
   
3. 　　@Override
   
4. 　　public boolean matches(MainSubSupplierApply argument) {
   
5. 　　        // mock 插入，生成主键
   
6. 　　argument.setId(applyId);
   
7. 　　        return true;
   
8. 　　}
   
9. 　　}));

复制代码

5.4 修改mock方法的出参

1. PowerMockito.when(DBEncryptUtil.encrypt(anyString())).thenAnswer(new Answer<String>() {
   
2. 　　@Override
   
3. 　　public String answer(InvocationOnMock invocation) throws Throwable {
   
4. 　　    // mock方法返回值，使用第一个入参返回
   
5. 　　    return invocation.getArgument(0);
   
6. 　　}
   
7. 　　});

复制代码

5.5 mock入参null值

1. // mockito2.x ArgumentMatchers.anyString() 与 ArgumentMatchers.isNull() 互斥，可以看下方法注释
   
2. 　　PowerMockito.when(inquiryGoodsSnapshotDao.getBySkuIdsAndPlanSendTimeBetween2Days(
   
3. 　　ArgumentMatchers.<ObjectPair<String, Long>>anyList(),
   
4. 　　ArgumentMatchers.<Integer>anyList(),
   
5. 　　ArgumentMatchers.<Long>isNull())
   
6. 　　).thenReturn(inquiryGoodsSnapshotList);

复制代码