[JUnit] @AfterAll – 전체 테스트 종료 후 한 번만 실행되는 작업

JUnit 테스트를 작성하다 보면 테스트가 모두 끝난 뒤 공통으로 수행해야 하는 정리 작업이 생길 때가 있습니다. 예를 들어 데이터베이스 연결 해제, 외부 리소스 종료, 임시 파일 삭제 같은 작업입니다. 이런 작업을 매 테스트마다 반복할 필요는 없고, 테스트 전체가 끝났을 때 한 번만 수행하는 것이 효율적입니다. 이를 위해 JUnit은 @AfterAll 기능을 제공합니다.

이 글에서는 @AfterAll의 역할과 사용법, 주의할 점을 초보자도 이해하기 쉽게 정리합니다.

 

1. @AfterAll이란?

@AfterAll이 붙은 메서드는 테스트 클래스의 모든 테스트가 끝난 후 한 번만 실행되는 정리 코드입니다.
보통 @BeforeAll에서 열어둔 리소스를 종료하거나 테스트 전체가 끝날 때 해야 할 마무리 작업에 사용됩니다.

실행 순서

• 테스트 클래스 시작
• @BeforeAll 실행
• @BeforeEach → @Test → @AfterEach 반복
• 모든 테스트 종료 후 @AfterAll 실행

 

2. @AfterAll 사용 조건

JUnit5에서는 다음 조건을 만족해야 실행됩니다.

1. 메서드는 static이어야 함
2. 반환 타입은 void
3. 매개변수는 일반적으로 없음

단, @TestInstance(TestInstance.Lifecycle.PER_CLASS)를 사용하면 static이 아닌 메서드도 가능합니다.

기본 형태

@AfterAll
static void tearDownAll() {
    // 전체 테스트 종료 후 수행할 정리 작업
}

 

3. 예제: 테스트 종료 후 DB 연결 정리하기

아래 예제는 테스트 전체에서 사용하는 DB 연결을 테스트 마지막에 한 번만 종료하는 코드입니다.

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;

class DatabaseTest {

    private static DatabaseConnection connection;

    @BeforeAll
    static void setUp() {
        connection = new DatabaseConnection();
        connection.connect();
    }

    @AfterAll
    static void tearDown() {
        connection.close(); // 테스트 끝날 때 DB 연결 정리
    }

    @Test
    void testQuery() {
        String result = connection.select("SELECT name FROM user WHERE id = 1");
        assertEquals("Tom", result);
    }

    @Test
    void testInsert() {
        boolean success = connection.insert("INSERT INTO user(name) VALUES('Bob')");
        assertTrue(success);
    }
}

코드 설명

• 테스트 전체에서 DB 연결을 공유
• 테스트가 모두 끝난 뒤 @AfterAll에서 연결을 종료
• 불필요한 리소스 점유를 방지하고 테스트 환경을 깔끔하게 유지

 

4. @AfterAll과 @AfterEach의 차이

기능	실행 시점	실행 횟수	사용 목적
@AfterAll	모든 테스트 종료 후	1회	전체 리소스 정리
@AfterEach	각 테스트 종료 후	테스트마다	테스트별 개별 정리

즉,

• 전체 테스트의 공통 리소스 종료 → @AfterAll
• 각 테스트마다 개별 정리 필요 → @AfterEach

 

5. @AfterAll 사용 시 주의할 점

1. 전역 리소스만 정리하도록 설계하기
   • @AfterAll은 단 한 번만 실행되므로, 전체 테스트가 공유하는 리소스만 종료해야 합니다.
2. 테스트 도중 리소스가 변경되지 않도록 주의
   • 공유 객체가 테스트에서 수정되면 종료 시점에 예기치 않은 문제가 발생할 수 있습니다.
   • 가능하면 읽기 전용 객체만 사용하거나, 변경을 최소화해야 합니다.
3. 무거운 리소스 종료는 반드시 여기서 처리하기
   • DB, 파일 핸들, 스레드풀 등은 테스트 종료 후 종료하는 것이 메모리 누수 방지에 필수입니다.
4. @BeforeAll과 항상 세트로 생각하기
   • @BeforeAll에서 리소스를 열었다면
     → @AfterAll에서 반드시 닫는 흐름을 유지해야 코드 품질이 좋아집니다.