[java] TestContainers를 사용하여 AWS Lambda 테스트하기

16 Nov 2023

java

AWS Lambda는 서버리스 아키텍처를 구현하기 위한 서비스로, 코드를 실행하고 스케일링 및 관리를 자동으로 처리합니다. 이러한 Lambda 함수를 테스트하는 것은 매우 중요한데, 특히 다른 AWS 서비스와의 통합이 있는 경우 실제 환경과 유사한 테스트를 수행해야 합니다.

TestContainers는 도커를 사용하여 테스트 환경을 구성해주는 도구입니다. 이를 활용하여 AWS Lambda 함수를 테스트할 수 있습니다. 이 글에서는 TestContainers를 사용하여 AWS Lambda를 테스트하는 방법을 알아보겠습니다.

Step 1: TestContainers 라이브러리 추가

TestContainers는 Java로 작성된 테스트 라이브러리이므로, 프로젝트의 의존성 관리 도구 (예: Maven 또는 Gradle)에 다음과 같은 의존성을 추가해야 합니다.

<dependency>
    <groupId>org.testcontainers</groupId>
    <artifactId>testcontainers</artifactId>
    <version>1.15.3</version>
    <scope>test</scope>
</dependency>

Step 2: AWS Lambda 테스트 클래스 작성

다음으로, AWS Lambda 함수를 테스트하기 위한 테스트 클래스를 작성해야 합니다. 이 테스트 클래스는 @RunWith 어노테이션을 사용하여 TestContainers와 연동되도록 설정해야 합니다. 또한 @Container 어노테이션을 사용하여 테스트 컨테이너를 생성해야 합니다.

@RunWith(DockerTestRunner.class)
public class LambdaFunctionTest {

    @Container
    public static GenericContainer<?> dockerContainer = new GenericContainer<>("amazon/aws-lambda-java:11");

    @Before
    public void setup() {
        // 테스트 환경 설정 코드
    }

    @Test
    public void testLambdaFunction() {
        // Lambda 함수 테스트 코드
    }

    @After
    public void tearDown() {
        // 테스트 종료 후 처리 코드
    }
}

위의 코드에서 dockerContainer는 AWS Lambda 기반의 테스트 컨테이너를 생성합니다. 필요한 경우 테스트 컨테이너의 설정을 변경하여 사용자 정의 환경으로 구성할 수도 있습니다.

Step 3: AWS Lambda 함수 테스트

실제로 AWS Lambda 함수를 테스트하는 코드를 작성해야 합니다. 이 코드는 테스트 컨테이너를 사용하여 로컬 환경에서 Lambda 함수를 실행하는 내용을 포함해야 합니다.

@Test
public void testLambdaFunction() {
    AWSLambda lambdaClient = AWSLambdaClientBuilder.defaultClient();

    // 테스트할 Lambda 함수 호출
    InvokeRequest request = new InvokeRequest()
                                .withFunctionName("my-lambda-function")
                                .withPayload("{\"key1\":\"value1\", \"key2\":\"value2\"}");

    InvokeResult result = lambdaClient.invoke(request);

    // 결과 처리 코드
    String output = new String(result.getPayload().array());
    // 테스트 결과 검증 코드
}

위의 코드에서 my-lambda-function은 테스트할 Lambda 함수의 이름을 의미합니다. 필요한 경우 호출할 함수의 입력값을 변경하여 여러 시나리오에 대한 테스트를 수행할 수도 있습니다.

결론

TestContainers를 사용하여 AWS Lambda를 테스트하는 것은 매우 간단하고 효과적입니다. AWS Lambda와 관련된 다양한 환경에서의 테스트를 수행할 수 있으며, 더 신뢰성 높은 애플리케이션을 개발하는 데 도움이 됩니다.

더 자세한 정보는 TestContainers 공식 문서를 참조하십시오.