如何使用 JUnit 在 Spring Boot 中的公共方法中模拟私有方法

Question

我想问你几个问题并征求你的意见：

我想测试我的公共方法（我使用 Spring Boot、Mockito、JUnit）：

@Service
public class MyClass{

public Long getClientId(List<String> nameSurname) throws AuthorizationException {
        Long operatorId;
        if(...){
        (... something not interesting ...)
            User user = getUserByLogin("AnthonyGates2");
            operatorId = nonNull(user) ? user.getOperatorId() : null;
        } else {
            List<User> users = getUserListByLogin("AnthinyGates");
            operatorId = isNotEmpty(users) ? return operatorId;
         return operatorId;
    }

如何测试方法getClientId？

方法 getUserByLogin 和 getUserListByLogin 在此类 (MyClass) 中是私有的，但我必须模拟这些私有方法的结果，因为这些方法从外部服务检索数据。 这些私有方法看起来像：

User user = DelegateImpl.getDelegate().getUserByLogin(nameAndSurname);

DelegateImpl.getDelegate().getUserByLogin 从数据库中获取数据，并且必须像这样模拟数据：

when(DelegateImpl.getDelegate().getUserByLogin(any())).thenReturn(user);

如何测试我的公开课？我应该使用 PowerMock/PowerMockito 吗？在我看来，公开这些方法很难看，因为这些方法只在MyClass 中调用。我在 Internet 上找不到适合我的案例的好的教程（Spring Boot、Mockito、JUnit）。

非常感谢您的所有提示！

最好的问候 马修

Answer 1

仅通过调用公共方法来测试单元。我认为您的示例是服务层中的一个类（包含业务逻辑），两个 getUser... 方法应该在不同的类中（我认为在数据层中），它们可以是公共的。通过构造函数将该类作为依赖项（在服务对象中）注入，以便在测试服务类时模拟它。数据层类（使用 getUser... 方法）也可以通过它自己的单元测试进行测试。

Answer 2

如果你不能对一个方法/类进行单元测试，那很可能意味着它做的太多了。尝试将您的私有方法提取到单独的类中。它不需要公开 - 例如，您可以将其本地包放在同一个包中。

稍后，在测试中，您必须注入该类的模拟并模拟其行为。

MyClass 在其单元测试中的设置可能类似于：

AnotherClass anotherClassMock = Mockito.mock(AnotherClass.class);
MyClass myClass = new MyClass(anotherClassMock);

AnotherClass 将有方法 getUserListByLogin 和 getUserByLogin。

编辑： 您的私有方法中的逻辑似乎已经调用了外部类。问题是您通过调用另一个类中的静态getDelegate() 方法来获取对象的实例。

您可以这样做：

• 在MyClass 中创建一个新字段，该字段的类型与getDelegate() 方法返回的类型相同（我不知道那是什么，我称之为Delegate）
• 有 2 个构造函数：一个默认构造函数将 getDelegate 方法的结果分配给您的新字段，另一个将 Delegate 的实例作为参数并将其分配给您的字段
• 在测试中使用第二个构造函数创建MyClass 的实例并传递Delegate 类的模拟

看起来更像是这样：

class MyClass() {
    private Delegate delegate;

    MyClass() {
        this.delegate = DelegateImpl.getDelegate();
    }

    MyClass(Delegate delegate) {
        this.delegate = delegate;
    }
    // ... the rest
}