com.sdl.selenium.extjs6.grid.Row Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Testy Show documentation
Automated Acceptance Testing. Selenium and Selenium WebDriver test framework for web applications. (optimized for dynamic html, ExtJS, Bootstrap, complex UI, simple web applications/sites)
The newest version!
package com.sdl.selenium.extjs6.grid;

import com.google.common.base.Strings;
import com.sdl.selenium.WebLocatorUtils;
import com.sdl.selenium.utils.config.WebDriverConfig;
import com.sdl.selenium.web.SearchType;
import com.sdl.selenium.web.WebLocator;
import com.sdl.selenium.web.table.AbstractCell;
import com.sdl.selenium.web.utils.RetryUtils;
import com.sdl.selenium.web.utils.Utils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.time.Duration;
import java.util.*;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Stream;

public class Row extends com.sdl.selenium.web.table.Row {
    private static final Logger log = LogManager.getLogger(Row.class);
    private String version;

    public Row() {
        super();
        setTag("table");
    }

    public Row(WebLocator grid) {
        this();
        setContainer(grid);
    }

    public Row(WebLocator grid, int indexRow) {
        this(grid);
        setPosition(indexRow);
    }

    public Row(WebLocator grid, String searchElement, SearchType... searchTypes) {
        this(grid);
        setText(searchElement, searchTypes);
        if (isGridLocked()) {
            String index = getAttribute("data-recordindex");
            setTag("*");
            String xPath = getXPath().replace(grid.getXPath(), "");
            setElPath(xPath + "//table[@data-recordindex='" + index + "']");
        }
    }

    public Row clone(Row row) {
        return new Row(row.getPathBuilder().getContainer(), row.getPathBuilder().getCells());
    }

    private boolean isGridLocked() {
        Grid grid = getGridAsContainer();
        if (grid == null) {
            return false;
        }
        String aClass;
        try {
            aClass = WebDriverConfig.getDriver() == null ? null : grid.getAttributeClass();
        } catch (NullPointerException e) {
            grid = (Grid) getPathBuilder().getContainer().getPathBuilder().getContainer();
            aClass = WebDriverConfig.getDriver() == null ? null : grid.getAttributeClass();
        }
        return aClass != null && aClass.contains("x-grid-locked");
    }

    private Grid getGridAsContainer() {
        Grid grid;
        if (getPathBuilder().getContainer() instanceof Grid) {
            grid = (Grid) getPathBuilder().getContainer();
        } else if (getPathBuilder().getContainer().getPathBuilder().getContainer() instanceof Grid) {
            grid = (Grid) getPathBuilder().getContainer().getPathBuilder().getContainer();
        } else if (getPathBuilder().getContainer().getPathBuilder().getContainer() != null && getPathBuilder().getContainer().getPathBuilder().getContainer().getPathBuilder().getContainer() instanceof Grid) {
            grid = (Grid) getPathBuilder().getContainer().getPathBuilder().getContainer().getPathBuilder().getContainer();
        } else {
            return null;
        }
        return grid;
    }

    public Row(WebLocator grid, AbstractCell... cells) {
        this(grid, false, cells);
    }

    public Row(Row row, boolean clone) {
        this(row.getPathBuilder().getContainer(), row.getPathBuilder().getCells());
    }

