테스트 코드 작성 입문 | Arrange-Act-Assert(AAA) 패턴으로 단위 테스트의 기본 구조 이해하기

pytest로 테스트를 작성하기 시작하면 “테스트 코드를 어떻게 구성해야 할까?”라고 고민하게 됩니다. 그럴 때 꼭 알아두고 싶은 것이 Arrange-Act-Assert(AAA 패턴)입니다. 「준비 → 실행 → 확인」의 3단 구조로 작성하기만 하면, 읽기 쉽고 유지보수하기 쉬운 테스트 코드를 작성할 수 있게 됩니다.

📌 이 글의 대상 독자

테스트 코드를 처음 작성했을 때, 필자도 이런 코드를 작성했습니다.

# ❌ 구조가 없어 무엇을 테스트하는지 알기 어렵다
def test_calc():
    assert add(1, 2) == 3
    assert add(-1, 1) == 0
    assert add(0, 0) == 0
    result = add(100, 200)
    assert result == 300
    assert add(1.5, 2.5) == 4.0

얼핏 보면 문제없어 보이지만, 어느 하나가 실패했을 때 「어떤 테스트가 실패한 건지」 바로 알 수 없습니다. 또 여러 검증이 혼재해 있어 테스트 의도가 전달되기 어렵습니다. 이 문제를 해결하는 것이 Arrange-Act-Assert 패턴입니다.

Arrange-Act-Assert(AAA) 패턴이란?

AAA 패턴이란 테스트 코드를 3개의 페이즈로 나눠서 작성하는 구조입니다. 어떤 프로그래밍 언어·테스트 프레임워크에서도 사용할 수 있는 보편적인 패턴으로, 테스트 업계에서는 사실상 표준으로 널리 사용되고 있습니다.

AAA 패턴의 기본형

앞서의 테스트를 AAA 패턴으로 다시 작성하면 이렇게 됩니다.

# ✅ AAA 패턴으로 작성한 경우
def test_add_returns_correct_sum_with_positive_numbers():
    # Arrange（준비）: 테스트에 필요한 데이터를 준비한다
    a = 2
    b = 3

    # Act（실행）: 테스트 대상의 처리를 실행한다
    result = add(a, b)

    # Assert（검증）: 결과가 기대대로인지 확인한다
    assert result == 5


def test_add_returns_zero_when_adding_opposite_numbers():
    # Arrange
    a = -1
    b = 1

    # Act
    result = add(a, b)

    # Assert
    assert result == 0

각 페이즈의 상세와 작성법

① Arrange（준비）：테스트의 전제 조건을 만든다

Arrange 페이즈에서는 테스트에 필요한 변수·객체·데이터를 준비합니다. 이 페이즈가 길어지는 경우는 pytest fixture로 꺼내면 정리됩니다.

def test_user_full_name():
    # Arrange: 테스트에 필요한 객체를 생성한다
    user = User(first_name="김", last_name="민준")

    # Act
    result = user.get_full_name()

    # Assert
    assert result == "김 민준"

② Act（실행）：검증하고 싶은 처리를 실행한다

Act 페이즈에서는 검증하고 싶은 처리를 실행합니다. 이상적으로는 「하나의 목적에 집중한 조작」이지만, 실무에서는 로그인 후에 조작하거나 카트에 추가한 후 합계를 가져오는 등 여러 단계가 필요한 케이스도 보통 있습니다. 그런 경우에도 「이 테스트에서 무엇을 검증하고 싶은지」가 명확해지도록 의식합시다.

# ⚠️ Arrange와 Act가 혼재（구조를 알기 어렵다）
def test_bad_example():
    cart = ShoppingCart()
    cart.add_item("apple", price=100)
    cart.add_item("banana", price=150)
    cart.apply_discount(10)       # ← 할인을 테스트하는 건지? 합계만?
    result = cart.total()
    assert result == 225

# ✅ 무엇을 검증하는지를 명확하게 한 작성법
def test_total_is_correct_after_adding_two_items():
    # Arrange（전제 조건：카트에 2개의 상품이 들어 있다）
    cart = ShoppingCart()
    cart.add_item("apple", price=100)
    cart.add_item("banana", price=150)

    # Act（검증 대상：합계 금액 취득）
    result = cart.total()

    # Assert
    assert result == 250

