在公共方法中拦截私有方法返回的对象

Question

我需要对一个方法进行单元测试，并且我想模拟该行为，以便我可以测试方法中代码的必要部分。

为此，我想访问由我尝试测试的方法中的私有方法返回的对象。我创建了一个示例代码来大致了解我想要实现的目标。

主类

Class Main {
  public String getUserName(String userId) {
    User user = null;
    user = getUser(userId);
    if(user.getName().equals("Stack")) {
      throw new CustomException("StackOverflow");
    }

    return user.getName();

 }

 private User getUser(String userId) {
  // find the user details in database
  String name = ""; // Get from db
  String address = ""; // Get from db
  return new User(name, address);
 }
}

测试类

@Test (expected = CustomException.class)
public void getUserName_UserId_ThrowsException() {
  Main main = new Main();
  // I need to access the user object returned by getUser(userId)
  // and spy it, so that when user.getName() is called it returns Stack
  main.getUserName("124");
}  

Answer 1

访问私有只有两种方式：

1. 使用反射
2. 扩大范围
3. 也许正在等待 Java 9 使用新的作用域机制？

我会将范围修饰符从私有更改为包范围。使用反射对于重构来说是不稳定的。是否使用 PowerMock 之类的助手并不重要。它们只会减少反射周围的样板代码。

但最重要的一点是，你不应该在 whitbox 测试中测试得太深。这会使测试设置爆炸。尝试将您的代码分割成更小的部分。

方法“getUserName”需要来自用户对象的唯一信息是名称。它将验证名称并抛出异常或返回它。所以应该没有必要在测试中引入 User-object。

所以我的建议是你应该将检索用户对象名称的代码提取到一个单独的方法中，并使这个方法包范围。现在不需要只模拟主对象的用户对象。但是该方法只有最少的信息才能正常工作。

class Main {

    public String getUserName(String userId) {
        String username = getUserNameFromInternal(userId);
        if (userName.equals("Stack")) {
            throw new CustomException("StackOverflow");
        }
        return user.getName();
    }

    String getUserNameFromInternal(String userId) {
        User user = getUser(userId);
        return user.getName();
    }

    ...

}

测试：

@Test (expected = CustomException.class)
public void getUserName_UserId_ThrowsException() {
  Main main = Mockito.mock(new Main());
  Mockito.when(main.getUserNameInternal("124")).thenReturn("Stack");
  main.getUserName("124");
}

Answer 2

您在私有方法中调用new 的问题。

答案是不要求助于 PowerMock；或更改该方法的可见性。

合理的答案是将对“给我一个用户对象的东西”的依赖“提取”到它自己的类中；并将该类的实例提供给您的“主”类。因为这样你就可以简单地模拟那个“工厂”对象；让它做任何你想做的事情。

含义：您当前的代码很难测试。与其解决由此引起的问题，不如花时间学习如何编写易于测试的代码；例如通过观看这些videos 作为起点。

鉴于您的最新评论：当您处理遗留代码时，您真的希望使用 PowerMockito。要理解的关键部分：您不要“嘲笑”该私有方法；您宁愿考虑嘲笑对new User() 的调用；如here所述。

Answer 3

您可以使用 PowerMock 的 mockPrivate，但我不推荐它。 如果你有这样的问题，通常意味着你的设计很糟糕。 为什么不保护方法？