PowerMock은 Java의 테스트 프레임워크 중 하나로, 모의 객체(Mock Object)를 사용하여 테스트를 수행할 수 있습니다. 이 중에서도 PowerMock의 객체 생성 감시 기능은 특히 유용한 테스트 방법입니다.

객체 생성 감시란, 테스트 중에 특정 클래스의 객체가 생성되는지를 확인하고, 필요한 경우 이를 가짜(Mock) 객체로 대체할 수 있는 기능을 말합니다. 이를 통해 테스트 대상 코드와 연관된 의존 객체를 통제할 수 있으며, 테스트의 격리성을 높일 수 있습니다.

@PrepareForTest 어노테이션

PowerMock에서는 @PrepareForTest 어노테이션을 사용하여 객체 생성 감시를 설정합니다. 이 어노테이션은 대상 클래스나 메소드가 호출되기 전에 해당 클래스의 생성자를 감시할 것을 명시합니다.

@RunWith(PowerMockRunner.class)
@PrepareForTest(TargetClass.class)
public class MyTestClass {
    // 테스트 코드 작성
}

위의 예시 코드에서는 TargetClass의 생성자 호출을 감시하는 테스트 코드를 작성하고 있습니다. 해당 테스트 클래스는 PowerMockRunner를 사용하여 실행됩니다.

expectNew() 메소드

객체 생성 감시를 사용하는 가장 간단한 방법은 PowerMockito의 expectNew() 메소드를 사용하는 것입니다. 이를 통해 객체가 생성될 때 원하는 모의 객체를 반환하거나, 예외를 발생시킬 수 있습니다.

@RunWith(PowerMockRunner.class)
@PrepareForTest(TargetClass.class)
public class MyTestClass {
    
    @Test
    public void testMethod() {
        // 객체 생성 감시 설정
        PowerMockito.whenNew(TargetClass.class).withNoArguments().thenReturn(mockTarget);
        
        // 테스트 코드 작성
        // ...
    }
}

위의 예시 코드에서는 whenNew() 메소드를 사용하여 TargetClass의 생성자 호출을 감시하고, thenReturn() 메소드를 사용하여 가짜(Mock) 객체를 반환하도록 설정하였습니다. 이후 테스트 코드에서는 생성된 모의 객체를 사용하게 됩니다.

verifyNew() 메소드

객체 생성 감시 이후에는 생성된 객체가 예상대로 사용되었는지를 확인할 필요가 있습니다. 이 때에 사용할 수 있는 메소드가 verifyNew() 입니다. 이 메소드는 객체 생성 횟수와 매개변수 등을 확인하여 테스트의 검증을 수행합니다.

@RunWith(PowerMockRunner.class)
@PrepareForTest(TargetClass.class)
public class MyTestClass {
    
    @Test
    public void testMethod() {
        // 객체 생성 감시 설정
        PowerMockito.whenNew(TargetClass.class).withArguments(arg1, arg2).thenReturn(mockTarget);
        
        // 테스트 코드 작성
        
        // 객체 생성 확인
        PowerMockito.verifyNew(TargetClass.class).withArguments(arg1, arg2);
    }
}

위의 예시 코드에서는 whenNew() 메소드를 사용하여 TargetClass의 생성자 호출을 감시하고, thenReturn() 메소드를 사용하여 가짜(Mock) 객체를 반환하도록 설정하였습니다. 이후 테스트 코드에서는 생성된 모의 객체를 사용하며, verifyNew() 메소드를 사용하여 생성이 예상대로 이루어졌는지를 확인합니다.

객체 생성 감시는 테스트 코드의 외부 환경을 통제하여 격리된 테스트를 수행할 수 있는 강력한 도구입니다. PowerMock에서 제공하는 객체 생성 감시 기능을 적절히 사용하여 테스트를 보다 안정적이고 확신할 수 있는 수준으로 개선할 수 있습니다.

참고 문서: PowerMock - Expect New Object