def test_close(self):
cache = BrowserCache()
browser = mock()
cache.register(browser)
verify(browser, times=0).quit() # sanity check
cache.close()
verify(browser, times=1).quit()
def test_close_only_called_once(self):
cache = BrowserCache()
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 = BrowserCache()
browser1 = mock()
browser2 = mock()
browser3 = mock()
cache.register(browser1)
cache.register(browser2)
cache.register(browser3)
browsers = cache.get_open_browsers()
self.assertEqual(len(browsers), 3)
self.assertEqual(browsers[0], browser1)
self.assertEqual(browsers[1], browser2)
self.assertEqual(browsers[2], browser3)
cache.close()
browsers = cache.get_open_browsers()
self.assertEqual(len(browsers), 2)
self.assertEqual(browsers[0], browser1)
self.assertEqual(browsers[1], browser2)