如何用Test Containers或开源工具测试Spring Boot的AWS Athena集成？

1. Test Containers + LocalStack开源版（免费可用） #

LocalStack并非只有付费版，它的开源核心版完全支持Athena服务模拟，能直接通过Test Containers集成到Spring Boot测试流程中，无需付费。

配置步骤： #

• 添加Test Containers LocalStack依赖：
  Maven（pom.xml）：

  <dependency>
      <groupId>org.testcontainers</groupId>
      <artifactId>localstack</artifactId>
      <scope>test</scope>
  </dependency>
  

  Gradle（build.gradle）：

  testImplementation 'org.testcontainers:localstack'
  

• 在测试类中配置LocalStack容器，指定启用Athena服务：

  @Testcontainers
  @SpringBootTest
  public class AthenaIntegrationTest {
  
      @Container
      private static final LocalStackContainer localStack = new LocalStackContainer(DockerImageName.parse("localstack/localstack:latest"))
              .withServices(LocalStackContainer.Service.ATHENA);
  
      @DynamicPropertySource
      static void setAwsProperties(DynamicPropertyRegistry registry) {
          registry.add("aws.access-key", localStack::getAccessKey);
          registry.add("aws.secret-key", localStack::getSecretKey);
          registry.add("aws.region", localStack::getRegion);
          registry.add("aws.athena.endpoint", () -> localStack.getEndpointOverride(LocalStackContainer.Service.ATHENA).toString());
      }
  
      // 编写Athena测试用例，比如提交查询、校验执行结果等
      @Test
      public void testQueryExecution() {
          // 注入AthenaClient，执行具体测试逻辑
      }
  }
  

  这套配置可以在测试环境中模拟完整的Athena服务，覆盖集成测试场景。

2. 开源替代：Trino（原PrestoSQL） #

Athena底层基于Presto，Trino是Presto的开源分支，SQL语法与Athena高度兼容，可用来模拟Athena的查询环境。

测试思路： #

• 用Test Containers启动Trino容器，同时搭配MinIO（开源S3兼容存储）模拟Athena的数据源存储。
• 配置Spring Boot的Athena客户端指向Trino的端点，即可运行真实的查询测试，适合需要验证SQL兼容性的场景。

3. 单元测试：Mock Athena客户端 #

如果仅需验证业务逻辑，无需真实查询执行，可以用Mockito直接Mock AthenaClient：

@ExtendWith(MockitoExtension.class)
public class AthenaServiceUnitTest {

    @Mock
    private AthenaClient athenaClient;

    @InjectMocks
    private YourAthenaBusinessService athenaService;

    @Test
    public void testExecuteQuery() {
        // 模拟客户端返回的查询ID
        when(athenaClient.startQueryExecution(any(StartQueryExecutionRequest.class)))
                .thenReturn(StartQueryExecutionResponse.builder()
                        .queryExecutionId("test-query-id")
                        .build());

        String result = athenaService.runSampleQuery();
        assertEquals("test-query-id", result);
    }
}

这种方式轻量快速，适合单元测试阶段，无需依赖任何外部服务。

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