io.github.sskorol.model.DataSupplierMetaData Maven / Gradle / Ivy

Go to download
package io.github.sskorol.model;

import io.github.sskorol.core.DataSupplier;
import io.github.sskorol.utils.ReflectionUtils;
import one.util.streamex.EntryStream;
import one.util.streamex.IntStreamEx;
import org.testng.ITestContext;
import org.testng.ITestNGMethod;

import java.lang.reflect.Method;
import java.util.List;

import static io.github.sskorol.utils.ReflectionUtils.invokeDataSupplier;
import static io.github.sskorol.utils.ReflectionUtils.streamOf;
import static java.util.Collections.singletonList;
import static java.util.Optional.ofNullable;

/**
 * Base container for DataSupplier meta data.
 */
@SuppressWarnings("FinalLocalVariable")
public class DataSupplierMetaData {

    private final List testData;
    private final boolean transpose;
    private final boolean flatMap;
    private final int[] indices;
    private final TestNGMethod testNGMethod;

    public DataSupplierMetaData(final ITestContext context, final ITestNGMethod testMethod) {
        this.testNGMethod = new TestNGMethod(context, testMethod);
        this.transpose = testNGMethod.getDataSupplierArg(DataSupplier::transpose, false);
        this.flatMap = testNGMethod.getDataSupplierArg(DataSupplier::flatMap, false);
        this.indices = testNGMethod.getDataSupplierArg(DataSupplier::indices, new int[0]);
        this.testData = transform();
    }

    public List getTestData() {
        return testData;
    }

    public ITestNGMethod getTestMethod() {
        return testNGMethod.getTestMethod();
    }

    public Method getDataSupplierMethod() {
        return testNGMethod.getDataSupplierMethod();
    }

    private List transform() {
        var data = streamOf(obtainReturnValue()).toList();
        var indicesList = indicesList(data.size());
        var wrappedReturnValue = EntryStream.of(data).filterKeys(indicesList::contains).values();

        if (transpose) {
            return singletonList(flatMap
                ? wrappedReturnValue.flatMap(ReflectionUtils::streamOf).toArray()
                : wrappedReturnValue.toArray());
        }

        return wrappedReturnValue.map(ob -> flatMap ? streamOf(ob).toArray() : new Object[] {ob}).toList();
    }

    private Object obtainReturnValue() {
        return invokeDataSupplier(testNGMethod.getDataSupplierMetaData());
    }

    private List indicesList(final int collectionSize) {
        return ofNullable(indices)
            .filter(indicesArray -> indicesArray.length > 0)
            .map(IntStreamEx::of)
            .orElseGet(() -> IntStreamEx.range(0, collectionSize))
            .boxed()
            .toList();
    }
}