Как получить размер веб-страницы с помощью селена webdriver?

Question

Мой сценарий прокручивает 500 пикселей до тех пор, пока он не достигнет конца страницы, чтобы я мог делать скриншот через каждые 500 прокручиваемых пикселей.

Ниже мой код:

    try {           
//this provides the last height of the web page
                int lastHeight = ((Number) ((JavascriptExecutor) webDriver).executeScript("return document.body.scrollHeight")).intValue();

                do {
                    ((JavascriptExecutor) webDriver).executeScript("window.scrollBy(0,500)");

//This returns the height of the window
                    contentHeight = ((Number) ((JavascriptExecutor) webDriver).executeScript("return window.innerHeight")).intValue();
                    Thread.sleep(2000);                 
                }while(!(lastHeight == contentHeight));
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

Поэтому вместо приведенного ниже кода

contentHeight = ((Number) ((JavascriptExecutor) webDriver).executeScript("return window.innerHeight")).intValue();

Я хочу получить высоту веб-страницы в текущей позиции.