def close_driver(driver, timeout=10): ut.timeout(timeout) try: driver.close() except (sel_exceptions.UnexpectedAlertPresentException, cm.TimeExceededError): pass ut.cancel_timeout()
def test_cancel_timeout(self): start_time = time() ut.timeout(1) sleep(0.3) ut.cancel_timeout() elapsed_time = time() - start_time self.assertLess(elapsed_time, 5, 'Cannot cancel timer: %s' % elapsed_time)
def driver_get(driver, visit_info, timeout=cm.SOFT_TIMEOUT): ut.timeout(cm.SOFT_TIMEOUT) t0 = time.time() ############################################################# # The real visit happens here ############################### driver.get(visit_info.url) ############################################################# ############################################################# visit_info.duration = time.time() - t0 ut.cancel_timeout()