com.nordstrom.automation.selenium.examples.TableComponent Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of selenium-foundation Show documentation

Selenium Foundation is an automation framework designed to extend and enhance the capabilities provided by Selenium (WebDriver).

There is a newer version: 28.3.1-s4

Show newest version

package com.nordstrom.automation.selenium.examples;

import java.util.ArrayList;
import java.util.List;

import org.openqa.selenium.By;
import org.openqa.selenium.SearchContext;
import org.openqa.selenium.WebElement;

import com.nordstrom.automation.selenium.model.ComponentContainer;
import com.nordstrom.automation.selenium.model.PageComponent;
import com.nordstrom.automation.selenium.model.RobustWebElement;

public class TableComponent extends PageComponent {

    public TableComponent(By locator, ComponentContainer parent) {
        super(locator, parent);
    }
    
    public TableComponent(RobustWebElement element, ComponentContainer parent) {
        super(element, parent);
    }
    
    private TableRowComponent tableHdr;
    private List tableRows;
    private int refreshCount;
    
    protected enum Using implements ByEnum {
        HDR_ROW(By.cssSelector("tr[id*='-h']")),
        TBL_ROW(By.cssSelector("tr[id*='-r']"));
        
        private final By locator;
        
        Using(By locator) {
            this.locator = locator;
        }

        @Override
        public By locator() {
            return locator;
        }
    }
    
    public List getHeadings() {
        return getTableHdr().getContent();
    }
    
    public List> getContent() {
        List> result = new ArrayList<>();
        for (TableRowComponent row : getTableRows()) {
            result.add(row.getContent());
        }
        return result;
    }
    
    private TableRowComponent getTableHdr() {
        if (tableHdr == null) {
            tableHdr = new TableRowComponent(Using.HDR_ROW.locator, this);
        }
        return tableHdr;
    }
    
    private List getTableRows() {
        if (tableRows == null) {
            tableRows = newComponentList(TableRowComponent.class, Using.TBL_ROW.locator);
        }
        return tableRows;
    }
    
    public static Object getKey(SearchContext context) {
        return ((WebElement) context).getAttribute("id");
    }

    @Override
    public SearchContext refreshContext(long expiration) {
        refreshCount++;
        return super.refreshContext(expiration);
    }
    
    public int getRefreshCount() {
        return refreshCount;
    }
    
    public int getHeadRefreshCount() {
        return getTableHdr().getRefreshCount();
    }
    
    public int[] getBodyRefreshCounts() {
        List tableRows = getTableRows();
        int[] counts = new int[tableRows.size()];
        for (int i = 0; i < tableRows.size(); i++) {
            counts[i] = tableRows.get(i).getRefreshCount();
        }
        return counts;
    }
    
}