def test_wait_for_loading_return_False(self): driver = WebDriverMock() driver.expect("window_handles", [1, 2, 3]) driver.expect("current_url", "http://www.google.co.th") page = Page("google", "http://www.google.com") page.driver = driver with pytest.raises(RuntimeError): page.wait_for_loading(timeout=0)
def test_wait_for_loading_return_True(self): driver = WebDriverMock() driver.expect("window_handles", [1, 2, 3]) driver.expect("current_url", "http://www.google.com") page = Page("google", "http://www.google.com") page.driver = driver assert page.wait_for_loading(timeout=0) == True