All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.tngtech.junit.dataprovider.DataProviderParameterResolver Maven / Gradle / Ivy

Go to download

A TestNG like dataprovider runner for JUnit Jupiter which is feature comparable to JUnit4 dataprovider.

There is a newer version: 2.10
Show newest version
package com.tngtech.junit.dataprovider;

import static com.tngtech.junit.dataprovider.Preconditions.checkNotNull;

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

import org.junit.jupiter.api.extension.ExtensionContext;
import org.junit.jupiter.api.extension.ParameterContext;
import org.junit.jupiter.api.extension.ParameterResolutionException;
import org.junit.jupiter.api.extension.ParameterResolver;

class DataProviderParameterResolver implements ParameterResolver {

    private final List arguments;

    DataProviderParameterResolver(List arguments) {
        this.arguments = new ArrayList<>(checkNotNull(arguments, "'arguments' must not be null"));
    }

    @Override
    public boolean supportsParameter(ParameterContext parameterContext, ExtensionContext extensionContext) {
        Executable declaringExecutable = parameterContext.getParameter().getDeclaringExecutable();
        Method testMethod = extensionContext.getTestMethod().orElse(null);
        return declaringExecutable.equals(testMethod) && parameterContext.getIndex() < arguments.size();
    }

    @Override
    public Object resolveParameter(ParameterContext parameterContext, ExtensionContext extensionContext)
            throws ParameterResolutionException {
        Object result = arguments.get(parameterContext.getIndex());

        // TODO workaround for https://github.com/junit-team/junit5/issues/1092
        Class parameterType = parameterContext.getParameter().getType();
        if (parameterType.isPrimitive()) {
            return convertToBoxedTypeAsWorkaroundForNotWorkingWideningAndUnboxingConversion(result, parameterType);
        }

        return result;
    }

    private Object convertToBoxedTypeAsWorkaroundForNotWorkingWideningAndUnboxingConversion(Object result,
            Class parameterType) {
        if (short.class.equals(parameterType)) {
            return ((Number) result).shortValue();
        } else if (byte.class.equals(parameterType)) {
            return ((Number) result).byteValue();
        } else if (int.class.equals(parameterType)) {
            return ((Number) result).intValue();
        } else if (long.class.equals(parameterType)) {
            return ((Number) result).longValue();
        } else if (float.class.equals(parameterType)) {
            return ((Number) result).floatValue();
        } else if (double.class.equals(parameterType)) {
            return ((Number) result).doubleValue();
        }
        return result;
    }
}