SmartSelect.java

package org.lightadmin.field;

import org.lightadmin.SeleniumContext;
import org.openqa.selenium.By;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.internal.Locatable;
import org.openqa.selenium.support.FindBy;

public class SmartSelect extends BaseField implements BaseSelect {

	private final WebElement theElement;

	@FindBy( className = "chzn-single" )
	private WebElement dropDownContainer;

	@FindBy( className = "search-choice-close" )
	private WebElement clearButton;

	private WebElement valueList;

	@FindBy( tagName = "input" )
	private WebElement searchField;

	private WebElement link;

	public SmartSelect( WebElement theElement, SeleniumContext seleniumContext ) {
		super( seleniumContext );
		this.theElement = theElement;

		link = theElement.findElement(By.tagName("a"));
		valueList = theElement.findElement(By.className("chzn-results"));
	}

	@Override
	public void select( String... optionNames ) {
		expandOptionList();

		webDriver().waitForElementVisible( valueList );

		final WebElement optionToSelect = valueList.findElement( By.xpath( String.format( "//li[contains(text(),'%s')]", optionNames[ 0 ] ) ) );

		( ( Locatable ) optionToSelect ).getCoordinates().inViewPort();

		optionToSelect.click();

		webDriver().waitForElementVisible( getSelectedOption( optionNames[ 0 ] ) );
	}

	@Override
	public void clear() {
		clearButton.click();
		webDriver().waitForElementVisible( getSelectedOption( "Select " ) );
	}

	@Override
	public void searchAndSelect( String searchString, String labelToSelect ) {
		expandOptionList();
		searchField.sendKeys( searchString );
		selectFromFilteredList( labelToSelect );
	}

	private void selectFromFilteredList( String labelToSelect ) {
		for ( WebElement option : valueList.findElements( By.tagName( "li" ) ) ) {
			if ( option.getText().equals( labelToSelect ) )
				option.click();
		}
	}

	private void expandOptionList() {
		if ( !dropDownContainer.getAttribute( "class" ).contains( "with-drop" ) ) {
			link.click();
		}
	}

	private WebElement getSelectedOption( String textLabel ) {
		return theElement.findElement( By.xpath( String.format( "//a[span[contains(text(),'%s')]]", textLabel ) ) );
	}
}