DefaultTable.java

package br.com.caelum.seleniumdsl.table;

import br.com.caelum.seleniumdsl.search.RowMatcher;
import br.com.caelum.seleniumdsl.search.RowVisitor;
import br.com.caelum.seleniumdsl.search.TableCriteria;
import br.com.caelum.seleniumdsl.table.layout.TableLayout;
import br.com.caelum.seleniumdsl.table.layout.TableLayoutChooser;

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

public class DefaultTable implements Table {
	private final Selenium selenium;

	private final String id;

	/**
	 * The property that holds this table's key, usually "id"
	 */
	private final String type;

	private final TableLayout layout;

	private Integer colCount;

	private Integer rowCount;

	public DefaultTable(Selenium selenium, String id) {
		this(selenium, id, "id");
	}

	public DefaultTable(Selenium selenium, String value, String type) {
		this.selenium = selenium;
		this.id = value;
		this.type = type;

		layout = new TableLayoutChooser(selenium, value, type).choose();
	}

	public String getType() {
		return type;
	}

	public Column column(int columnIndex) {
		return new DefaultColumn(this, columnIndex);
	}

	public Column column(String columnName) {
		return column(findColumn(columnName));
	}

	public int getColCount() {
		if (this.colCount == null) {
			this.colCount = layout.getColCount();
		}
		return this.colCount;
	}

	public int getRowCount() {
		if (this.rowCount == null) {
			this.rowCount = layout.getRowCount();
		}
		return this.rowCount;
	}

	public int getContentCount() {
		return layout.getContentCount();
	}

	public Cell cell(int row, int col) {
		return new DefaultCell(selenium, this, row, col);
	}

	public Cell cell(int row, String col) {
		return new DefaultCell(selenium, this, row, findColumn(col));
	}

	public String getId() {
		return id;
	}

	public boolean exists() {
		return selenium.isElementPresent(getId());
	}

	public void iterate(RowVisitor visitor) {
		int count = getRowCount();
		for (int row = 1; row <= count; row++) {
			visitor.visit(new DefaultRow(this, selenium, row));
		}
	}

	public Row header() {
		return row(1);
	}

	public Row row(Integer row) {
		return new DefaultRow(this, selenium, row);
	}

	public Integer findColumn(String columnName) {
		Row row = new DefaultRow(this, selenium, 1);
		int colCount = getColCount();
		for (int i = 0; i < colCount; i++) {
			String current;
			try {
				current = row.cell(i + 1).headerValue();
			} catch (SeleniumException e) {
				current = row.cell(i + 1).value();
			}
			if (columnName.equals(current))
				return i + 1;
		}
		throw new IllegalArgumentException("Column " + columnName + " not found");
	}

	public RowMatcher select(RowMatcher matcher) {
		matcher.setTable(this);
		return matcher;
	}

	public TableCriteria createCriteria() {
		return new TableCriteria(this);
	}

	public TableLayout getLayout() {
		return layout;
	}
}