Yes, I'll split this into a separate unit test rather than the current end-to-end test so that I can mock the webdriver session and assert the right calls are made to it.