[java] TestNG를 활용한 데이터 프로바이더와 테스트 데이터 관리

20 Nov 2023

java

TestNG는 Java 기반의 테스트 프레임워크로, 데이터 프로바이더를 통해 테스트 데이터를 관리할 수 있습니다. 이를 통해 테스트 케이스를 더 유연하게 관리할 수 있고, 코드의 재사용성을 높일 수 있습니다.

데이터 프로바이더란?

데이터 프로바이더는 테스트 메서드에 데이터를 제공하는 역할을 합니다. 각 테스트 메서드는 데이터 프로바이더에 의해 호출되어 실행됩니다. 이를 통해 여러 데이터 세트에 대한 반복 테스트를 간편하게 작성할 수 있습니다.

데이터 프로바이더 작성하기

데이터 프로바이더를 작성하기 위해 @DataProvider 어노테이션을 사용합니다. 이 어노테이션은 메서드를 데이터 프로바이더로 지정합니다. 아래의 예시를 확인해보겠습니다.

import org.testng.annotations.DataProvider;

public class MyDataProvider {

    @DataProvider
    public Object[][] testData() {
        return new Object[][] {
            { "data1", 10 },
            { "data2", 20 },
            { "data3", 30 }
        };
    }
}

위의 코드에서 testData() 메서드는 @DataProvider 어노테이션으로 지정되어 있습니다. Object[][] 타입을 반환하는 이 메서드는 각 데이터 세트를 2차원 배열로 반환합니다.

데이터 프로바이더 사용하기

테스트 메서드에서 데이터 프로바이더를 사용하기 위해서는 @Test 어노테이션의 dataProvider 속성을 사용합니다. 데이터 프로바이더 메서드의 이름을 문자열로 전달하여 연결합니다. 아래의 예시를 확인해보겠습니다.

import org.testng.annotations.Test;

public class MyTest {

    @Test(dataProvider = "testData", dataProviderClass = MyDataProvider.class)
    public void testMethod(String data, int value) {
        // 테스트 실행 코드
    }
}

위의 코드에서 testMethod() 메서드는 @Test 어노테이션의 dataProvider 속성을 통해 testData() 메서드와 연결됩니다. 테스트 메서드의 파라미터를 통해 데이터를 전달받을 수 있습니다.

Conclusion

TestNG를 사용하여 데이터 프로바이더를 활용하면 테스트 데이터를 효율적으로 관리할 수 있습니다. 데이터 프로바이더를 이용한 테스트 케이스 작성은 코드의 재사용성을 높이고, 테스트의 유연성을 확보하는데 도움을 줍니다.

더 자세한 내용은 TestNG 공식 문서를 참조하세요.

References

TestNG 공식 문서: https://testng.org/doc/documentation-main.html