如何使用mockito来模拟webdriver?

时间:2014-06-16 11:18:15

标签: java selenium webdriver mockito

我希望能够为我的Selenium集成测试进行单元测试,为此我需要能够模拟驱动程序和测试所需的元素,这是一个返回名称的函数的快速示例元素。

public class demo {

  private WebDriver mockDriver;
  private WebElement mockElement;

  @Before
  public void setUp(){
    this.mockDriver = mock(WebDriver.class);
    this.mockElement = mock(WebElement.class, withSettings().name("elementName"));
    when(this.mockDriver.findElement(By.id("testmock"))).thenReturn(mockElement);
  }

  public String getName(String id){
    WebElement testElement = mockDriver.findElement(By.id(id));
    return testElement.getAttribute("name");
  }

  @Test
  public void assertElementName() throws InterruptedException {
    Assert.assertTrue(getName("testmock").equals("elementName"));
  }
}

这会在java.lang.NullPointerException的回复中给我一个getName()。 我显然使用了这个错误,但我无法弄清楚如何。任何有这方面经验的人都能指出我正确的方向吗?

1 个答案:

答案 0 :(得分:4)

您误解了这段代码的作用:

this.mockElement = mock(WebElement.class, withSettings().name("elementName"));

withSettings()子句为模拟对象提供了一个名称,这意味着Mockito生成的某些错误消息将使用此名称。您WebElement对象上设置任何属性。

所以......当你的代码到达这个部分时:

return testElement.getAttribute("name");

它返回null,因为没有具有该值的属性。如果你想拥有一个属性,那么你需要添加如下内容:

when(this.mockElement.getAttribute("name")).thenReturn("elementName");