[java] Mockito를 사용하여 특정 메소드에 전달된 인자들로 다른 메소드를 호출했을 때 클래스의 private 필드 값을 변경하는 방법은?

17 Nov 2023

java

Mockito는 Java에서 테스트를 위한 목 프레임워크로, 테스트 중에 객체의 동작을 가짜로 대체하고 결과를 조작할 수 있습니다. 따라서 Mockito를 사용하여 private 필드를 변경하는 방법은 목 객체를 생성하고 해당 목 객체를 통해 다른 메소드를 호출하는 것입니다.

다음은 Mockito를 사용하여 private 필드 값을 변경하는 예제 코드입니다.

import org.mockito.Mockito;

public class ExampleClass {

    private String privateField = "default value";

    public void methodA(String arg) {
        // arg를 사용하여 다른 메소드 호출
        String result = methodB(arg);
        
        // 결과에 따라 private 필드 값 변경
        if (result.equals("success")) {
            privateField = "new value";
        } else {
            privateField = "default value";
        }
    }
    
    private String methodB(String arg) {
        // 실제 구현 코드
        return "success";
    }
}

public class ExampleClassTest {

    @Test
    public void testMethodA() {
        // 목 객체 생성
        ExampleClass example = Mockito.spy(new ExampleClass());

        // methodB의 결과를 가짜로 설정
        Mockito.when(example.methodB(Mockito.anyString())).thenReturn("failure");

        // methodA 호출
        example.methodA("test");

        // private 필드 값 확인
        Assert.assertEquals("default value", example.privateField);

        // methodB의 결과를 가짜로 설정
        Mockito.when(example.methodB(Mockito.anyString())).thenReturn("success");

        // methodA 호출
        example.methodA("test");

        // private 필드 값 확인
        Assert.assertEquals("new value", example.privateField);
    }
}

위의 코드에서 ExampleClass는 변경하려는 private 필드 privateField를 가지고 있습니다. methodA는 전달된 인자를 사용하여 methodB를 호출한 다음, methodB의 결과에 따라 privateField 값을 변경합니다.

ExampleClassTest에서는 ExampleClass의 객체를 생성하고 Mockito의 spy 메소드를 사용하여 목 객체를 생성합니다. methodB의 결과를 가짜로 설정하여 methodA를 호출한 뒤, privateField의 값을 확인합니다.

이를 통해 Mockito를 이용하여 private 필드 값을 변경하는 방법을 알아봤습니다. Mockito를 사용하면 테스트 중에 객체의 동작을 조작하고 결과를 제어할 수 있으므로 유용하게 사용할 수 있습니다.

더 자세한 내용은 Mockito 공식 문서를 참고하세요.