    public Row(WebLocator grid, boolean size, AbstractCell... cells) {
        this(grid);
        AbstractCell[] childNodes = Stream.of(cells).filter(t -> t != null && (t.getPathBuilder().getText() != null || (t.getPathBuilder().getChildNodes() != null && !t.getPathBuilder().getChildNodes().isEmpty()))).toArray(AbstractCell[]::new);
        if (isGridLocked()) {
            Integer index = null;
            if (!size) {
                List> ids = new ArrayList<>();
                for (AbstractCell cell : childNodes) {
                    ((Grid) grid).scrollTop();
                    Details details = getCellPosition(cell);
                    int indexCurrent = details.lockedPosition();
                    cell.setTemplateValue("tagAndPosition", indexCurrent + "");
                    Row tmpEl;
                    if (details.firstColumns() >= details.actualPosition()) {
                        WebLocator containerLocked = new WebLocator(grid).setClasses("x-grid-scrollbar-clipper", "x-grid-scrollbar-clipper-locked");
                        tmpEl = new Row(containerLocked).setChildNodes(cell);
                    } else {
                        WebLocator containerUnLocked = new WebLocator(grid).setClasses("x-grid-scrollbar-clipper").setExcludeClasses("x-grid-scrollbar-clipper-locked");
                        tmpEl = new Row(containerUnLocked).setChildNodes(cell);
                    }
                    boolean isScrollBottom;
                    Set list = new LinkedHashSet<>();
                    do {
                        int count = tmpEl.size();
                        for (int j = 1; j <= count; j++) {
                            tmpEl.setResultIdx(j);
                            String indexValue = tmpEl.getAttribute("data-recordindex");
                            int i1 = Integer.parseInt(indexValue);
                            list.add(i1);
                        }
                        isScrollBottom = ((Grid) grid).isScrollBottom();
                        if (!isScrollBottom) {
                            ((Grid) grid).scrollPageDown();
                            ((Grid) grid).scrollPageDown();
                            ((Grid) grid).scrollPageDown();
                            ((Grid) grid).scrollPageDown();
                            ((Grid) grid).scrollPageDown();
                        }
                        tmpEl.setResultIdx(0);
                    } while (!isScrollBottom);
                    if (list.isEmpty()) {
                        log.info("No found child with value: '{}'", cell.getPathBuilder().getText());
                    }
                    ids.add(list);
                }
                ((Grid) grid).scrollTop();
                List> collect = new ArrayList<>();
                for (Set id : ids) {
                    List strings = new ArrayList<>(id);
                    collect.add(strings);
                }
                List theMinList = getTheMinList(collect);
                List commonId = findCommonId(collect, theMinList);
                if (commonId.size() == 1) {
                    index = commonId.get(0);
                } else if (commonId.isEmpty()) {
                    log.error("No found commonId!!!");
                } else {
                    log.error("Find more row that one!!!");
                }
            }
            for (AbstractCell cell : childNodes) {
                if (size) {
                    int indexCurrent = getCellPosition(cell).lockedPosition();
                    cell.setTemplateValue("tagAndPosition", indexCurrent + "");
                } else {
                    cell.setTag("table[@data-recordindex='" + index + "']//td");
                }
            }
            setChildNodes(childNodes);
            if (size) {
                setTag("table");
            } else {
                setTag("*");
                setFinalXPath("//table[@data-recordindex='" + index + "']");
            }
        } else {
            setChildNodes(childNodes);
        }
    }

    private List findCommonId(List> ids, List theMinList) {
        Set intersection = new HashSet<>(theMinList);
        for (List list : ids) {
            Set newIntersection = new HashSet<>();
            for (Integer i : list) {
                if (intersection.contains(i)) {
                    newIntersection.add(i);
                }
            }
            intersection = newIntersection;
        }
        return new ArrayList<>(intersection);
    }

    private static List getTheMinList(List> lists) {
        int size = -1;
        for (List list : lists) {
            int sizeTmp = list.size();
            if (size == -1) {
                size = sizeTmp;
            } else if (sizeTmp > 0) {
                if (sizeTmp < size) {
                    size = sizeTmp;
                }
            }
        }
        for (List list : lists) {
            if (list.size() == size) {
                return list;
            }
        }
        return null;
    }

    private Details getCellPosition(AbstractCell cell) {
        int firstColumns = getLockedCells();
        String positions = cell.getPathBuilder().getTemplatesValues().get("tagAndPosition")[0];
        int actualPosition = Integer.parseInt(positions);
        int lockedPosition = getLockedPosition(firstColumns, actualPosition);
        return new Details(firstColumns, actualPosition, lockedPosition);
    }

    record Details(int firstColumns, int actualPosition, int lockedPosition) {

        @Override
        public String toString() {
            return "Details{" +
                    "firstColumns=" + firstColumns +
                    ", actualPosition=" + actualPosition +
                    ", lockedPosition=" + lockedPosition +
                    '}';
        }
    }

    private int getLockedPosition(int firstColumns, int actualPosition) {
        if (actualPosition <= firstColumns) {
            return actualPosition;
        } else {
            return actualPosition - firstColumns;
        }
    }

    public Row(WebLocator grid, int indexRow, AbstractCell... cells) {
        this(grid, cells);
        setPosition(indexRow);
    }

