使用Mockito通过new（）调用测试类

Question

我有一个旧类，其中包含用于实例化LoginContext（）的new（）调用：public class TestedClass {  public LoginContext login(String user, String password) {    LoginContext lc = new LoginContext("login", callbackHandler);  }}我想使用Mockito测试类来模拟LoginContext，因为它要求在实例化之前设置JAAS安全性，但是我不确定如何在不更改login（）方法以外部化LoginContext的情况下进行此操作。是否可以使用Mockito模拟LoginContext类？

Answer 1

对于将来，我会推荐伊兰·哈雷尔（Eran Harel）的答案（将new工厂搬迁到可以嘲笑的工厂）。但是，如果您不想更改原始源代码，请使用非常方便且独特的功能：spies。从文档中：

您可以创建真实对象的间谍。当您使用间谍时，将调用实际方法（除非对方法进行了加注）。

真正的间谍应该小心谨慎地使用，例如在处理遗留代码时。

在您的情况下，您应该写：

TestedClass tc = spy(new TestedClass());

LoginContext lcMock = mock(LoginContext.class);

when(tc.login(anyString(), anyString())).thenReturn(lcMock);

Answer 2

我全力支持Eran Harel的解决方案，在不可能的情况下，Tomasz Nurkiewicz的间谍建议是极好的。但是，值得注意的是，在某些情况下都不适用。例如，如果该login方法有点“骗子”：

public class TestedClass {

    public LoginContext login(String user, String password) {

        LoginContext lc = new LoginContext("login", callbackHandler);

        lc.doThis();

        lc.doThat();

    }

}

...这是旧代码，无法重构以提取新的初始化LoginContext为其自己的方法并应用上述解决方案之一。

为了完整起见，值得一提的是第三种技术- 在调用运算符时使用PowerMock注入模拟对象new。但是，PowerMock并不是灵丹妙药。它通过在其模拟的类上应用字节码操作来工作，如果所测试的类采用字节码操作或反射，并且至少从我的个人经验出发，已知该类会给测试带来性能损失，那么这可能是狡猾的实践。再说一次，如果没有其他选择，则唯一的选择必须是好的选择：

@RunWith(PowerMockRunner.class)

@PrepareForTest(TestedClass.class)

public class TestedClassTest {

    @Test

    public void testLogin() {

        LoginContext lcMock = mock(LoginContext.class);

        whenNew(LoginContext.class).withArguments(anyString(), anyString()).thenReturn(lcMock);

        TestedClass tc = new TestedClass();

        tc.login ("something", "something else");

        // test the login's logic

    }

}

Answer 3

您可以使用工厂来创建登录上下文。然后，您可以模拟工厂并返回您想要进行测试的任何内容。

public class TestedClass {

  private final LoginContextFactory loginContextFactory;

  public TestedClass(final LoginContextFactory loginContextFactory) {

    this.loginContextFactory = loginContextFactory;

  }

  public LoginContext login(String user, String password) {

    LoginContext lc = loginContextFactory.createLoginContext();

  }

}

public interface LoginContextFactory {

  public LoginContext createLoginContext();

}