[java] JUnit으로 AWS 리소스 테스트하기

14 Nov 2023

java

AWS(Amazon Web Services)는 클라우드 컴퓨팅 플랫폼으로 다양한 서비스를 제공합니다. 이러한 AWS 리소스를 테스트하기 위해서는 JUnit과 AWS SDK를 함께 사용할 수 있습니다. 이번 블로그에서는 JUnit을 사용하여 AWS 리소스를 테스트하는 방법에 대해 알아보겠습니다.

1. AWS SDK 설정하기

먼저 AWS SDK를 프로젝트에 추가해야 합니다. 여러분의 프로젝트에 맞는 SDK를 선택하고, 해당 SDK를 Maven이나 Gradle 등의 의존성 관리 도구를 사용하여 추가해주세요.

<dependency>
    <groupId>software.amazon.awssdk</groupId>
    <artifactId>aws-sdk-java</artifactId>
    <version>2.17.60</version>
</dependency>

2. 테스트 클래스 작성하기

JUnit을 사용하여 AWS 리소스를 테스트하기 위해선 다음과 같이 테스트 클래스를 작성해야 합니다.

import software.amazon.awssdk.services.s3.S3Client;
import software.amazon.awssdk.services.s3.model.CreateBucketRequest;
import software.amazon.awssdk.services.s3.model.DeleteBucketRequest;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import static org.junit.Assert.*;

public class AwsResourceTest {

    private S3Client s3Client;
    private String bucketName;

    @Before
    public void setUp() {
        // AWS 클라이언트 생성
        s3Client = S3Client.create();
        // 버킷 이름 생성
        bucketName = "my-test-bucket";
    }

    @After
    public void tearDown() {
        // 테스트 종료 후 생성한 버킷 삭제
        DeleteBucketRequest deleteBucketRequest = DeleteBucketRequest.builder()
            .bucket(bucketName)
            .build();
        s3Client.deleteBucket(deleteBucketRequest);
    }

    @Test
    public void testCreateBucket() {
        // 버킷 생성 요청
        CreateBucketRequest createBucketRequest = CreateBucketRequest.builder()
            .bucket(bucketName)
            .build();
        s3Client.createBucket(createBucketRequest);
        
        // 생성한 버킷이 존재하는지 검증
        assertTrue(s3Client.listBuckets().buckets().stream()
                .anyMatch(bucket -> bucket.name().equals(bucketName)));
    }
}

3. AWS 리소스 테스트하기

위에서 작성한 테스트 클래스는 JUnit에 의해 실행될 때 @Before 어노테이션이 지정된 setUp 메서드가 가장 먼저 실행됩니다. setUp 메서드에서는 S3 클라이언트를 생성하고, 테스트에 사용할 버킷 이름을 설정합니다.

테스트 메서드인 testCreateBucket에서는 S3 버킷 생성을 요청하고, 생성된 버킷이 정상적으로 존재하는지를 확인합니다. assertTrue를 사용하여 검증 결과를 확인합니다.

테스트가 완료되면 @After 어노테이션이 지정된 tearDown 메서드가 실행됩니다. 이 메서드에서는 테스트 종료 후에 생성한 버킷을 삭제합니다.

4. 테스트 실행하기

JUnit을 이용하여 테스트를 실행하려면 위에서 작성한 테스트 클래스를 실행하면 됩니다. 대부분의 IDE는 JUnit을 지원하기 때문에, IDE 상에서도 테스트 실행이 가능합니다.

테스트 실행 결과는 성공한 테스트와 실패한 테스트에 대한 정보를 제공합니다. 성공한 테스트의 경우에는 녹색으로 표시되고, 실패한 테스트의 경우에는 빨간색으로 표시됩니다.

5. 마무리

JUnit을 사용하여 AWS 리소스를 테스트하는 방법에 대해 알아보았습니다. JUnit을 활용하면 AWS 리소스의 생성, 수정, 삭제 등의 작업을 테스트할 수 있으며, 이를 통해 개발 및 테스트 작업을 보다 효율적으로 수행할 수 있습니다.

더 자세한 정보는 JUnit 공식 문서와 AWS SDK for Java 문서를 참고하시기 바랍니다.