[java] PowerMock에서의 동적 SQL 모킹

29 Nov 2023

java

PowerMock은 자바 단위 테스트 프레임워크인데, 다른 라이브러리에서 호출된 정적 메서드나 생성자를 모킹하는 기능을 제공합니다. 이 기능을 이용하면 테스트에서 외부 시스템과의 연동이나 시간이 오래 걸리는 작업을 피할 수 있습니다.

이번에는 PowerMock에서 동적 SQL 모킹을 하는 방법에 대해 알아보겠습니다.

필요한 의존성 추가

먼저 프로젝트에 PowerMock 의존성을 추가해야 합니다. Maven을 사용한다면 pom.xml 파일에 다음과 같이 의존성을 추가하세요:

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito2</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>

Gradle을 사용한다면 build.gradle 파일에 다음과 같이 의존성을 추가하세요:

testImplementation 'org.powermock:powermock-module-junit4:2.0.9'
testImplementation 'org.powermock:powermock-api-mockito2:2.0.9'

동적 SQL 모킹하기

@RunWith(PowerMockRunner.class) 어노테이션을 사용하여 테스트 클래스를 PowerMockRunner로 실행하도록 설정합니다.
@PrepareForTest 어노테이션을 사용하여 테스트 중에 모킹할 클래스를 준비합니다. SQL 실행 코드가 포함된 클래스를 준비해야 합니다.
PowerMockito.mockStatic() 메서드를 사용하여 정적 메서드를 모킹할 클래스를 지정합니다.
PowerMockito.when() 메서드와 thenReturn() 메서드를 사용하여 실행한 SQL에 대한 결과를 지정합니다.

아래는 위 방법을 이용하여 동적 SQL을 모킹하는 예제 코드입니다.

import static org.powermock.api.mockito.PowerMockito.*;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import static org.junit.Assert.assertEquals;

@RunWith(PowerMockRunner.class)
@PrepareForTest({ Database.class }) // 동적 모킹할 클래스를 여기에 추가
public class DatabaseTest {

    @Test
    public void testQuery() {
        // 모킹할 클래스를 정의
        mockStatic(Database.class);

        // 모킹 결과 지정
        when(Database.executeQuery("SELECT COUNT(*) FROM users")).thenReturn(10);

        // 실제 테스트
        int result = Database.executeQuery("SELECT COUNT(*) FROM users");
        assertEquals(10, result);
    }
}

위의 코드에서 Database는 동적으로 호출되는 SQL을 실행하는 클래스입니다. executeQuery() 메서드는 SQL 쿼리를 받아 실행하고 결과를 반환합니다. 테스트 메서드에서는 모킹 결과를 지정하고, 실제로 동일한 SQL을 실행해 결과를 비교하는 방식으로 테스트를 진행합니다.

정리

PowerMock을 사용하면 동적 SQL을 모킹하여 테스트할 수 있습니다. 위에 설명한 방법을 따라가면 PowerMock을 사용하여 동적 SQL을 모킹하는 테스트를 작성할 수 있습니다. 관련된 자세한 내용은 PowerMock의 공식 문서를 참고하세요.

참고 자료

PowerMock 공식 문서