[java] PowerMock에서의 리플렉션 활용 방법

29 Nov 2023

java

PowerMock는 자바의 테스트 프레임워크로서, 리플렉션(reflection)을 사용하여 테스트 중인 클래스의 내부 상태를 조작할 수 있습니다. 리플렉션을 사용하면 private 메소드, 필드, 생성자 등에 접근할 수 있어서 테스트케이스 작성에 유용합니다.

PowerMock와 리플렉션을 함께 사용하는 방법을 살펴보겠습니다.

Step 1: Maven Dependencies 설정

<dependencies>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-module-junit4</artifactId>
        <version>2.0.7</version>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-api-mockito2</artifactId>
        <version>2.0.7</version>
    </dependency>
</dependencies>

위의 Maven 의존성을 추가하여 PowerMock를 프로젝트에 포함시킵니다.

Step 2: 리플렉션을 사용하여 테스트하기

리플렉션을 사용하여 테스트하려는 클래스의 메소드에 접근하는 방법은 다음과 같습니다.

@RunWith(PowerMockRunner.class)
@PrepareForTest(TargetClass.class)
public class ReflectionTest {

    @Test
    public void testPrivateMethodAccess() throws Exception {
        TargetClass target = new TargetClass();
        Method privateMethod = TargetClass.class.getDeclaredMethod("privateMethod", String.class);
        privateMethod.setAccessible(true);
        String result = (String) privateMethod.invoke(target, "test");
        
        assertEquals("test", result);
    }
    
    @Test
    public void testPrivateFieldAccess() throws Exception {
        TargetClass target = new TargetClass();
        Field privateField = TargetClass.class.getDeclaredField("privateField");
        privateField.setAccessible(true);
        privateField.set(target, "test");
        String result = (String) privateField.get(target);
        
        assertEquals("test", result);
    }
    
    // 테스트할 클래스
    public static class TargetClass {
        private String privateField;
        
        private String privateMethod(String arg) {
            return arg;
        }
    }
}

@RunWith(PowerMockRunner.class) 어노테이션을 테스트 클래스에 추가하여 PowerMock을 사용합니다. @PrepareForTest(TargetClass.class) 어노테이션은 리플렉션을 사용하기 위해 준비해야 하는 클래스를 지정합니다.

testPrivateMethodAccess() 메소드와 testPrivateFieldAccess() 메소드에서는 리플렉션을 사용하여 privateMethod()과 privateField에 접근합니다. 리플렉션을 사용하기 위해서는 setAccessible(true)을 호출하여 이를 허용해야 합니다.

위의 예제에서는 PowerMock와 Mockito를 함께 사용하였는데, powermock-api-mockito2 의존성을 추가하여 Mockito와 연동하여도 잘 작동합니다.

이렇게 PowerMock를 사용하여 리플렉션을 활용하여 테스트 케이스를 작성할 수 있습니다. PowerMock는 테스트 중인 코드의 내부 상태를 다룰 수 있기 때문에 복잡한 테스트 케이스를 작성하는 데 유용합니다.

더 자세한 내용은 PowerMock 공식 문서를 참고하시기 바랍니다.