扩展Selenium WebDriver WebElement？

Question

我遵循Selenium建议的Page Object模式，但我如何为页面创建更专业的WebElement。具体来说，我们的页面上有表，我编写了一些帮助函数来获取表的特定行、返回表的内容等。

目前，以下是我创建的一个页面对象的片段，其中包含一个表：

public class PermissionsPage  {

    @FindBy(id = "studyPermissionsTable")
    private WebElement permissionTable;

    @FindBy(id = "studyPermissionAddPermission")
    private WebElement addPermissionButton;

    ... 

}

所以，我想做的是让permissionsTable成为一个更定制的WebElement，它具有我前面提到的一些方法。

例如：

public class TableWebElement extends WebElement {
    WebElement table;
    // a WebDriver needs to come into play here too I think

    public List<Map<String, String>> getTableData() {
        // code to do this
    }

    public int getTableSize() {
        // code to do this
    }

    public WebElement getElementFromTable(String id) {
        // code to do this
    }
}

我希望我所要解释的有道理。我想我要寻找的是一种让这个自定义WebElement做一些特定于表的其他事情的方法。将此自定义元素添加到页面，并利用Selenium基于注释将webelements连接到页面的方式。

有可能吗？如果是这样的话，有人知道如何做到这一点吗？

Answer 1

作为后续，这就是我最终所做的（以防其他人有同样的问题）。下面是我作为WebElement的包装器创建的类的片段：

public class CustomTable {

private WebDriver driver;
private WebElement tableWebElement;

public CustomTable(WebElement table, WebDriver driver) {
    this.driver = driver;
    tableWebElement = table;
}

public WebElement getTableWebElement() {
    return tableWebElement;
}

public List<WebElement> getTableRows() {
    String id = tableWebElement.getAttribute("id");
    return driver.findElements(By.xpath("//*[@id='" + id + "']/tbody/tr"));
}

public List<WebElement> getTableHeader() {
    String id = tableWebElement.getAttribute("id");
    return tableWebElement.findElements(By.xpath("//*[@id='" + id + "']/thead/tr/th"));
}   
.... more utility functions here
}

然后我在任何通过引用创建的页面中使用它，如下所示：

public class TestPage {

@FindBy(id = "testTable")
private WebElement myTestTable;

/**
 * @return the myTestTable
 */
public CustomTable getBrowserTable() {
    return new CustomTable(myTestTable, getDriver());
}

我唯一不喜欢的是，当页面想要获取表时，它会创建一个“新”表。我这样做是为了避免在更新表中的数据（即单元格更新、行添加、行删除）时发生StaleReferenceException。如果有人对我如何避免每次请求时都创建一个新实例有任何建议，而是返回更新后的WebElement，那就太好了！

Answer 2

您可以使用WebDriver扩展框架创建自定义WebElement，该框架提供了一个WebComponent类，该类实现WebElement接口

创建您的自定义WebElement

public class Table extends WebComponent {
    @FindBy(tagName = "tr")
    List<Row> rows;

    public Row getRow(int row) {
        return rows.get(row - 1);
    }

    public int getTableSize() {
        return rows.size();
    }

    public static class Row extends WebComponent {
        @FindBy(tagName = "td")
        List<WebElement> columns;

        public WebElement getCell(int column) {
            return columns.get(column - 1);
        }
    }
}

...然后使用@FindBy注释将其添加到您的PageObject中，并在调用PageFactory.initElements方法时使用WebDriverExtensionFieldDecorator

public class PermissionPage {
    public PermissionPage(WebDriver driver) {
        PageFactory.initElements(new WebDriverExtensionFieldDecorator(driver), this);
    }

    @FindBy(id = "studyPermissionsTable")
    public Table permissionTable;

    @FindBy(id = "studyPermissionAddPermission")
    public WebElement addPermissionButton;
}

。。。然后在测试中使用它

public class PermissionPageTest {
    @Test
    public void exampleTest() {
        WebDriver driver = new FirefoxDriver();
        PermissionPage permissionPage = new PermissionPage(driver);

        driver.get("http://www.url-to-permission-page.com");
        assertEquals(25, permissionPage.permissionTable.getTableSize());
        assertEquals("READ", permissionPage.permissionTable.getRow(2).getCell(1).getText());
        assertEquals("WRITE", permissionPage.permissionTable.getRow(2).getCell(2).getText());
        assertEquals("EXECUTE", permissionPage.permissionTable.getRow(2).getCell(3).getText());
    }
}

或者更好地使用WebDriver Extensions PageObject实现

public class PermissionPage extends WebPage {
    @FindBy(id = "studyPermissionsTable")
    public Table permissionTable;

    @FindBy(id = "studyPermissionAddPermission")
    public WebElement addPermissionButton;

    @Override
    public void open(Object... arguments) {
        open("http://www.url-to-permission-page.com");
        assertIsOpen();
    }

    @Override
    public void assertIsOpen(Object... arguments) throws AssertionError {
        assertIsDisabled(permissionTable);
        assertIsDisabled(addPermissionButton);
    }
}

以及具有WebElements静态断言方法的JUnitRunner

import static com.github.webdriverextensions.Bot.*;

@RunWith(WebDriverRunner.class)
public class PermissionPageTest {

    PermissionPage permissionPage;

    @Test
    @Firefox
    public void exampleTest() {
        open(permissionPage);
        assertSizeEquals(25, permissionPage.permissionTable.rows);
        assertTextEquals("READ", permissionPage.permissionTable.getRow(2).getCell(1));
        assertTextEquals("WRITE", permissionPage.permissionTable.getRow(2).getCell(2));
        assertTextEquals("EXECUTE", permissionPage.permissionTable.getRow(2).getCell(3));
    }
}

Answer 3

我创建了一个结合了所有WebDriver界面的界面：

public interface Element extends WebElement, WrapsElement, Locatable {}

它只是用来包装WebElements在包装元素时可以做的所有事情。

然后是实现：

public class ElementImpl implements Element {

    private final WebElement element;

    public ElementImpl(final WebElement element) {
        this.element = element;
    }

    @Override
    public void click() {
        element.click();
    }

    @Override
    public void sendKeys(CharSequence... keysToSend) {
        element.sendKeys(keysToSend);
    }

    // And so on, delegates all the way down...

}

然后，例如复选框：

public class CheckBox extends ElementImpl {

    public CheckBox(WebElement element) {
        super(element);
    }

    public void toggle() {
        getWrappedElement().click();
    }

    public void check() {
        if (!isChecked()) {
            toggle();
        }
    }

    public void uncheck() {
        if (isChecked()) {
            toggle();
        }
    }

    public boolean isChecked() {
        return getWrappedElement().isSelected();
    }
}

在我的脚本中使用它时：

CheckBox cb = new CheckBox(element);
cb.uncheck();

我还提出了一种包装元素类的方法。您必须创建一些工厂来替换内置的PageFactory，但这是可行的，并且提供了很大的灵活性。

我在我的网站上记录了这个过程：

• 包装WebElement：第1部分
• 包装WebElement：第2部分

我还有一个名为selophane的项目，它的灵感来自于这个问题和其他问题：selophane