def test_create_scenario_queue_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.create_scenario_queue()
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