在selenium中运行Function时，为了流畅的等待获取空指针异常

Question

按钮为null

时的颜色

  

线程中的异常＆＃34; main＆＃34;显示java.lang.NullPointerException

public static void main(String[] args) throws InterruptedException {

    WebDriver driver = new FirefoxDriver();
    driver.get("http://toolsqa.com/automation-practice-switch-windows/");

    FluentWait<WebDriver> wait = new FluentWait<WebDriver>(driver);
    wait.pollingEvery(250,  TimeUnit.MILLISECONDS);
    wait.withTimeout(2, TimeUnit.SECONDS);

    Function<WebDriver, Boolean> function = new Function<WebDriver, Boolean>()
            {
                public Boolean apply(WebDriver arg0) {
                    WebElement element = arg0.findElement(By.id("colorVar"));
                    String color = element.getAttribute("color");
                    System.out.println("The color if the button is " + color);
                    if(color.equals("red"))
                    {
                        return true;
                    }
                    return false;
                }
            };

    wait.until(function);
}