PowerMock – 快速入门

引言

<dependencies>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-module-junit4</artifactId>
        <version>1.6.5</version>
        <scope>test</scope>
    </dependency>

    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-api-mockito</artifactId>
        <version>1.6.5</version>
        <scope>test</scope>
    </dependency>
</dependencies>

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Book {
  private String name;
  private Double price;
}

public class BookDao {
  public int count() {
    throw new UnsupportedOperationException("BookDao does not support count() operation.");
  }

  public void insert(Book book) {
    throw new UnsupportedOperationException("BookDao does not support insert() operation.");
  }
}

@Data
@NoArgsConstructor
@AllArgsConstructor
public class BookService {
  private BookDao bookDao;

  public int count() {
     return bookDao.count();
  }

  public void save(Book book){
    bookDao.insert(book);
  }
}

public class BookServiceTestWithJUnit {
  private BookService bookService;

  @Before
  public void setup() {
    BookDao bookDao = new BookDao();
    bookService = new BookService(bookDao);
  }

  @Test
  public void count() {
    try {
      int count = bookService.count();
      Assert.fail("Procedure will not be here.");
    } catch (Exception e) {
      Assert.assertTrue(e instanceof UnsupportedOperationException);
    }
  }

  @Test
  public void save() {
    try {
      Book book = new Book("C++ Primer", 20.0);
      BookDao bookDao = new BookDao();
      BookService bookService = new BookService(bookDao);
      bookService.save(book);
      Assert.fail("Procedure will not be here.");
    } catch (Exception e) {
      Assert.assertTrue(e instanceof UnsupportedOperationException);
    }
  }
}

Mockito单元测试

Book，BookDao和BookService我们直接使用上面示例中的即可，重点是Mockito的使用。

public class BookServiceTestWithMockito {
  /**
   * 初始化Mock对象，代替不使用注解的场景中的Mockito.mock(xxx.class)以达到精简代码的目的.
   */
  @Mock
  private BookDao bookDao;

  @Before
  public void setup() {
    //参数this指代所写的测试类
    MockitoAnnotations.initMocks(this);
  }

  @Test
  public void count() {
    //录制bookDao的行为以便后续进行回放，具体含义为：当bookDao执行count()时返回一个数值20.
    Mockito.when(bookDao.count()).thenReturn(20);
    //创建Service实例
    BookService bookService = new BookService(bookDao);
    //执行bookService.count()时会在其中回放第一步录制的行为，也就是说会拿到bookDao.count()返回的值20.
    int count = bookService.count();
    //断言
    Assert.assertEquals(20, count);
  }

  @Test
  public void save() {
    //新建Book实例
    Book book = new Book("C++ Primer", 20.9);
    //录制bookDao的行为以便后续进行回放，insert为void类型，没有返回值。由于具体含义为：当bookDao执行insert(book)什么也不做.
    Mockito.doNothing().when(bookDao).insert(book);
    //创建Service实例
    BookService bookService = new BookService(bookDao);
    //执行保存操作
    bookService.save(book);
    //验证Mock BookDao对象insert方法没有被调用
    Mockito.verify(bookDao).insert(book);
  }
}

使用Mockito不仅省去了我们new各种实例的繁琐操作，而且生成的mock对象可以复用。在无法访问依赖服务的情况下需要测试的对象可以安装提前录制好的行为在调用时进行回放，结合断言可以预测返回结果。

Power Mock单元测试

同样Book，BookDao和BookService直接使用上面示例中的即可，此次采用Power Mock进行测试。

public class BookServiceTestWithPowerMock {
  //普通 Mock 不需要加 @RunWith 和 @PrepareForTest 注解。
  @Test
  public void count() {
    //Mock对象，也可以使用 org.mockito.Mock 注解标记来实现
    BookDao bookDao = PowerMockito.mock(BookDao.class);
    //录制bookDao的行为以便后续进行回放，具体含义为：当bookDao执行count()时返回一个数值15.
    PowerMockito.when(bookDao.count()).thenReturn(15);
    //创建Service实例
    BookService bookService = new BookService(bookDao);
    //执行bookService.count()时会在其中回放第一步录制的行为，也就是说会拿到bookDao.count()返回的值20.
    int count = bookService.count();
    Assert.assertEquals(15, count);
  }

  @Test
  public void save() {
    //创建Book实例
    Book book = new Book("C++ Primer", 30.5);
    //Mock对象，也可以使用 org.mockito.Mock 注解标记来实现
    BookDao bookDao = PowerMockito.mock(BookDao.class);
    //录制bookDao的行为以便后续进行回放，insert为void类型，没有返回值。由于具体含义为：当bookDao执行insert(book)什么也不做.
    PowerMockito.doNothing().when(bookDao).insert(book);
    BookService bookService = new BookService(bookDao);
    //执行保存操作
    bookService.save(book);
    //验证Mock BookDao对象insert方法没有被调用
    Mockito.verify(bookDao).insert(book);
  }
}

通过对比可能发现Mockito和Power Mock API使用起来差别不大，但是为什么选择Power Mock而不是直接用Mockito呢？PowerMock可以模拟静态方法，删除静态初始化程序，允许模拟而不依赖于注入等，而Mockito则不支持某些操作，比如对final修饰类的Mock等。本次通过对比先熟悉下Power Mock的由来与简单使用，后续将从变量、方法、常用类与接口方面详细介绍。

转载请注明：雪后西塘 » PowerMock – 快速入门