用Mockito测试类，其中包含一个新的（）调用

Question

我有一个遗留类，它包含一个new（）调用来实例化LoginContext（）:

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

您可以使用工厂来创建登录上下文。然后你可以嘲弄工厂并返回任何你想要的测试。

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();
}

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();
        return lc;
    }
}

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

为了完整起见，值得一提的是第三种技术--在调用new运算符时使用PowerMock注入mock对象。不过，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

对于未来，我将推荐EranHarel的答案（重构，将new移动到工厂，可以模仿）。但是如果你不想改变原来的源代码，使用非常得心应手和独特的功能:间谍。从文档中:

你可以创造真实物体的间谍。当您使用spy时，将调用真正的方法（除非某个方法被替换）。

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

在你的情况下，你应该写:

TestedClass tc = spy(new TestedClass());
LoginContext lcMock = mock(LoginContext.class);
when(tc.login(anyString(), anyString())).thenReturn(lcMock);