[java] PowerMock에서의 필드 주입 방법

29 Nov 2023

java

PowerMock은 Java에서 작성된 코드의 단위 테스트 시에 테스트 대상 클래스의 private 필드에 값을 주입하기 위해 사용되는 도구입니다. 이 글에서는 PowerMock을 사용하여 필드 주입을 수행하는 방법에 대해 알아보겠습니다.

1. PowerMockito.mockStatic 메서드를 사용한 필드 주입

PowerMockito 클래스의 mockStatic 메서드를 사용하여 static 필드를 주입할 수 있습니다. 아래의 예시 코드를 참고해주세요.

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

public class MyClass {
    private static String myField;
    
    public static String getMyField() {
        return myField;
    }
}

public class MyClassTest {
    @Test
    public void testMyFieldInjection() {
        mockStatic(MyClass.class);
        MyClass myClass = new MyClass();
        Mockito.when(MyClass.getMyField()).thenReturn("Injected Value");
        
        String result = myClass.getMyField();
        
        assertEquals("Injected Value", result);
    }
}

위의 예시 코드에서는 MyClass 클래스의 static 필드인 myField를 주입하기 위해 mockStatic 메서드를 사용하였습니다. 그 후에 Mockito의 Mock.when 메서드를 사용하여 해당 필드의 반환 값을 지정하였습니다. 따라서 테스트에서 MyClass.getMyField()를 호출하면 “Injected Value”가 반환될 것입니다.

2. @PrepareForTest 어노테이션을 사용한 필드 주입

때로는 인스턴스 변수에 대한 필드 주입이 필요할 수 있습니다. 이 경우에는 @PrepareForTest 어노테이션을 사용하여 필드 주입을 수행할 클래스를 지정해야 합니다.

import org.powermock.core.classloader.annotations.PrepareForTest;

@PrepareForTest(MyClass.class)
public class MyClassTest {
    @Test
    public void testMyFieldInjection() {
        MyClass myClass = new MyClass();
        Mockito.when(myClass.getMyField()).thenReturn("Injected Value");
        
        String result = myClass.getMyField();
        
        assertEquals("Injected Value", result);
    }
}

위의 예시 코드에서는 @PrepareForTest 어노테이션을 사용하여 MyClass 클래스를 지정하여 필드 주입을 수행하였습니다. 이제는 MyClass 인스턴스의 getMyField 메서드가 호출될 때 “Injected Value”가 반환될 것입니다.

결론

PowerMock은 Java에서 테스트 가능한 코드를 작성할 수 있도록 도와주는 강력한 도구입니다. 필드 주입을 통해 테스트 대상 클래스의 상태를 적절하게 조작할 수 있으므로, 효율적이고 신뢰성있는 단위 테스트를 작성할 수 있습니다.

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