[java] PowerMock에서의 네트워크 요청 모킹

29 Nov 2023

java

PowerMock은 자바에서 단위 테스트를 작성할 때 사용되는 강력한 도구입니다. 이는 네트워크 요청과 같은 외부 종속성이 있는 코드를 모킹하고 테스트할 수 있도록 도와줍니다. 네트워크 요청의 모킹은 테스트할 때 실제 네트워크 연결을 필요로하지 않고도 코드를 독립적으로 테스트할 수 있는 장점이 있습니다.

아래는 PowerMock를 사용하여 네트워크 요청을 모킹하는 예제입니다.

@RunWith(PowerMockRunner.class)
@PrepareForTest({MyClass.class, HttpURLConnection.class})
public class MyTest {

    @Test
    public void testNetworkRequest() throws Exception {
        // Mock HttpURLConnection 객체 생성
        HttpURLConnection connectionMock = PowerMockito.mock(HttpURLConnection.class);

        // 모킹된 HttpURLConnection이 반환할 응답 코드 설정
        PowerMockito.when(connectionMock.getResponseCode()).thenReturn(200);

        // URL 객체 생성 및 모킹된 HttpURLConnection 반환 설정
        URL url = new URL("http://example.com");
        PowerMockito.whenNew(URL.class).withArguments(Mockito.anyString()).thenReturn(url);
        PowerMockito.when(url.openConnection()).thenReturn(connectionMock);

        // 테스트할 클래스의 메서드 호출
        MyClass myClass = new MyClass();
        boolean result = myClass.makeNetworkRequest();

        // 테스트 결과 검증
        assertTrue(result);
        PowerMockito.verifyStatic(HttpURLConnection.class); // 정적 메서드 모킹 검증
        HttpURLConnection.setFollowRedirects(Mockito.anyBoolean());
        PowerMockito.verify(connectionMock).connect(); // 호출된 메서드 검증
    }
}

위의 예제에서는 MyClass라는 클래스에서 makeNetworkRequest() 메서드를 테스트하는 방법을 보여줍니다. HttpURLConnection 클래스는 PowerMock에 의해 모킹되고, 테스트에 필요한 응답 코드가 설정되었습니다. URL 객체도 모킹되어 connectionMock 객체를 반환하도록 설정되었습니다. 테스트 결과를 확인하기 위해 assertEquals() 메서드를 사용하거나 다른 검증 메서드를 사용할 수 있습니다.

PowerMock을 사용하여 네트워크 요청을 모킹하는 것은 테스트를 더욱 견고하고 독립적으로 만들어줍니다. 이를 통해 외부 종속성에 의존하지 않고 자체 코드 로직을 테스트할 수 있습니다.