def test_close_and_same_alias(self):
cache = WebDriverCache()
cache.register(mock(), "foo")
cache.register(mock(), "bar")
cache.close()
index = cache.get_index("bar")
self.assertEqual(index, None)
def test_close(self):
cache = WebDriverCache()
browser = mock()
cache.register(browser)
verify(browser, times=0).quit() # sanity check
cache.close()
verify(browser, times=1).quit()
def test_close_no_error(self):
cache = WebDriverCache()
driver = mock()
when(driver).quit().thenReturn(None)
cache.register(driver, "bar")
cache.close()
self.assertTrue(isinstance(cache.current, NoConnection))
self.assertTrue(driver in cache._closed)
def test_close(self):
cache = WebDriverCache()
browser = mock()
cache.register(browser)
verify(browser, times=0).quit() # sanity check
cache.close()
verify(browser, times=1).quit()
def test_close_no_error(self):
cache = WebDriverCache()
driver = mock()
when(driver).quit().thenReturn(None)
cache.register(driver, 'bar')
cache.close()
self.assertTrue(isinstance(cache.current, NoConnection))
self.assertTrue(driver in cache._closed)
def test_close_and_same_alias(self):
cache = WebDriverCache()
cache.register(mock(), 'foo')
cache.register(mock(), 'bar')
cache.close()
index = cache.get_index('bar')
self.assertEqual(index, None)
def test_close_quite_fails(self):
cache = WebDriverCache()
driver = mock()
when(driver).quit().thenRaise(TimeoutException("timeout."))
cache.register(driver, "bar")
with self.assertRaises(TimeoutException):
cache.close()
self.assertTrue(isinstance(cache.current, NoConnection))
self.assertTrue(driver in cache._closed)
def test_close_quite_fails(self):
cache = WebDriverCache()
driver = mock()
when(driver).quit().thenRaise(TimeoutException('timeout.'))
cache.register(driver, 'bar')
with self.assertRaises(TimeoutException):
cache.close()
self.assertTrue(isinstance(cache.current, NoConnection))
self.assertTrue(driver in cache._closed)
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()
def test_get_open_browsers(self):
cache = WebDriverCache()
driver1 = mock()
driver2 = mock()
driver3 = mock()
cache.register(driver1)
cache.register(driver2)
cache.register(driver3)
drivers = cache.active_drivers
self.assertEqual(len(drivers), 3)
self.assertEqual(drivers[0], driver1)
self.assertEqual(drivers[1], driver2)
self.assertEqual(drivers[2], driver3)
cache.close()
drivers = cache.active_drivers
self.assertEqual(len(drivers), 2)
self.assertEqual(drivers[0], driver1)
self.assertEqual(drivers[1], driver2)
def test_get_open_browsers(self):
cache = WebDriverCache()
driver1 = mock()
driver2 = mock()
driver3 = mock()
cache.register(driver1)
cache.register(driver2)
cache.register(driver3)
drivers = cache.active_drivers
self.assertEqual(len(drivers), 3)
self.assertEqual(drivers[0], driver1)
self.assertEqual(drivers[1], driver2)
self.assertEqual(drivers[2], driver3)
cache.close()
drivers = cache.active_drivers
self.assertEqual(len(drivers), 2)
self.assertEqual(drivers[0], driver1)
self.assertEqual(drivers[1], driver2)
def test_same_alias_new_browser(self):
cache = WebDriverCache()
cache.close()
index = cache.get_index("bar")
self.assertEqual(index, None)
def test_same_alias_new_browser(self):
cache = WebDriverCache()
cache.close()
index = cache.get_index('bar')
self.assertEqual(index, None)