51Testing软件测试论坛

标题: powermock+mockito+testng测试用例编写 [打印本页]

作者: lsekfe    时间: 2017-9-29 09:47
标题: powermock+mockito+testng测试用例编写
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);
复制代码








欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/) Powered by Discuz! X3.2