[java] PowerMock에서의 로그인 작업 모킹

29 Nov 2023

java

PowerMock은 Java 프레임워크에서 유용하게 사용되는 모킹 라이브러리입니다. 모킹은 테스트 중에 외부 의존성을 가지는 코드를 테스트하기 위해 가짜 객체를 생성하는 과정을 말합니다. 이번 글에서는 PowerMock을 사용하여 로그인 작업을 모킹하는 방법에 대해 알아보겠습니다.

1. PowerMock 설정하기

PowerMock을 사용하기 위해 먼저 프로젝트의 의존성에 PowerMock 모듈을 추가해야 합니다. Gradle을 사용하는 경우, build.gradle 파일에 다음 의존성을 추가합니다:

testImplementation 'org.powermock:powermock-module-junit4:2.0.9'
testImplementation 'org.powermock:powermock-api-mockito2:2.0.9'

Maven을 사용하는 경우, pom.xml 파일에 다음 의존성을 추가합니다:

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito2</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>

2. 로그인 작업 모킹하기

다음은 로그인 작업을 수행하는 UserService 클래스가 있는 예제입니다:

public class UserService {
    private UserRepository userRepository;

    public UserService(UserRepository userRepository) {
        this.userRepository = userRepository;
    }

    public boolean login(String username, String password) {
        User user = userRepository.findByUsername(username);

        if (user != null && user.getPassword().equals(password)) {
            return true;
        }

        return false;
    }
}

이제 UserService를 테스트하기 위해 UserRepository를 모킹해야 합니다. PowerMock에서는 Mockito와 함께 사용되므로, UserRepository를 Mockito mock 객체로 생성한 후 mock 객체를 알맞은 응답을 반환하도록 설정합니다:

@RunWith(PowerMockRunner.class)
@PrepareForTest(UserRepository.class)
public class UserServiceTest {
    @Test
    public void testLogin() throws Exception {
        // Mock 객체 생성
        UserRepository userRepository = Mockito.mock(UserRepository.class);
        
        // Mock 객체의 응답 설정
        Mockito.when(userRepository.findByUsername(Mockito.anyString())).thenReturn(new User("testuser", "password"));

        // UserService 객체 생성 및 로그인 메소드 호출
        UserService userService = new UserService(userRepository);
        boolean result = userService.login("testuser", "password");

        // 검증
        Assert.assertTrue(result);
    }
}

위의 예제에서 @RunWith(PowerMockRunner.class)와 @PrepareForTest(UserRepository.class) 어노테이션은 PowerMock과 함께 JUnit을 실행하기 위해 필요한 설정입니다. Mockito.mock()을 사용하여 UserRepository의 mock 객체를 생성하고, Mockito.when().thenReturn()을 사용하여 mock 객체의 응답을 설정합니다. 마지막으로 UserService 객체를 생성하고 login 메소드를 호출하여 결과를 검증합니다.

3. 마무리

이제 PowerMock을 사용하여 로그인 작업을 모킹하는 방법을 알아보았습니다. PowerMock은 외부 의존성을 가지는 코드를 테스트하기 위해 강력한 도구입니다. 추가적인 기능과 사용 방법에 대해서는 PowerMock의 공식 문서를 참조하시기 바랍니다.