io.github.xinyangpan.cucumber.element.BaseElement Maven / Gradle / Ivy

Go to download
package io.github.xinyangpan.cucumber.element;

import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Objects;
import java.util.Optional;
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Collectors;

import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanWrapperImpl;
import org.springframework.util.Assert;

import com.google.common.base.Preconditions;
import com.google.common.base.Splitter;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;

import io.github.xinyangpan.cucumber.keyword.YesOrNo;
import io.github.xinyangpan.cucumber.util.ElementUtils;

public class BaseElement {
	public static final String IGNORE_ROW = "_ignoreRow";
	public static final String EXPECT_FAIL = "_expectFail";
	// 
	protected final Map keyValueMap;

	public BaseElement(Map keyValueMap) {
		this.keyValueMap = Maps.newHashMap();
		for (Entry e : keyValueMap.entrySet()) {
			if (StringUtils.isNotEmpty(e.getValue())) {
				this.keyValueMap.put(e.getKey(), e.getValue());
			}
		}
	}

	public  Optional getValueOpt(String key, Class type) {
		if (keyValueMap.containsKey(key)) {
			return Optional.of(this.getValue(key, type));
		} else {
			return Optional.empty();
		}
	}

	public  T getValue(String key, Class type) {
		String value = keyValueMap.get(key);
		Assert.notNull(value, String.format("Not value found for key[%s]", key));
		return ElementUtils.defaultConversionService().convert(value, type);
	}

	public  T getValue(String key, T defaultValue) {
		Preconditions.checkNotNull(defaultValue);
		String value = keyValueMap.get(key);
		if (value == null) {
			return defaultValue;
		}
		// 
		@SuppressWarnings("unchecked")
		T t = this.getValue(key, (Class) defaultValue.getClass());
		if (t == null) {
			return defaultValue;
		} else {
			return t;
		}
	}

	public void putValuesTo(Object target) {
		BeanWrapperImpl beanWrapperImpl = ElementUtils.newBeanWrapperImpl(target);
		for (Entry e : this.getEligibleEntries()) {
			initNullObjectIfNeeded(beanWrapperImpl, e.getKey());
			beanWrapperImpl.setPropertyValue(e.getKey(), e.getValue());
		}
	}

	private void initNullObjectIfNeeded(BeanWrapperImpl beanWrapperImpl, String fieldString) {
		List strings = Splitter.on('.').splitToList(fieldString);
		String propertyName = "";
		for (int i = 1; i < strings.size(); i++) {
			propertyName += strings.get(i - 1);
			Object propertyValue = beanWrapperImpl.getPropertyValue(propertyName);
			if (propertyValue == null) {
				Class propertyType = beanWrapperImpl.getPropertyType(propertyName);
				Object instantiate = ElementUtils.newInstance(propertyType);
				beanWrapperImpl.setPropertyValue(propertyName, instantiate);
			}
			propertyName += ".";
		}
	}

	// -----------------------------
	// ----- built-in
	// -----------------------------

	public boolean isIgnoreRow() {
		YesOrNo yesOrNo = this.getValue(IGNORE_ROW, YesOrNo.NO);
		return yesOrNo.isYes();
	}

	public boolean isNotIgnoreRow() {
		return !this.isIgnoreRow();
	}

	public Map dataMap() {
		return keyValueMap;
	}

	/**
	 * To Evaluate if self values matches Target Obejct's fields' values according to keys
	 * 
	 * Field Match Logic
	 * 
	 * For Comparable - using compare
	 * For None Comparable - using equals
	
	 * @param target
	 * @return
	 */
	@SuppressWarnings("unchecked")
	public boolean matches(Object target) {
		// 
		BeanWrapperImpl beanWrapperImpl = ElementUtils.newBeanWrapperImpl(target);
		for (Entry e : this.getEligibleEntries()) {
			Object actualValue = beanWrapperImpl.getPropertyValue(e.getKey());
			Object expectedValue = ElementUtils.defaultConversionService().convert(e.getValue(), beanWrapperImpl.getPropertyType(e.getKey()));
			// 
			if (actualValue instanceof Comparable && expectedValue instanceof Comparable) {
				if (((Comparable