def test_create_list_from_comments_args(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--comments", ["test_url", "10"]]) l_type = "comments" assert Cli.GetPRAWScrapeSettings().create_list(args, l_type) == ["test_url"]
def test_create_list_from_redditor_args(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--redditor", ["test_redditor", "10"]]) l_type = "redditor" assert Cli.GetPRAWScrapeSettings().create_list(args, l_type) == ["test_redditor"]
def test_list_switch_method_third_switch(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--comments", "test_url"]) index = 2 assert Cli.GetPRAWScrapeSettings()._list_switch(args, index) == ["test_url"]
def test_list_switch_method_second_switch(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--redditor", "test_redditor"]) index = 1 assert Cli.GetPRAWScrapeSettings()._list_switch(args, index) == ["test_redditor"]
def test_list_switch_method_first_switch(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--subreddit", "test_subreddit"]) index = 0 assert Cli.GetPRAWScrapeSettings()._list_switch(args, index) == ["test_subreddit"]
def test_two_arg_settings_one_arg_comments(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--comments", ["test_url", "2"]]) master = {"test_url": None} invalids = [] Cli.GetPRAWScrapeSettings()._two_arg_settings(args.comments, invalids, master) assert master == {"test_url": "2"}
def test_two_arg_settings_one_arg_redditor(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--redditor", ["test_redditor", "10"]]) master = {"test_redditor": None} invalids = [] Cli.GetPRAWScrapeSettings()._two_arg_settings(args.redditor, invalids, master) assert master == {"test_redditor": "10"}
def test_subreddit_settings_one_subreddit(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--subreddit", ["test_subreddit", "h", "10"]]) master = {"test_subreddit": []} invalids = [] Cli.GetPRAWScrapeSettings()._subreddit_settings(args.subreddit, invalids, master) assert master == {"test_subreddit": [["h", "10", None]]}
def test_create_list_from_subreddit_args(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--subreddit", ["test_subreddit", "h", "10"]]) l_type = "subreddit" assert Cli.GetPRAWScrapeSettings().create_list(args, l_type) == \ ["test_subreddit"]
def test_get_settings_with_comments_args(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--comments", ["test_url", "2"]]) master = {"test_url": None} s_type = "comments" invalids = [] Cli.GetPRAWScrapeSettings().get_settings(args, invalids, master, s_type) assert master == {"test_url": "2"}
def test_get_settings_with_redditor_args(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--redditor", ["test_redditor", "10"]]) master = {"test_redditor": None} s_type = "redditor" invalids = [] Cli.GetPRAWScrapeSettings().get_settings(args, invalids, master, s_type) assert master == {"test_redditor": "10"}
def test_get_settings_with_subreddit_args(self): parser = MakeArgs.make_scraper_args() args = parser.parse_args(["--subreddit", ["test_subreddit", "h", "10"]]) master = {"test_subreddit": []} s_type = "subreddit" invalids = [] Cli.GetPRAWScrapeSettings().get_settings(args, invalids, master, s_type) assert master == {"test_subreddit": [["h", "10", None]]}
def test_set_sub_settings_len_is_four(self): sub = ["test_one", "S", "20", "day"] settings = Cli.GetPRAWScrapeSettings()._set_sub_settings(sub) assert ["S", "20", "day"] == settings
def test_set_sub_settings_len_is_three_with_non_filterable_category(self): sub = ["test_one", "H", "20"] settings = Cli.GetPRAWScrapeSettings()._set_sub_settings(sub) assert ["H", "20", None] == settings