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_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_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)