class CrawlerTestCase(unittest.TestCase):
def setUp(self):
self.crawler = Crawler()
def test_scrape(self):
flag = self.crawler.scrape('手機')
self.assertTrue(flag)
def test_scrape_return_none(self):
flag = self.crawler.scrape()
self.assertFalse(flag)
def main():
"""main
Crawling from the keyword
"""
flag = False
crawler = Crawler()
parser = argparse.ArgumentParser()
parser.add_argument("-k",
"--keyword",
help="keyword to search",
default="templates/ssh.j2")
args = parser.parse_args()
flag = crawler.scrape(args.keyword)
return flag
