Spring Boot 단위 테스트: @Mock 과 @InjectionMocks의 원리와 활용

🧐 1. Mockito란? (왜 사용하는가?)

Mockito 란 Java에서 단위 테스트를 위한 Mocking 프레임워크입니다. 특정 객체를 실제 객체처럼 흉내 내는 가짜(Mock) 객체로 만들어서, 테스트 중 특정 동작을 정의하고 예상 결과를 검증할 수 있다.

✅ 왜 Mocking이 필요할까?

• 데이터베이스 의존성 제거: Repository나 외부 API를 직접 호출하면 테스트가 느려지고 환경에 의존적이 된다.
• 정확한 테스트 보장: 특정 메서드가 정확한 값을 반환하는지 독립적으로 테스트가 가능하다.
• 테스트 속도 향상: 실제 DB 연결 없이 원하는 값을 변환하도록 설정하면 빠른 테스트가 가능하다.

🔥 Mock을 사용해야 하는 경우 vs 사용하지 않아야 하는 경우

Mock을 사용해야 하는 경우	Mock을 사용하지 않아야 하는 경우
데이터베이스 없이 독립적인 테스트가 필요할 때	실제 데이터의 흐름을 검증해야 할 때
외부 API 호출 없이 빠르게 테스트하고 싶을 때	통합 테스트나 E2E 테스트를 진행할 때
특정 메서드의 동작을 단독으로 검증하고 싶을 때	트랜잭션이 올바르게 처리되는지 확인할 때

 

즉, 단위 테스트에서는 Mock을 적극적으로 사용하고,

통합 테스트(E2E 테스트)에서는 실제 DB와 API를 활용하는 것이 더 적절하다.

🏗 2. @Mock과 @InjectMocks의 원리

✅ @Mock

• Mockito에서 특정 객체를 Mocking할 때 사용한다.
• 원래 객체를 생성하는 대신, 해당 객체의 가짜(Stub) 객체를 생성한다.
• when().thenReturn()으로 원하는 동작을 미리 정의 가능하다.

✅ @InjectMocks

• @Mock으로 만들어진 객체를 자동으로 주입(Inject)하여,
  우리가 테스트하려는 @InjectMocks 대상 클래스에서 사용할 수 있도록 한다.

📌 예제 코드

@Mock
private SolutionRepository solutionRepository;  // Mock 객체 생성

@InjectMocks
private SolutionService solutionService;  // solutionRepository가 자동 주입됨

 

위 코드에서 solutionRepository는 Mock 객체이지만, solutionService에서는 원래 객체처럼 사용 가능하다.

 

🔍 3. Mockito가 실제로 어떻게 동작하는가?

Mockito는 프록시(Proxy) 객체를 생성해서 원래 객체처럼 동작하게 만들어.

1. @Mock을 사용하면, 실제 객체가 아니라 Proxy 객체가 생성된다.
2. when().thenReturn()을 사용하면 특정 메서드가 호출될 때 미리 정의된 값을 반환하도록 설정된다.
3. @InjectMocks가 사용된 클래스에는 @Mock으로 만든 가짜 객체가 주입된다.
4. 실제 메서드 호출 시, Mock 객체는 원래 로직을 실행하지 않고 Mockito가 미리 설정한 값을 반환한다.

🛠 4. 코드 예제: Repository를 Mocking한 단위 테스트

@Test
void givenMoreThan100Solutions_whenCheckingThreshold_thenReturnsTrue() {
    // given
    String problemId = "1000";
    String language = "Java";

    // Mock 객체의 동작을 설정 (실제 DB에 접근하지 않음)
    when(solutionRepository.countByProblemIdAndLanguage(problemId, language))
            .thenReturn(150L); // 150개로 설정

    // when
    boolean result = solutionService.hasAtLeast100Solutions(problemId, language);

    // then
    assertTrue(result);
}

✅ 테스트 코드 실행 흐름

1. solutionRepository.countByProblemIdAndLanguage(problemId, language) 호출된다.
2. 실제 데이터베이스를 조회하지 않고, when().thenReturn(150L)이 설정한 값을 반환한다.
3. solutionService.hasAtLeast100Solutions()에서 count >= 100을 체크하여 true 반환한다.
4. assertTrue(result);에서 검증 성공! ✅

🕵️‍♂️ 5. Mockito 내부 동작을 디버깅해보기

테스트 중에 Mockito의 Mock 객체가 어떻게 동작하는지 확인하려면,
System.out.println()을 추가하거나 디버거를 활용할 수 있다.

@Test
void debugMockitoBehavior() {
    // given
    String problemId = "2000";
    String language = "Python";

    when(solutionRepository.countByProblemIdAndLanguage(problemId, language))
            .thenReturn(80L); // 80개로 설정

    System.out.println("Mock 객체 호출 전");  // 디버깅용 로그

    // when
    boolean result = solutionService.hasAtLeast100Solutions(problemId, language);

    System.out.println("Mock 객체가 반환한 값: " 
        + solutionRepository.countByProblemIdAndLanguage(problemId, language));

    // then
    assertFalse(result);
}

 

✅ 출력 결과

Mock 객체 호출 전
Mock 객체가 반환한 값: 80

• solutionRepository.countByProblemIdAndLanguage()가 실제 DB를 조회한 게 아니라,
  Mockito가 설정한 값(80)을 반환한 걸 확인할 수 있다.

✨ 결론

✅ @Mock은 가짜 객체를 만들어서 테스트 시 의존성을 제거하고 독립적인 테스트가 가능하게 함.
✅ @InjectMocks를 사용하면, Mock 객체를 자동으로 주입하여 원래 서비스 클래스처럼 사용 가능.
✅ when().thenReturn()을 활용하면 특정 메서드 호출 시 원하는 값을 반환하도록 설정할 수 있음.
✅ Mockito는 프록시 객체를 생성하여 실제 객체처럼 동작하게 하는 원리로 작동함.