为selenium测试设计页面对象结构。从WebElement继承导致问题

Question

我试图设计一些页面对象，并与整体设计有一些问题。

我知道测试中的一个常见问题是确定元素是否存在，解决方案是使用try/catch或使用findElements（…）如果结果列表的大小为0，则断言。

我想在所有测试中避免这些尝试/陷阱。是否可以使用继承将try/catch放在一个超类中，从而使测试更干净？我希望该方法返回null，如果元素不存在，那么我可以只assertNotNull（…）在测试中。

以下是我当前解决方案中页面对象的一个方法：

public WebElement getRegisterLink(){
    List<WebElement> list = driver.findElements(By.linktext("Register"));
    if(list.size() == 0){
        return null;
    }
        return list.get(0);
}

问题是有很多类似的重复代码。我的getLoginButton（）、getSomePageLink（）等看起来都一样。

这是我的一个想法。。。

public abstract class MyCompanyElement implements WebElement{

WebElement element;

public MyCompanyElement(WebElement element){
    this.element = element;
}

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

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

@Override
public MyCompanyElement findElement(By arg0) {      
    try{
        MyCompanyElement e = (MyCompanyElement) element.findElement(arg0);
        return e;
    } catch (NoSuchElementException ex){
        return null;
    }

}

@Override
public List<MyCompanyElement> findElements(By arg0) {       
    List<MyCompanyElement> list = new ArrayList<MyCompanyElement>();
    for(WebElement e : element.findElements(arg0)){
        list.add((MyCompanyElement) e);
    }
    return list;
}

//Rest of the methods here which just call the methods of element
}

然后我有一个BasePage类，所有页面对象都将扩展它。

public abstract class BasePage {

MyCompanyElement element;

public BasePage(WebDriver driver){
    element = (MyCompanyElement) driver.findElement(By.tagName("html"));        
}

//More methods here

}

然后，页面对象有一些方法，如getLoginButton（）、getCrysterLink（）等，它们返回MyCompanyElement而不是WebElement。（主页包含所有网页上的元素，如页眉和页脚）。

然后，我的测试可能会如下所示：

Homepage homepage = new Homepage(driver);
homepage.getLoginButton().click();

如果我想检查元素是否存在，那么。。。

assertNotNull(homepage.getRegisterLink());

上述解决方案的问题是：

• 在MyCompanyElement中，findElement（）可以返回MyCompanyElement。但是编译器不喜欢我在findElements（）中返回MyCompanyElement的列表。有解决办法吗

谢谢

Answer 1

我觉得你有点过分了。您不需要创建自己版本的WebElement。

您应该做的是，将一个方法放入接受给定By的基类中，例如：

public WebElement getFirstMatchingElement(By by)
{
    List<WebElement> list = driver.findElements(by);
    if(list.size() == 0) {
        return null;
    }
    return list.get(0);
}

那么它就足够通用了，可以这样称呼它。。。。

public WebElement getRegisterLink() {
    return getFirstMatchingElement(By.linktext("Register"));
}

就像。。。

public WebElement getLoginButton() {
    return getFirstMatchingElement(By.id("loginButton"));
}

这通过将代码抽象到另一个方法来解决您的问题，这意味着您的调用者（即您实际找到元素的地方）更干净、更简单，并且不需要“了解”所有“列表内容”。