MenuTreePageFragment.java

/**
 * <a href="http://www.openolat.org">
 * OpenOLAT - Online Learning and Training</a><br>
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License"); <br>
 * you may not use this file except in compliance with the License.<br>
 * You may obtain a copy of the License at the
 * <a href="http://www.apache.org/licenses/LICENSE-2.0">Apache homepage</a>
 * <p>
 * Unless required by applicable law or agreed to in writing,<br>
 * software distributed under the License is distributed on an "AS IS" BASIS, <br>
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. <br>
 * See the License for the specific language governing permissions and <br>
 * limitations under the License.
 * <p>
 * Initial code contributed and copyrighted by<br>
 * frentix GmbH, http://www.frentix.com
 * <p>
 */
package org.olat.selenium.page.core;

import java.util.List;

import org.junit.Assert;
import org.olat.selenium.page.graphene.OOGraphene;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;

/**
 * Fragment which contains the menu tree. The WebElement to create
 * this fragment must be a parent of the div.o_tree
 * 
 * Initial date: 20.06.2014<br>
 * @author srosse, stephane.rosse@frentix.com, http://www.frentix.com
 *
 */
public class MenuTreePageFragment {
	
	public static final By treeBy = By.className("o_tree");
	
	private final  WebDriver browser;

	public MenuTreePageFragment(WebDriver browser) {
		this.browser = browser;
	}
	
	/**
	 * Click the root link in the tree.
	 * 
	 * @return The menu page fragment
	 */
	public MenuTreePageFragment selectRoot() {
		WebElement tree = browser.findElement(treeBy);
		List<WebElement> rootLinks = tree.findElements(By.cssSelector("span.o_tree_link>a"));
		Assert.assertNotNull(rootLinks);
		Assert.assertFalse(rootLinks.isEmpty());
		
		rootLinks.get(0).click();
		OOGraphene.waitBusy(browser);
		return this;
	}
	
	public MenuTreePageFragment selectWithTitle(String title) {
		boolean found = false;
		WebElement tree = browser.findElement(treeBy);
		List<WebElement> nodeLinks = tree.findElements(By.cssSelector("li>div>span.o_tree_link>a"));
		for(WebElement nodeLink:nodeLinks) {
			String text = nodeLink.getText().toLowerCase();
			if(text.contains(title.toLowerCase())) {
				nodeLink.click();
				OOGraphene.waitBusy(browser);
				found = true;
			}
		}
		
		Assert.assertTrue("Link not found with title: " + title, found);
		return this;
	}
	

	public MenuTreePageFragment assertWithTitle(String title) {
		boolean found = false;
		By titleBy = By.cssSelector(".o_tree li>div>span.o_tree_link>a");
		List<WebElement> nodeLinks = browser.findElements(titleBy);
		for(WebElement nodeLink:nodeLinks) {
			String text = nodeLink.getText();
			if(text.contains(title)) {
				found = true;
			}
		}
		
		Assert.assertTrue("Link not found with title: " + title, found);
		return this;
	}

	public MenuTreePageFragment assertTitleNotExists(String title) {
		boolean found = false;
		WebElement tree = browser.findElement(treeBy);
		List<WebElement> nodeLinks = tree.findElements(By.cssSelector("li>div>span.o_tree_link>a"));
		for(WebElement nodeLink:nodeLinks) {
			String text = nodeLink.getText();
			if(text.contains(title)) {
				OOGraphene.waitBusy(browser);
				found = true;
			}
		}
		
		Assert.assertFalse("Link found with title: " + title, found);
		return this;
	}
}