def setup_method(self, method):
sys.argv.clear()
sys.argv.append("spam")
sys.argv.append("spam")
cmd_parser = CommandArgumentParser()
self._cmd_args = cmd_parser.parse()
self._log_file = os.path.join(env.BASE_DIR, "logs", "app.log")
runner = ScenarioRunner(self._cmd_args)
runner.add_system_path()
def test_execute_scenario_ok(self):
test_data = {
"scenario": [{
"arguments": {
"src_dir": self._pj_dir,
"src_pattern": r"(.*)\.csv"
},
"class": "FileRename",
"step": "",
}]
}
with open(self._scenario_file, "w") as f:
f.write(yaml.dump(test_data, default_flow_style=False))
runner = ScenarioRunner(self._cmd_args)
runner.execute_scenario()
def test_create_scenario_queue_ok(self):
os.makedirs(self._pj_dir)
test_data = {
"scenario": [{
"arguments": {
"retry_count": 10
},
"class": "SftpFileExtract",
"step": "sftp_file_extract",
}]
}
with open(self._scenario_file, "w") as f:
f.write(yaml.dump(test_data, default_flow_style=False))
is_completed_queue_creation = True
try:
runner = ScenarioRunner(self._cmd_args)
runner.create_scenario_queue()
except Exception:
is_completed_queue_creation = False
else:
shutil.rmtree(self._pj_dir)
assert is_completed_queue_creation is True
def test_add_system_path(self):
runner = ScenarioRunner(self._cmd_args)
runner.add_system_path()
includes_spam = any("project/spam" in p for p in sys.path)
assert includes_spam is True