def test_nonexistent_project():
args = docopt(doc, "generate project_unknown project2")
gc = generate.GenerateCommand(args)
with output() as out:
gc.execute_command()
alert = out.getvalue().strip()
expected_alert = "project_unknown.json does not exist. Use ``scrapple genconfig``."
assert_in(expected_alert, alert)
def test_css_scraper_generate():
args = docopt(doc, "generate project2 project2")
gc = generate.GenerateCommand(args)
gc.execute_command()
with open(os.path.join(os.getcwd(), 'project2.py'), 'r') as f:
program = f.read()
assert_in("from scrapple.selectors.css import CssSelector", program)
assert_in('page0 = CssSelector("http://pyvideo.org/category/15/bostonpy")',
program)
def test_xpath_scraper_generate():
args = docopt(doc, "generate project1 project1")
gc = generate.GenerateCommand(args)
gc.execute_command()
with open(os.path.join(os.getcwd(), 'project1.py'), 'r') as f:
program = f.read()
assert_in("from scrapple.selectors.xpath import XpathSelector", program)
assert_in('page0 = XpathSelector("https://trakt.tv/shows/mr-robot")',
program)
def test_xpath_scraper_generate():
args = docopt(doc, "generate project1 project1")
gc = generate.GenerateCommand(args)
gc.execute_command()
with open(os.path.join(os.getcwd(), 'project1.py'), 'r') as f:
program = f.read()
assert_in("from scrapple.selectors.xpath import XpathSelector", program)
assert_in(
'page0 = XpathSelector("http://pyvideo.org/video/1785/python-for-humans-1")',
program)
def test_css_scraper_generate():
args = docopt(doc, "generate project2 project2")
gc = generate.GenerateCommand(args)
gc.execute_command()
with open(os.path.join(os.getcwd(), 'project2.py'), 'r') as f:
program = f.read()
assert_in("from scrapple.selectors.css import CssSelector", program)
assert_in(
'page0 = CssSelector("https://www.basketball-reference.com/teams/")',
program)
def test_if_generate_instance_created():
args = docopt(doc, "generate project1 test_project")
gc = generate.GenerateCommand(args)
assert_is_instance(gc, generate.GenerateCommand)
