[java] JUnit 프레임워크 확장하기

14 Nov 2023

java

JUnit은 Java에서 단위 테스트를 위한 가장 널리 사용되는 프레임워크 중 하나입니다. 그러나 때로는 기본적으로 제공되는 JUnit의 기능만으로는 테스트의 특정 요구 사항을 충족시키기 어려울 수 있습니다. 이런 경우에는 JUnit을 확장하여 필요한 기능을 추가할 수 있습니다.

이 글에서는 JUnit을 확장하는 방법에 대해 알아보겠습니다.

1. 확장 가능한 테스트 러너 만들기

JUnit은 테스트 러너(Test Runner)를 통해 테스트 메소드를 실행합니다. 우리는 JUnit의 테스트 러너를 확장하여 테스트 메소드 실행 전후에 추가적인 동작을 수행할 수 있습니다.

public class CustomRunner extends BlockJUnit4ClassRunner {
    
    public CustomRunner(Class<?> klass) throws InitializationError {
        super(klass);
    }
    
    @Override
    protected void beforeInvokeMethod(FrameworkMethod method, RunNotifier notifier) {
        // 메소드 실행 전에 필요한 동작 수행
        System.out.println("Before invoking method: " + method.getName());
    }
    
    @Override
    protected void afterInvokeMethod(FrameworkMethod method, RunNotifier notifier) {
        // 메소드 실행 후에 필요한 동작 수행
        System.out.println("After invoking method: " + method.getName());
    }
}

위 코드에서 CustomRunner는 BlockJUnit4ClassRunner를 확장한 사용자 정의 테스트 러너입니다. beforeInvokeMethod와 afterInvokeMethod 메소드에서는 각각 테스트 메소드 실행 전후에 수행되어야 할 동작을 정의할 수 있습니다.

2. 확장 가능한 어노테이션 만들기

JUnit은 어노테이션을 사용하여 테스트의 실행 순서를 제어할 수 있습니다. 우리는 JUnit의 어노테이션을 확장하여 추가적인 기능을 제공할 수 있습니다.

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface RetryTest {
    
    int maxRetries() default 1;
}

위 코드에서 RetryTest 어노테이션은 테스트 메소드를 재시도할 수 있는 횟수를 정의하는 데 사용됩니다.

3. 확장 가능한 테스트 규칙 만들기

JUnit은 테스트 규칙(Test Rule)을 사용하여 테스트에 부수적으로 적용되는 동작을 제어할 수 있습니다. 우리는 JUnit의 테스트 규칙을 확장하여 추가적인 동작을 수행할 수 있습니다.

public class TimingTestRule implements TestRule {
    
    @Override
    public Statement apply(Statement base, Description description) {
        return new Statement() {
            
            @Override
            public void evaluate() throws Throwable {
                long start = System.currentTimeMillis();
                try {
                    base.evaluate();
                } finally {
                    long end = System.currentTimeMillis();
                    long duration = end - start;
                    System.out.println("Execution time: " + duration + "ms");
                }
            }
        };
    }
}

위 코드에서 TimingTestRule은 테스트 메소드의 실행 시간을 측정하고 출력하는 테스트 규칙입니다. 사용자 정의 테스트 규칙은 apply 메소드를 구현하여 원하는 동작을 수행할 수 있습니다.

결론

JUnit을 확장함으로써 우리는 테스트에 필요한 추가적인 기능을 제공할 수 있습니다. 테스트 러너, 어노테이션, 테스트 규칙을 확장하여 자신의 프로젝트에 적합한 테스트 환경을 구성할 수 있습니다.

참고 문서:

JUnit 확장하기