[java] Mockito를 사용하여 객체의 private 메소드에 값을 주입하는 방법은?

17 Nov 2023

java

Mockito는 자바에서 유닛 테스트를 작성할 때 많이 사용되는 프레임워크입니다. Mockito를 사용하면 테스트 대상 객체의 메소드 호출을 가로채고, 원하는 값을 반환하거나 원하는 동작을 수행할 수 있습니다. 하지만 Mockito는 기본적으로 public 메소드에만 접근이 가능하므로, private 메소드에 값을 주입하는 것은 어려운 작업입니다.

하지만 Mockito는 리플렉션(Reflection)을 사용하여 private 메소드에 접근하고 값을 주입하는 기능도 제공합니다. 아래의 예제 코드를 통해 private 메소드에 값을 주입하는 방법을 확인해보세요.

import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;

class ExampleClass {
    private String privateMethod() {
        return "Original Value";
    }
}

public class MockitoPrivateMethodInjectionExample {
    public static void main(String[] args) throws IllegalAccessException, NoSuchMethodException, InvocationTargetException {
        ExampleClass exampleObject = new ExampleClass();

        // Mockito를 사용하여 private 메소드에 값을 주입하기 위해 리플렉션을 사용합니다.
        Method privateMethod = ExampleClass.class.getDeclaredMethod("privateMethod");
        privateMethod.setAccessible(true);
        privateMethod.invoke(exampleObject); // 이 부분에서 privateMethod에 어떤 값을 주입하고 싶은 경우 리플렉션을 사용하여 값을 주입할 수 있습니다.
    }
}

위의 예제 코드에서는 ExampleClass라는 클래스를 정의하고, 이 클래스에는 privateMethod()라는 private 메소드가 있습니다. MockitoPrivateMethodInjectionExample 클래스에서는 private 메소드에 값을 주입해보는 코드를 작성하였습니다.

Method privateMethod = ExampleClass.class.getDeclaredMethod("privateMethod")에서는 ExampleClass의 private 메소드를 얻어옵니다. 이때 setAccessible(true)를 사용하여 접근할 수 있도록 설정합니다. 그리고 privateMethod.invoke(exampleObject)를 통해 private 메소드를 호출합니다.

이렇게 하면 Mockito를 사용하여 private 메소드에 값을 주입할 수 있습니다. 하지만 private 메소드에 값을 주입하는 방법은 특수한 상황에서만 사용해야 하며, 테스트할 코드를 리팩토링하여 public 메소드로 분리하는 것이 더 좋은 방법일 수도 있습니다.