如何为使用 ClientBuilders 的 AWSLambda/AWSStepFunctions 模拟新创建的 object？

Question

我有一个 class，我正在尝试测试它在哪里创建一个新的 AWSStepFunctions 类型的 object，并且我试图模拟它以返回相同类型的模拟。 我无法更改这行原始代码，我只能添加测试，所以我想知道如何 go 关于 mocking 它。

class 看起来像这样——

Class class{
    public Object handleRequest(Object object, Context context) {
        AWSStepFunctions client = AWSStepFunctionsClientBuilder.standard().withClientConfiguration(new ClientConfiguration()).build();
        client.startExecution(...);
    }
}

测试代码如下所示 -

public class ClassTest {
    @Test
    public void testHandlerRequest() {
        mockStatic(AWSStepFunctionsClientBuilder.class);  //mockStatic() and when() from PowerMockito
        AWSStepFunctions awsStepFunctionsMock = mock(AWSStepFunctions.class);
        AWSStepFunctionsClientBuilder awsStepFunctionsClientBuilder = mock(AWSStepFunctionsClientBuilder.class);
        ClientConfiguration configuration = mock(ClientConfiguration.class);

        PowerMockito.whenNew(ClientConfiguration.class).withAnyArguments().thenReturn(awsStepFunctionsMock);

        when(awsStepFunctionsClientBuilder.standard()).thenReturn(awsStepFunctionsClientBuilder);        
        when(awsStepFunctionsClientBuilder.withClientConfiguration()).thenReturn(awsStepFunctionsClientBuilder);
        when(awsStepFunctionsClientBuilder.build()).thenReturn(awsStepFunctionsMock);
        ... more when-thenreturns
    }
}

对于 clientBuilder 的模拟，我遇到了诸如 NoSuchMethodError 之类的错误。

我尝试使用 PowerMockito 的 whenNew 来模拟 AWSStepFunctions - PowerMockito.whenNew(AWSStepFunctions.class).withAnyArguments().thenReturn(awsStepFunctionsMock) 类型的新 object 的创建，但这似乎也不起作用。 有没有办法正确返回这个模拟？

Answer 1

您可以使用 Mockito 和 Junit5 直接模拟 static 方法，而无需使用 Powermock。

类测试

    @Test
    void test() throws IOException {
        try (MockedStatic<AWSStepFunctionsClientBuilder> awsMock = Mockito.mockStatic(AWSStepFunctionsClientBuilder.class, Mockito.RETURNS_DEEP_STUBS)) {
            AWSStepFunctions awsStepFunctionsMock = mock(AWSStepFunctions.class);
            // You can mock methods chaining when you specify Mockito.RETURNS_DEEP_STUBS
            awsMock.when(() -> AWSStepFunctionsClientBuilder.standard().withClientConfiguration(Mockito.any()).build()).thenReturn(awsStepFunctionsMock);
            
        }
    }

您可以阅读这篇文章以获取有关 MockedStatic 的更多说明： https://www.baeldung.com/mockito-mock-static-methods
而这个关于 Mockito.RETURNS_DEEP_STUBS: https://www.baeldung.com/mockito-fluent-apis

不要忘记配置 Mockito 来处理 static 模拟：
测试/资源/mockito-extensions/org.mockito.plugins.MockMaker

mock-maker-inline