③ Assert（검증）：기댓값과 실제 결과를 비교한다

Assert 페이즈에서는 assert 문으로 결과를 검증합니다. 1개의 테스트 함수에 assert는 1~3개 정도가 기준입니다. pytest는 assert 문 실패 시 상세한 차이를 자동으로 표시해 주므로, 복잡한 assertion 라이브러리 없이도 심플하게 작성할 수 있습니다.

# ✅ 심플한 assert（권장）
assert result == 5
assert result is not None
assert "error" not in response_body
assert status_code in (200, 201)

# ✅ 에러 메시지 포함（실패 시 원인을 알기 쉽다）
assert result == 5, f"기댓값: 5, 실제 값: {result}"

# ❌ 1개의 함수에 assert를 너무 많이 넣음
def test_user():
    user = create_user("김민준", "minjun@example.com")
    assert user.name == "김민준"
    assert user.email == "minjun@example.com"
    assert user.is_active is True
    assert user.role == "user"
    assert user.created_at is not None  # 이것만으로 5개의 확인

읽기 쉬운 테스트 함수명 작성법

AAA 패턴과 마찬가지로 중요한 것이 테스트 함수명입니다. 함수명만 읽어도 「무엇을 테스트하는지」를 알 수 있으면, 실패 시 조사가 훨씬 빨라집니다.

권장 패턴：test_[무엇을]_[어떤 조건에서]_[어떤 결과가 나오는지]

길어 보일 수 있지만, 테스트 결과 목록에 나열되었을 때 함수명만으로 「무엇이 실패했는지」를 즉시 알 수 있습니다. 로그를 읽는 시간이 크게 줄어듭니다.

import pytest

@pytest.mark.parametrize("a, b, expected", [
    (2,  3,   5),   # 정상계
    (0,  0,   0),   # 경계값
    (-1, 1,   0),   # 음수
])
def test_add_returns_correct_sum(a, b, expected):
    result = add(a, b)          # Act
    assert result == expected   # Assert

실전：자주 있는 테스트를 AAA 패턴으로 작성하기

예①：계산 함수 테스트

def add(a: float, b: float) -> float:
    return a + b


def test_add_returns_correct_sum_with_positive_numbers():
    # Arrange
    a = 3
    b = 4

    # Act
    result = add(a, b)

    # Assert
    assert result == 7


def test_add_returns_negative_when_both_negative():
    # Arrange
    a = -3
    b = -4

    # Act
    result = add(a, b)

    # Assert
    assert result == -7

예②：예외가 발생하는 것을 테스트하기

import pytest


def divide(a: float, b: float) -> float:
    if b == 0:
        raise ValueError("0으로 나눌 수 없습니다")
    return a / b


def test_divide_raises_value_error_when_divisor_is_zero():
    # Arrange
    a = 10
    b = 0

    # Act & Assert（예외 테스트는 Act와 Assert가 하나가 된다）
    with pytest.raises(ValueError, match="0으로 나눌 수 없습니다"):
        divide(a, b)

예③：클래스 메서드 테스트

class ShoppingCart:
    def __init__(self) -> None:
        self._items: list[dict] = []

    def add_item(self, name: str, price: int) -> None:
        self._items.append({"name": name, "price": price})

    def total(self) -> int:
        return sum(item["price"] for item in self._items)


def test_total_returns_sum_of_all_item_prices():
    # Arrange
    cart = ShoppingCart()
    cart.add_item("apple", price=100)
    cart.add_item("banana", price=200)

    # Act
    result = cart.total()

    # Assert
    assert result == 300


def test_total_returns_zero_when_cart_is_empty():
    # Arrange
    cart = ShoppingCart()

    # Act
    result = cart.total()

    # Assert
    assert result == 0

AAA 패턴에서 빠지기 쉬운 주의 포인트

자주 묻는 질문 (FAQ)

정리

AAA 패턴은 「어려운 테크닉」이 아니라, 테스트 코드를 읽는 사람에 대한 배려입니다. 미래의 자신이나 팀 멤버가 「무엇을 테스트하는지」를 한눈에 이해할 수 있는 코드를 작성하는 것이, 유지보수하기 쉬운 테스트 스위트로의 첫 걸음입니다. 다음에 작성할 테스트 함수부터 AAA 패턴을 의식해보세요.