def setUp(self): self.manager = UrlManager('www.google.com')
class TestUrlManager(unittest.TestCase): def setUp(self): self.manager = UrlManager('www.google.com') def tearDown(self): self.manager = None def test_initial(self): self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.google.com'])) def test_add_url(self): self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.google.com'])) self.manager.add_url('www.facebook.com') self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.google.com', 'www.facebook.com'])) self.manager.add_url('www.facebook.com') self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.google.com', 'www.facebook.com'])) google_url = self.manager.get_url() self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.facebook.com'])) self.manager.add_url('www.google.com') self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.facebook.com'])) def test_add_urls(self): self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.google.com'])) self.manager.add_urls(['www.facebook.com', 'www.twitter.com']) self.assertEqual( self.manager.urls_to_crawl, collections.deque( ['www.google.com', 'www.facebook.com', 'www.twitter.com']) ) def test_get_url(self): self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.google.com'])) self.assertEqual(self.manager.has_url(), True) google_url = self.manager.get_url() self.assertEqual(google_url, 'www.google.com') self.assertEqual( self.manager.urls_to_crawl, collections.deque([])) def test_has_url(self): self.assertEqual( self.manager.urls_to_crawl, collections.deque(['www.google.com'])) self.assertEqual(self.manager.has_url(), True) google_url = self.manager.get_url() self.assertEqual(self.manager.has_url(), False)