net.amygdalum.testrecorder.runtime.SelectedFieldsComparisonStrategy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of testrecorder-runtime Show documentation

The newest version!

package net.amygdalum.testrecorder.runtime;

import static java.util.Arrays.asList;
import static net.amygdalum.testrecorder.util.Types.allFields;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class SelectedFieldsComparisonStrategy implements ComparisonStrategy {

	private Set fields;
	private ComparisonStrategy next;
	
	public SelectedFieldsComparisonStrategy(Set fields) {
		this.fields = fields;
		this.next = this;
	}
	
	public SelectedFieldsComparisonStrategy andThen(ComparisonStrategy next) {
		this.next = next;
		return this;
	}	
	
	public static SelectedFieldsComparisonStrategy comparingFields(Collection fields) {
		return new SelectedFieldsComparisonStrategy(new HashSet<>(fields));
	}
	
	public static SelectedFieldsComparisonStrategy comparingFields(String... fields) {
		return comparingFields(asList(fields));
	}

	@Override
	public List extend(GenericComparison comparison) throws ComparisonException {
		List todo = new ArrayList<>();
		Class clazz = comparison.requireSameClass();
		for (Field field : allFields(clazz)) {
			String fieldName = field.getName();
			if (!fields.contains(fieldName)) {
				continue;
			}
			todo.add(comparison.newComparison(fieldName));
		}
		return todo;
	}
	
	@Override
	public ComparisonStrategy next() {
		return next;
	}

}