TableLayoutHelper.java

package br.com.caelum.seleniumdsl.table.layout;

import java.util.logging.Logger;

import br.com.caelum.seleniumdsl.table.Table;

import com.thoughtworks.selenium.Selenium;
import com.thoughtworks.selenium.SeleniumException;

class TableLayoutHelper {
	private static final Logger log = Logger.getLogger(TableLayoutHelper.class.getName());

	private final Selenium selenium;
	private final String id;
	private final String type;

	TableLayoutHelper(Selenium selenium, String id, String type) {
		this.selenium = selenium;
		this.id = id;
		this.type = type;
	}

	int getRowCount() {
		return countXPath("/*/tr");
	}

	int countXPath(String expr) {
		return selenium.getXpathCount("//table[@" + type + "='" + id + "']" + expr)
				.intValue();
	}

	String getXPathText(String expr) {
		try {
			return selenium.getText("xpath=//table[@" + type + "='" + id + "']" + expr);
		} catch (SeleniumException e) {
			log.info(e.getMessage());
		}

		return null;
	}

	boolean contains(Table table, String col, String content) {
		for (int i = 1; i < table.getRowCount(); i++) {
			if (table.cell(i, col)
					.contains(content)) {
				return true;
			}
		}
		return false;
	}
}