    @Override
    public Cell getCell(int columnIndex) {
        if (isGridLocked()) {
            int firstColumns = getLockedCells();
            int position = 1;
            if (columnIndex > firstColumns) {
                columnIndex = columnIndex - firstColumns;
                position = 2;
            }
            return new Cell(this, columnIndex).setResultIdx(position);
        } else {
            return new Cell(this, columnIndex);
        }
    }

    private int getLockedCells() {
        Grid grid = getGridAsContainer();
        WebLocator containerLocked = new WebLocator(grid).setClasses("x-grid-scrollbar-clipper", "x-grid-scrollbar-clipper-locked");
        return new Row(containerLocked, 1).getCells();
    }

    public List getCellsText(int... excludedColumns) {
        return getCellsText((short) 0, excludedColumns);
    }

    public  V getCellsText(Class type, int... excludedColumns) {
        return getCellsText(type, (short) 0, excludedColumns);
    }

    public  V getCellsValues(V type, int... excludedColumns) {
        return getCellsValues(type, (short) 0, excludedColumns);
    }

    public  V getCellsText(Class type, short columnLanguages, int... excludedColumns) {
        return getCellsText(type, t -> t == columnLanguages, Cell::getLanguages, excludedColumns);
    }

    public  V getCellsValues(V type, short columnLanguages, int... excludedColumns) {
        return getCellsValues(type, t -> t == columnLanguages, Cell::getLanguages, excludedColumns);
    }

    public  V getCellsText(Class type, Predicate predicate, Function function, int... excludedColumns) {
        List cellsText = getCellsText(predicate, function, excludedColumns);
        return transformToObject(type, cellsText);
    }

    public  V getCellsValues(V type, Predicate predicate, Function function, int... excludedColumns) {
        List cellsText = getCellsText(predicate, function, excludedColumns);
        List columnsList = Arrays.stream(excludedColumns).boxed().toList();
        return transformToObject(type, cellsText, columnsList);
    }

    public List getCellsText(short columnLanguages, int... excludedColumns) {
        if (isGridLocked()) {
            return getLockedCellsText(t -> t == columnLanguages, Cell::getLanguages, excludedColumns);
        } else {
            return getCellsText(t -> t == columnLanguages, Cell::getLanguages, excludedColumns);
        }
    }

    public List getCellsText(Predicate predicate, Function function, int... excludedColumns) {
        List columns = getColumns(excludedColumns);
        return getValues(predicate, function, columns);
    }

    public List getValues(Predicate predicate, Function function, List columns) {
        List list = new ArrayList<>();
        for (int j : columns) {
            Cell cell = new Cell(this, j);
            if (predicate.test(j)) {
                list.add(function.apply(cell));
            } else {
                try {
                    list.add(cell.getText(true).replaceAll("\n", " ").trim());
                } catch (NullPointerException e) {
                    Utils.sleep(1);
                }
            }
        }
        return list;
    }

    public  List getValues(Options options, List columns) {
        List list = new ArrayList<>();
        for (int j : columns) {
            Cell cell = new Cell(this, j);
            if (options.getPredicate().test(j)) {
                list.add(options.getFunction().apply(cell));
            } else {
                try {
                    list.add(cell.getText(true).replaceAll("\n", " ").trim());
                } catch (NullPointerException e) {
                    Utils.sleep(1);
                }
            }
        }
        return list;
    }

    public List getLockedCellsText(Predicate predicate, Function function, int... excludedColumns) {
        int firstColumns = getLockedCells();
        List columns = getColumns(excludedColumns);
        List list = new ArrayList<>();
        WebLocator container = getPathBuilder().getContainer();
        WebLocator containerLocked = new WebLocator().setClasses("x-grid-scrollbar-clipper", "x-grid-scrollbar-clipper-locked");
        WebLocator containerUnLocked = new WebLocator().setClasses("x-grid-scrollbar-clipper").setExcludeClasses("x-grid-scrollbar-clipper-locked");
        String finalXPath = getPathBuilder().getFinalXPath();
        for (int j : columns) {
            if (j > firstColumns) {
                setFinalXPath(containerUnLocked.getXPath() + finalXPath);
            } else {
                setFinalXPath(containerLocked.getXPath() + finalXPath);
            }
            int currentPosition = getLockedPosition(firstColumns, j);
            Cell cell = new Cell(this, currentPosition);
            if (predicate.test(j)) {
                list.add(function.apply(cell));
            } else {
                list.add(cell.getText().trim());
            }
        }
        setFinalXPath(finalXPath);
        setContainer(container);
        return list;
    }

