com.automationrockstars.gir.ui.ListOfUiObjects Maven / Gradle / Ivy

Go to download
package com.automationrockstars.gir.ui;

import java.util.Iterator;
import java.util.List;

import org.openqa.selenium.NoSuchElementException;
import org.openqa.selenium.WebElement;

import com.automationrockstars.design.gir.webdriver.UiObject;
import com.google.common.base.Optional;
import com.google.common.base.Predicate;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.Iterators;
public class ListOfUiObjects implements Iterable{

	private FluentIterable internal;
	
	@Override
	public Iterator iterator() {
		return Iterators.unmodifiableIterator(internal.iterator());
	}
	
	private ListOfUiObjects(List internal){
		this.internal = FluentIterable.from(internal);
	}
	
	public static ListOfUiObjects from(List of){
		return new ListOfUiObjects(of);
	}
	
	public UiObject elementWithText(String text){
		return firstMatching(UiParts.withText(text));
	}
	
	public UiObject first(){
		return internal.first().get();
	}
	
	public UiObject firstMatching(Predicate filter){
		Optional result = internal.firstMatch(filter);
		if (result.isPresent()){
			return result.get();
		} else {
			throw new NoSuchElementException(String.format("Element identified by %s with filter %s not found",internal.first().get().getLocator(),filter));
		}
	}
	
	public ListOfUiObjects filter(Predicate filter){
		return from(internal.filter(filter).toList());
	}

	public FluentIterable fluentIterable(){
		return internal;
	}
	
	public List list(){
		return internal.toList();
	}
}