def test_set_headless_command(self):
""" --headless command should be set """
parse_args(self.argparse_mock)
self.argparse_mock.add_argument.assert_any_call(
"-head",
"--headless",
required=False,
help="Should browser be run headlessly",
action="count")
def test_set_ad_block_path_command(self):
""" --block command should be set """
parse_args(self.argparse_mock)
self.argparse_mock.add_argument.assert_any_call(
"-bl",
"--block",
required=False,
help="Path to ad blocking extension",
action="store",
type=str)
def test_set_bot_command(self):
""" --bot command should be set """
parse_args(self.argparse_mock)
self.argparse_mock.add_argument.assert_any_call(
"-b",
"--bot",
required=True,
help="Which bot should be used",
action="store",
type=str)
def test_set_scema_command(self):
""" --path command should be set """
parse_args(self.argparse_mock)
self.argparse_mock.add_argument.assert_any_call(
"-s",
"--scema",
required=True,
help="Path to database scema script",
action="store",
type=str)
def test_set_path_command(self):
""" --path command should be set """
parse_args(self.argparse_mock)
self.argparse_mock.add_argument.assert_any_call(
"-p",
"--path",
required=True,
help="Path to output files",
action="store",
type=str)
""" Bot that deals with js applications """
from argparse import ArgumentParser
from interactive_bots.commons.utils import parse_args
from interactive_bots.diagnosis_online_bot.reactive_bot import run as diagnosis_online_run
if __name__ == "__main__":
args = parse_args(ArgumentParser())
if args.bot == "diagnosis_online":
diagnosis_online_run(args)
else:
print("Bot not found")
def test_parse_args_should_be_called(self):
""" After setting up commands arguments must be parsed """
parse_args(self.argparse_mock)
self.assertTrue(self.argparse_mock.parse_args.called)