def test_page_count(self): w = WebSite(None, None) w.domainBase = "google.com" w.max_page_count = 2 w.get_urls(WEB_SITE) self.assertEqual(w.max_page_count, 0) self.assertEqual(len(w.toProcess), 2)
def test_find_urls(self): sites = ["http://www.google.com/help", "http://www.google.com/mail", "https://mail.google.com"] w = WebSite(None, None) w.domainBase = "google.com" w.max_page_count=10 w.get_urls(WEB_SITE) for site in sites: self.assertTrue(site in w.toProcess) self.assertTrue(w is not None)