    private String getVersion() {
        if (version == null) {
            WebLocator container = getPathBuilder().getContainer();
            Grid grid;
            if (container instanceof Grid) {
                grid = (Grid) container;
            } else {
                grid = (Grid) container.getPathBuilder().getContainer();
            }
            if (grid != null) {
                this.version = grid.getVersion();
            }
        }
        return version;
    }

    public  T setVersion(String version) {
        this.version = version;
        return (T) this;
    }

    public boolean select() {
        scrollInGrid();
        return isSelected() || selectPrivate() && isSelected();
    }

    public boolean doSelect() {
        scrollInGrid();
        return isSelected() || doSelectPrivate() && isSelected();
    }

    public boolean unSelect() {
        scrollInGrid();
        return !isSelected() || selectPrivate() && !isSelected();
    }

    public boolean doUnSelect() {
        scrollInGrid();
        return !isSelected() || doSelectPrivate() && !isSelected();
    }

    private boolean selectPrivate() {
        WebLocator checkBox = new WebLocator(this);
        if ("6.7.0".equals(getVersion())) {
            checkBox.setBaseCls("x-selmodel-column");
        } else {
            checkBox.setBaseCls("x-grid-row-checker");
        }
        return checkBox.click();
    }

    private boolean doSelectPrivate() {
        WebLocator checkBox = new WebLocator(this);
        if ("6.7.0".equals(getVersion())) {
            checkBox.setBaseCls("x-selmodel-column");
        } else {
            checkBox.setBaseCls("x-grid-row-checker");
        }
        return checkBox.doClick();
    }

    public boolean isSelected() {
        String aClass = getAttributeClass();
        return !Strings.isNullOrEmpty(aClass) && aClass.contains("x-grid-item-selected");
    }

    public boolean expand() {
        scrollInGrid();
        return !isCollapsed() || RetryUtils.retry(2, () -> doExpanded() && !isCollapsed());
    }

    public boolean collapse() {
        scrollInGrid();
        return isCollapsed() || RetryUtils.retry(2, () -> doExpanded() && isCollapsed());
    }

    protected boolean doExpanded() {
        WebLocator expander = new WebLocator(this).setBaseCls("x-grid-row-expander");
        return expander.doClick();
    }

    public boolean isCollapsed() {
        String aClass = getAttributeClass();
        return !Strings.isNullOrEmpty(aClass) && aClass.contains("x-grid-row-collapsed");
    }

    public boolean scrollTo() {
        return RetryUtils.retryIfNotSame(70, true, () -> {
            WebLocator container = getPathBuilder().getContainer();
            int lastRowVisibleIndex = new Row(container).findElements().size() - 1;
            Row row = new Row(container, lastRowVisibleIndex);
            WebLocatorUtils.scrollToWebLocator(row);
            return isPresent();
        });
    }

    public boolean scrollInGrid() {
        Grid grid = getGridAsContainer();
        if (grid.canItScroll()) {
            if (grid.isScrollBottom()) {
                grid.scrollTop();
            }
            return RetryUtils.retry(100, () -> {
                boolean render = waitToRender(Duration.ofMillis(100), false) || grid.isScrollBottom();
                if (!render) {
                    grid.scrollPageDown();
                }
                return render;
            });
        } else {
            return false;
        }
    }

    public int getCells() {
        return new Cell(this).size();
    }

    public Row getNextRow() {
        return getNextRow("/");
    }

    public Row getNextRow(String root) {
        return getNextRow(root, "following-sibling::table[1]");
    }

    public Row getNextRow(String root, String tag) {
        return new Row(this).setRoot(root).setTag(tag);
    }

    public Row getNextRow(AbstractCell... cells) {
        return getNextRow("/", cells);
    }

    public Row getNextRow(String root, AbstractCell... cells) {
        AbstractCell[] childNodes = Stream.of(cells).filter(t -> t != null && (t.getPathBuilder().getText() != null || (t.getPathBuilder().getChildNodes() != null && !t.getPathBuilder().getChildNodes().isEmpty()))).toArray(AbstractCell[]::new);
        return new Row(this).setRoot(root).setTag("following-sibling::table").setChildNodes(childNodes);
    }
}