使用Powermock模拟构造函数

  在代码中经常会出现在一个类中去new另一个类对象,并且调用该对象的一些方法的情况,这时我们如果不想真正执行该对象的方法,就需要对new出来的对象进行mock,即对构造函数进行mock。
  我们首先看一个例子:

public class User {
    private String username;
    private String password;

    public User(String username, String password) {
        this.username = username;
        this.password = password;
    }

    public void insert() {
        throw new UnsupportedOperationException();
    }
}

public class UserService {
    public void saveUser(String username, String password) {
        User user = new User(username, password);
        user.insert();
    }
}

  UserService 的方法saveUser中new了一个User对象,并且调用user的insert方法进行保存。我们看下如何测试saveUser方法。

@RunWith(PowerMockRunner.class)
@PrepareForTest({UserService.class})
public class UserServiceTest {
    @Mock
    private User user;

    @Test
    public void saveUser() throws Exception {
        String username = "user1";
        String password = "aaa";
        PowerMockito.whenNew(User.class).withArguments(username, password).thenReturn(user);
        PowerMockito.doNothing().when(user).insert();

        UserService userService = new UserService();
        userService.saveUser(username, password);

        Mockito.verify(user).insert();
    }
}

  1、首先我们要注意的是在@PrepareForTest后面的是使用构造函数的类,而不是构造函数所在的类。
  2、使用下面的语句对构造函数进行mock,即当new User.class类且参数为username和password时返回user这个已经mock的对象:

PowerMockito.whenNew(User.class).withArguments(username, password).thenReturn(user);

  该语句使用的是Powermock提供的下面的接口,该接口要求根据方法的定义对相应的参数进行匹配:

public interface WithExpectedArguments<T> {
    OngoingStubbing<T> withArguments(Object firstArgument, Object... additionalArguments) throws Exception;
}

  由此可以看出mock构造函数非常简单,但是此处要强调一点,由于我们mock时指定了参数必须为username和password,如果参数不匹配时,该mock语句是不起作用的。比如我们如果修改下,如下所示:

userService.saveUser("user2", password);

  则该测试用例执行就是错误的,会提示java.lang.NullPointerException错误,这是因为模拟了user类,但是构造函数又不匹配,则返回默认值Null,从而导致调用saveUser方法时抛空指针异常。
  除了上述的withArguments接口外,Powermock还提供了另外的参数匹配接口:
  1、withAnyArguments接口,原型如下:

public interface WithAnyArguments<T> {
    OngoingStubbing<T> withAnyArguments() throws Exception;
}

  该接口不像withArguments那样对参数作精确匹配,如果使用该接口进行mock,则不论入参是什么,该mock构造函数都有效。
  2、withNoArguments接口,原型如下:

public interface WithoutExpectedArguments<T> {
    OngoingStubbing<T> withNoArguments() throws Exception;
}

  此接口主要适用于无参构造函数。
  3、withParameterTypes接口,原型如下:

public interface WithExpectedParameterTypes<T> {
    WithExpectedArguments<T> withParameterTypes(Class<?> parameterType, Class<?>... additionalParameterTypes);
}

  该接口主要是匹配方法的参数,可以看出其返回值不是OngoingStubbing,而是WithExpectedArguments,即意味这它必须后续要和withArguments配和使用,如下所示:

PowerMockito.whenNew(User.class).withParameterTypes(String.class, String.class).withArguments(username, password)
                .thenReturn(user);

  也就是说,此时匹配构造函数时,不仅要求参数要匹配,而且参数类型也要匹配。如果我们把String.class换成别的类,比如改为如下所示:

PowerMockito.whenNew(User.class).withParameterTypes(Integer.class, String.class).withArguments(username, password)
                .thenReturn(user);

  这个时候执行该测试用例就会提示如下错误:

org.powermock.reflect.exceptions.ConstructorNotFoundException: Failed to lookup constructor with parameter types [ java.lang.Integer, java.lang.String ] in class org.zte.com.User.

  最后,再补充一点,就是如果我们使用Jacoco来统计Java代码测试覆率的话,我们会发现不论我们写了多少测试用例,该UserService类的覆盖率始终是0。这个是因为我们把UserService.class放在了PrepareForTest的后面,这个时候该类就已经被PrepareForTest处理过了,此时Powermock框架会通过ClassLoader的方式改变UserService的行为。即产生一个新的类,在运行过程中凡是碰到使用UserService的时候,都是使用新建的类且提前模拟好的行为运行,并不会执行真正执行UserService类的代码,所以也就没有覆盖率的产生。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 135,374评论 19 139
  • 这是16年5月份编辑的一份比较杂乱适合自己观看的学习记录文档,今天18年5月份再次想写文章,发现简书还为我保存起的...
    Jenaral阅读 7,916评论 2 9
  • 人在漫漫旅途中,常百无聊赖,于是常有各种约定,与自然的约定,与食物的约定,与他人的约定,与将来自己的约定,...
    阿苏sue阅读 1,865评论 0 1
  • 晚上爸问我联系她怎么样了,说话有没有更好一些。我随口一说,反正就是联系咯,哪有什么好的。然后他就挂了电话。 再打给...
    扶路阅读 1,651评论 0 0
  • 忍着想要联系的急切,忍着想要将项链甩到你脸上的愤怒,忍着想要抱抱你的冲动,忍着思念你的日日夜夜,忍着无法入睡的疼痛...
    涔离阅读 1,189评论 0 0