io.cucumber.core.stepexpression.ArgumentMatcher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cucumber-core Show documentation

The newest version!

package io.cucumber.core.stepexpression;

import io.cucumber.core.gherkin.DataTableArgument;
import io.cucumber.core.gherkin.DocStringArgument;
import io.cucumber.core.gherkin.Step;

import java.lang.reflect.Type;
import java.util.List;
import java.util.stream.Collectors;

public final class ArgumentMatcher {

    private final StepExpression expression;

    public ArgumentMatcher(StepExpression expression) {
        this.expression = expression;
    }

    public List argumentsFrom(Step step, Type... types) {
        io.cucumber.core.gherkin.Argument arg = step.getArgument();
        if (arg == null) {
            return expression.match(step.getText(), types);
        }

        if (arg instanceof io.cucumber.core.gherkin.DocStringArgument) {
            DocStringArgument docString = (DocStringArgument) arg;
            String content = docString.getContent();
            String contentType = docString.getMediaType();
            return expression.match(step.getText(), content, contentType, types);
        }

        if (arg instanceof io.cucumber.core.gherkin.DataTableArgument) {
            DataTableArgument table = (DataTableArgument) arg;
            List> cells = emptyCellsToNull(table.cells());
            return expression.match(step.getText(), cells, types);
        }

        throw new IllegalStateException("Argument was neither PickleString nor PickleTable");
    }

    private static List> emptyCellsToNull(List> cells) {
        return cells.stream()
                .map(row -> row.stream()
                        .map(s -> s.isEmpty() ? null : s)
                        .collect(Collectors.toList()))
                .collect(Collectors.toList());
    }

}