Что такое DataProvider в TestNG?

Что такое DataProvider в TestNG?

DataProvider в TestNG — это мощный механизм, позволяющий передавать параметризованные данные в тестовые методы. Основная цель — выполнить один и тот же тестовый метод многократно с разными наборами входных данных, что является краеугольным камнем параметризованного тестирования. В отличие от простой параметризации через testng.xml (которая статична), DataProvider предоставляет динамический и гибкий способ подачи данных, часто генерируемых или загружаемых непосредственно во время выполнения.

Ключевые особенности и назначение

• Повторное использование кода теста: Один тестовый метод проверяет множество сценариев, что уменьшает дублирование кода.
• Отделение данных от логики: Данные хранятся отдельно от тестового метода (в методе, помеченном @DataProvider), что улучшает читаемость и сопровождаемость.
• Гибкие источники данных: Данные могут быть получены откуда угодно: из массива, коллекции, внешнего файла (Excel, CSV, JSON), базы данных или сгенерированы алгоритмически.
• Разные данные для разных потоков: При параллельном выполнении можно настроить DataProvider для предоставления уникальных данных каждому потоку.

Базовая архитектура и синтаксис

DataProvider состоит из двух основных частей:

1. Метод, снабжающий данными (@DataProvider): Этот метод возвращает данные в виде двумерного массива (Object[][]) или итератора (Iterator<Object[]>). Каждое "вложенное" подмассива (Object[]) представляет собой набор аргументов для одного вызова тестового метода.
2. Тестовый метод (@Test): Метод, который использует эти данные, принимая аргументы в своих параметрах. Связь устанавливается через атрибут dataProvider в аннотации @Test.

Пример базового использования

Рассмотрим простейший пример проверки функции сложения.

import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import static org.testng.Assert.assertEquals;

public class CalculatorTest {

    // 1. Метод DataProvider. Должен возвращать Object[][] или Iterator<Object[]>
    @DataProvider(name = "sumData")
    public Object[][] provideSumData() {
        return new Object[][] {
            { 1, 2, 3 },   // Первый набор: a=1, b=2, expectedResult=3
            { -5, 10, 5 }, // Второй набор
            { 0, 0, 0 }    // Третий набор
        };
    }

    // 2. Тестовый метод, использующий DataProvider.
    // Количество параметров должно соответствовать количеству элементов во внутренних массивах DataProvider.
    @Test(dataProvider = "sumData")
    public void testAddition(int a, int b, int expectedResult) {
        Calculator calc = new Calculator();
        int actualResult = calc.add(a, b);
        assertEquals(actualResult, expectedResult, 
                     String.format("Сложение %d + %d работает некорректно", a, b));
    }
}

В этом примере метод testAddition будет выполнен три раза:

• Первый запуск: a=1, b=2, expectedResult=3
• Второй запуск: a=-5, b=10, expectedResult=5
• Третий запуск: a=0, b=0, expectedResult=0

Расширенные возможности DataProvider

Использование name и parallel

Атрибуты аннотации @DataProvider позволяют тонко настраивать его поведение.

@DataProvider(name = "userData", parallel = true) // Параллельное выполнение итераций
public Object[][] provideUserData() {
    return DataLoader.loadUsersFromCsv("users.csv"); // Загрузка данных из внешнего источника
}

@Test(dataProvider = "userData")
public void testUserLogin(String username, String password) {
    // Тест логина для каждого пользователя из CSV
}

Передача метаданных (ITestContext или Method)

DataProvider-метод может принимать параметры для более интеллектуальной выдачи данных.

import org.testng.ITestContext;
import org.testng.annotations.DataProvider;

@DataProvider(name = "contextAwareData")
public Object[][] provideDataBasedOnEnvironment(ITestContext context) {
    // Можем менять данные в зависимости от имени теста, группы или других атрибутов контекста
    String env = context.getCurrentXmlTest().getParameter("environment");
    if ("staging".equals(env)) {
        return new Object[][] { {"staging_user", "staging_pass"} };
    } else {
        return new Object[][] { {"prod_user", "prod_pass"} };
    }
}

Возврат Iterator<Object[]>

Полезно для работы с большими объемами данных, чтобы не загружать их все в память сразу.

@DataProvider
public Iterator<Object[]> largeDataProvider() {
    List<Object[]> data = new ArrayList<>();
    // Постепенное чтение данных из файла или БД
    try (BufferedReader br = new BufferedReader(new FileReader("huge_data.csv"))) {
        String line;
        while ((line = br.readLine()) != null) {
            String[] parts = line.split(",");
            data.add(new Object[]{parts[0], parts[1]});
        }
    }
    return data.iterator();
}

Важные аспекты для QA Automation

• Индексация в отчетах: Каждый вызов теста с разными данными в отчете TestNG будет отображен отдельно с своим индексом (например, testAddition[0], testAddition[1]), что упрощает анализ падений.
• Наследование: DataProvider может быть объявлен в базовом тестовом классе и использован в классах-наследниках.
• Ограничения: DataProvider-метод должен быть public и не может быть static, если он не находится в отдельном классе. Для статического метода в другом классе используется dataProviderClass в аннотации @Test.

В заключение, DataProvider — это не просто инструмент для подачи данных, это фундаментальный паттерн в TestNG для построения масштабируемых, легко поддерживаемых и комплексных наборов автоматизированных тестов. Его правильное использование позволяет эффективно покрывать граничные случаи и валидировать функциональность приложения с разнообразными входными данными, что напрямую влияет на повышение качества продукта.