Mock AWSSecretsManager时getSecret返回Null问题排查

核心问题原因 #

1. 待测试方法硬编码创建客户端：AWSClient的getSecretsManagerSecret方法内部直接通过AWSSecretsManagerClientBuilder.standard().build()生成真实的AWSSecretsManager实例，你Mock的secretsManagerClient完全没被用到，导致Mock配置无效。
2. 错误Mock了待测试类：测试代码中Mock了AWSClient本身，调用awsClient.getSecretsManagerSecret(ID)时，实际执行的是Mock对象的空实现，而非真实业务逻辑。

修复步骤 #

第一步：改造AWSClient类，支持依赖注入

修改类结构，通过构造函数传入AWSSecretsManager实例，既满足生产环境的默认创建，又支持测试时注入Mock对象：

public class AWSClient{

    private final AWSSecretsManager secretsManagerClient;

    // 测试用构造函数：注入自定义的SecretsManager客户端
    public AWSClient(AWSSecretsManager secretsManagerClient) {
        this.secretsManagerClient = secretsManagerClient;
    }

    // 生产用构造函数：默认创建真实客户端
    public AWSClient() {
        this.secretsManagerClient = AWSSecretsManagerClientBuilder.standard().build();
    }

    public String getSecretsManagerSecret(String id) {
        GetSecretValueRequest request = new GetSecretValueRequest().withSecretId(id);
        GetSecretValueResult result = secretsManagerClient.getSecretValue(request);
        return result.getSecretString();
    }
}

第二步：修正测试代码

使用真实的AWSClient实例，注入Mock的AWSSecretsManager客户端，确保Mock逻辑生效：

@Mock AWSSecretsManager secretsManagerClient;
@Mock GetSecretValueResult secretValueResult;

private static final String SECRET_VALUE = "value";
private static final String ID = "id";

private AWSClient awsClient;

@Before
public void setUp() {
    MockitoAnnotations.openMocks(this);
    // 注入Mock的SecretsManager客户端到真实的AWSClient实例
    awsClient = new AWSClient(secretsManagerClient);
}

@Test
public void handleRequest() {
    // 配置Mock行为
    when(secretsManagerClient.getSecretValue(any(GetSecretValueRequest.class)))
        .thenReturn(secretValueResult);
    when(secretValueResult.getSecretString()).thenReturn(SECRET_VALUE);

    String response = awsClient.getSecretsManagerSecret(ID);
    assertNotNull(response);
    assertEquals(SECRET_VALUE, response); // 验证返回值符合预期
}

关键说明 #

• 你无需MockAWSClient本身，反而应该使用真实实例，Mock它的外部依赖（AWSSecretsManager），这样才能测试真实的业务逻辑。
• 依赖注入的改造不会影响生产环境使用，无参构造仍保持原有逻辑，仅新增了测试用的构造函数。

内容的提问来源于stack exchange，提问作者Sparrow