def test_close_all_no_error(self):
cache = WebDriverCache()
driver0, driver1, driver2 = mock(), mock(), mock()
when(driver0).quit().thenReturn(None)
when(driver1).quit().thenReturn(None)
when(driver2).quit().thenReturn(None)
cache.close_all()
self.verify_cache(cache)
def test_close_all_no_error(self):
cache = WebDriverCache()
driver0, driver1, driver2 = mock(), mock(), mock()
when(driver0).quit().thenReturn(None)
when(driver1).quit().thenReturn(None)
when(driver2).quit().thenReturn(None)
cache.close_all()
self.verify_cache(cache)
def test_close_all_cache_first_quite_fails(self):
cache = WebDriverCache()
driver = mock()
when(driver).quit().thenRaise(TimeoutException("timeout."))
cache.register(driver, "bar")
with self.assertRaises(TimeoutException):
cache.close_all()
self.verify_cache(cache)
def test_close_all_cache_first_quite_fails(self):
cache = WebDriverCache()
driver = mock()
when(driver).quit().thenRaise(TimeoutException('timeout.'))
cache.register(driver, 'bar')
with self.assertRaises(TimeoutException):
cache.close_all()
self.verify_cache(cache)
def test_close_all_cache_not_selenium_error(self):
cache = WebDriverCache()
driver0, driver1, driver2 = mock(), mock(), mock()
when(driver0).quit().thenRaise(RemoteDriverServerException("stuff."))
when(driver1).quit().thenRaise(ValueError("stuff."))
when(driver2).quit().thenRaise(TimeoutException("timeout."))
cache.register(driver0, "bar0")
cache.register(driver1, "bar1")
cache.register(driver2, "bar2")
with self.assertRaises(TimeoutException):
cache.close_all()
self.verify_cache(cache)
def test_close_all_cache_middle_quite_fails(self):
cache = WebDriverCache()
driver0, driver1, driver2 = mock(), mock(), mock()
when(driver0).quit().thenReturn(None)
when(driver1).quit().thenRaise(TimeoutException("timeout."))
when(driver2).quit().thenReturn(None)
cache.register(driver0, "bar0")
cache.register(driver1, "bar1")
cache.register(driver2, "bar2")
with self.assertRaises(TimeoutException):
cache.close_all()
self.verify_cache(cache)
def test_close_all_cache_all_quite_fails(self):
cache = WebDriverCache()
driver0, driver1, driver2 = mock(), mock(), mock()
when(driver0).quit().thenRaise(RemoteDriverServerException('stuff.'))
when(driver1).quit().thenRaise(RemoteDriverServerException('stuff.'))
when(driver2).quit().thenRaise(TimeoutException('timeout.'))
cache.register(driver0, 'bar0')
cache.register(driver1, 'bar1')
cache.register(driver2, 'bar2')
with self.assertRaises(TimeoutException):
cache.close_all()
self.verify_cache(cache)
def test_close_all_cache_not_selenium_error(self):
cache = WebDriverCache()
driver0, driver1, driver2 = mock(), mock(), mock()
when(driver0).quit().thenRaise(RemoteDriverServerException('stuff.'))
when(driver1).quit().thenRaise(ValueError('stuff.'))
when(driver2).quit().thenRaise(TimeoutException('timeout.'))
cache.register(driver0, 'bar0')
cache.register(driver1, 'bar1')
cache.register(driver2, 'bar2')
with self.assertRaises(TimeoutException):
cache.close_all()
self.verify_cache(cache)
def test_close_all_cache_middle_quite_fails(self):
cache = WebDriverCache()
driver0, driver1, driver2 = mock(), mock(), mock()
when(driver0).quit().thenReturn(None)
when(driver1).quit().thenRaise(TimeoutException('timeout.'))
when(driver2).quit().thenReturn(None)
cache.register(driver0, 'bar0')
cache.register(driver1, 'bar1')
cache.register(driver2, 'bar2')
with self.assertRaises(TimeoutException):
cache.close_all()
self.verify_cache(cache)
def test_close_only_called_once(self):
cache = WebDriverCache()
browser1 = mock()
browser2 = mock()
browser3 = mock()
cache.register(browser1)
cache.register(browser2)
cache.register(browser3)
cache.close()
verify(browser3, times=1).quit()
cache.close_all()
verify(browser1, times=1).quit()
verify(browser2, times=1).quit()
verify(browser3, times=1).quit()
def test_close_only_called_once(self):
cache = WebDriverCache()
browser1 = mock()
browser2 = mock()
browser3 = mock()
cache.register(browser1)
cache.register(browser2)
cache.register(browser3)
cache.close()
verify(browser3, times=1).quit()
cache.close_all()
verify(browser1, times=1).quit()
verify(browser2, times=1).quit()
verify(browser3, times=1).quit()