@Override
public void setTimeout(Timeout timeout) {
WebDriver.Timeouts tm = getDriver().manage().timeouts();
if (timeout.getWait() != 0) {
tm.implicitlyWait(timeout.getWait(), TimeUnit.SECONDS);
}
if (timeout.getScript() != 0) {
tm.setScriptTimeout(timeout.getScript(), TimeUnit.SECONDS);
}
if (timeout.getLoad() != 0) {
tm.pageLoadTimeout(timeout.getLoad(), TimeUnit.SECONDS);
}
}
