com.github.vendigo.acetest.conversion.DataConverter Maven / Gradle / Ivy

Go to download
package com.github.vendigo.acetest.conversion;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

import static java.util.stream.Collectors.toList;

public class DataConverter {
    public static List> parseDataForAssertions(List> inputData) {
        return inputData.stream()
                .map(parseRow(String::toUpperCase, StringParser::parseString))
                .collect(toList());
    }

    public static List> convertDataForAssertions(List> inputData) {
        return inputData.stream()
                .map(parseRow(String::toUpperCase, ObjectConverter::convertForAssertion))
                .collect(Collectors.toList());
    }

    public static List> parseDataForInsert(List> inputData) {
        return inputData.stream()
                .map(parseRow(String::toUpperCase, StringParser::parseString))
                .map(parseRow(String::toUpperCase, ToStringConverter::convertForInserting))
                .collect(Collectors.toList());
    }

    private static  Function, Map> parseRow(Function keyConvert,
                                                                            Function valueConvert) {
        return (inputRow) -> {
            Map resultMap = new HashMap<>();
            inputRow.forEach((key, value) -> {
                String convertedKey = keyConvert.apply(key);
                U convertedValue = valueConvert.apply(value);
                if (convertedKey != null && convertedValue != null) {
                    resultMap.put(convertedKey, convertedValue);
                }
            });
            return resultMap;
        };
    }
}