i use this bit of code.

var url = await driver.executeScript("return window.location.href").then(function (url) {return url;});