de.retest.recheck.testng.Execute Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of recheck-testng-extension Show documentation

TestNG extension reducing boilerplate code in tests.

There is a newer version: 1.13.0

Show newest version

package de.retest.recheck.testng;

import java.lang.reflect.Field;
import java.util.function.Consumer;
import java.util.function.Predicate;
import java.util.stream.Stream;

import de.retest.recheck.RecheckLifecycle;

/**
 * Search fields matching {@link #recheckLifecycle} {@link Predicate} and call the consumer for the field values of the
 * given object.
 */
public class Execute {

	private static final Predicate recheckLifecycle =
			field -> RecheckLifecycle.class.isAssignableFrom( field.getType() );

	private final Consumer consumer;

	private Execute( final Consumer consumer ) {
		this.consumer = consumer;
	}

	/**
	 * Create an executor to call the consumer with all instances of the later given object.
	 *
	 * @param consumer
	 *            {@link Consumer} to be executed for all fields matching {@link #recheckLifecycle} {@link Predicate}
	 */
	public static Execute execute( final Consumer consumer ) {
		return new Execute( consumer );
	}

	/**
	 * Call {@link #consumer} on all field values matching {@link #recheckLifecycle} {@link Predicate}.
	 *
	 * @param object
	 *            object to retrieve fields from and call {@link #consumer} on
	 */
	public void on( final Object object ) {
		final Stream field = findRecheckFields( object );
		field.forEach( f -> execute( f, object ) );
	}

	private void execute( final Field field, final Object testInstance ) {
		unlock( field );
		doExecute( field, testInstance );
		lock( field );
	}

	private void doExecute( final Field field, final Object testInstance ) {
		try {
			consumer.accept( (RecheckLifecycle) field.get( testInstance ) );
		} catch ( IllegalArgumentException | IllegalAccessException cause ) {
			throw new RuntimeException( cause );
		}
	}

	private Stream findRecheckFields( final Object testInstance ) {
		return FindFields.matching( recheckLifecycle ).on( testInstance.getClass() );
	}

	private void unlock( final Field field ) {
		field.setAccessible( true );
	}

	private void lock( final Field field ) {
		field.setAccessible( false );
	}
}