LoadUiRobot.java

package com.eviware.loadui.ui.fx.util.test;

import com.eviware.loadui.util.test.TestUtils;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;
import javafx.scene.Node;
import javafx.scene.control.Label;
import javafx.stage.Window;
import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.junit.internal.matchers.TypeSafeMatcher;
import org.loadui.testfx.GuiTest;

import java.awt.*;
import java.util.Queue;
import java.util.Set;
import java.util.concurrent.Callable;
import java.util.concurrent.TimeUnit;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.not;
import static com.google.common.collect.Lists.newLinkedList;
import static org.junit.Assert.assertThat;
import static org.loadui.testfx.GuiTest.*;
import static org.loadui.testfx.matchers.ContainsNodesMatcher.contains;
import static org.loadui.testfx.matchers.EnabledMatcher.enabled;
import static org.loadui.testfx.matchers.VisibleNodesMatcher.visible;

public class LoadUiRobot
{
	public enum Component
	{
		FIXED_RATE_GENERATOR( "generators", "Fixed Rate" ), TABLE_LOG( "output", "Table Log" ), WEB_PAGE_RUNNER(
			"runners", "Web Page Runner" ), SCENARIO( "vu-scenario", "VU Scenario" );

		public final String category;
		public final String name;

		private Component( String category, String name )
		{
			this.category = category;
			this.name = name;
		}

		public String cssClass()
		{
			return name.toLowerCase().replace( ' ', '-' );
		}
	}

	private Queue<Point> predefinedPoints;
	private GuiTest controller;

	{
		resetPredefinedPoints();
	}

	private LoadUiRobot( GuiTest controller )
	{
		this.controller = controller;
	}

	public void resetPredefinedPoints()
	{
		predefinedPoints = newLinkedList( ImmutableList.of( new Point( 250, 250 ), new Point(
				450, 450 ) ) );
	}

	public static LoadUiRobot usingController( GuiTest controller )
	{
		return new LoadUiRobot( controller );
	}

	public ComponentHandle createComponent( final Component component )
	{
		if( findAll( ".canvas-object-view" ).isEmpty() )
		{
			resetPredefinedPoints();
		}

		Preconditions.checkNotNull( predefinedPoints.peek(),
				"All predefined points (x,y) for component placement are used. Please add new ones." );
		return createComponentAt( component, predefinedPoints.poll() );
	}

	public ComponentHandle createComponentAt( final Component component, Point targetPoint )
	{
		final int numberOfComponents = GuiTest.findAll( ".canvas-object-view" ).size();
		Set<Node> oldOutputs = findAll( ".canvas-object-view .terminal-view.output-terminal" );
		Set<Node> oldInputs = findAll( ".canvas-object-view .terminal-view.input-terminal" );

		expandCategoryOf( component );

		Window window = find( "#" + component.category + ".category" ).getScene().getWindow();
		int windowX = ( int )window.getX();
		int windowY = ( int )window.getY();
		controller.drag( matcherForIconOf( component ) )
				.to( windowX + targetPoint.x, windowY + targetPoint.y );

		TestUtils.awaitCondition( new Callable<Boolean>()
		{
			@Override
			public Boolean call() throws Exception
			{
				return GuiTest.findAll( ".canvas-object-view" ).size() == numberOfComponents + 1;
			}
		}, 25000 );

		Set<Node> outputs = findAll( ".canvas-object-view .terminal-view.output-terminal" );
		Set<Node> inputs = findAll( ".canvas-object-view .terminal-view.input-terminal" );
		inputs.removeAll( oldInputs );
		outputs.removeAll( oldOutputs );

		return new ComponentHandle( inputs, outputs, controller, this );
	}

	public Matcher<Node> matcherForIconOf( final Component component )
	{
		return new TypeSafeMatcher<Node>()
		{
			@Override
			public boolean matchesSafely( Node node )
			{
				String className = node.getClass().getSimpleName();
				if( className.equals( "ComponentDescriptorView" ) || className.equals( "NewScenarioIcon" ) )
				{
					return node.toString().equals( component.name );
				}
				return false;
			}

			@Override
			public void describeTo( Description description )
			{
				//To change body of implemented methods use File | Settings | File Templates.
			}
		};
	}

	public void expandCategoryOf( Component component )
	{
		if( GuiTest.findAll( "#" + component.category + ".category .expander-button" ).isEmpty() )
		{
			scrollToCategoryOf( component );
		}
		controller.click( "#" + component.category + ".category .expander-button" );
	}

	public void scrollToCategoryOf( Component component )
	{
		int maxToolboxCategories = 50;
		while( GuiTest.findAll( "#" + component.category + ".category" ).isEmpty() )
		{
			if( --maxToolboxCategories < 0 )
				throw new RuntimeException( "Could not find component category " + component.category
						+ " in component ToolBox." );
			controller.move( "#runners.category" ).scroll( 10 );
		}
	}

	public Node getComponentNode( Component component, String... optionalName )
	{
		if( optionalName.length > 0 )
		{
			return findComponentByName( optionalName[0], true );
		}
		else
		{
			return find( "." + component.cssClass() );
		}
	}

	@Deprecated
	private Node findComponentByName( String name, boolean exactMatch )
	{
		for( Node compNode : findAll( ".canvas-object-view" ) )
		{
			Set<Node> textLabels = findAll( "Label", find( "#topBar", compNode ) );
			for( Node label : textLabels )
			{
				String componentLabel = ( ( Label )label ).getText();
				boolean foundMatch = exactMatch ? componentLabel.equals( name ) : componentLabel.startsWith( name );
				if( foundMatch )
				{
					return compNode;
				}
			}
		}
		return null;
	}

	public void clickPlayStopButton()
	{
		Node playButton = find( ".project-playback-panel .play-button" );
		waitUntil( playButton, is( enabled() ) );
		System.out.println("      Clicking playbutton: " + playButton );
		controller.click( playButton );
		System.out.println("      There, I clicked it!");
	}

	public void pointAtPlayStopButton()
	{
		controller.move( ".project-playback-panel .play-button" );
	}

	public void runTestFor( int number, TimeUnit unit )
	{
		System.out.println("      aaa");
		clickPlayStopButton();
		controller.sleep( unit.toMillis( number ) );
		System.out.println("      bbb");
		clickPlayStopButton();
		waitUntil( "#abort-requests", is( not( visible() ) ) );
	}

	public void deleteAllComponentsFromProjectView()
	{
		waitUntil( "#abort-requests", is( not( visible() ) ) );

		controller.click( "#designTab" );

		int maxTries = 20;
		int tries = 0;
		while( tries++ < maxTries && !findAll( ".component-view" ).isEmpty() )
			controller.click( ".component-view #menu" ).click( "#delete-item" ).click( "#default" );

		assertThat( ".component-layer", contains( 0, ".component-view" ) );
		resetPredefinedPoints();
	}
}