def testSetupEnviromentShouldCallThePrepareFilesStrategyAndCopyTheScriptFile(self):
setup_enviroment = SetupEnviroment()
strategy = Mock()
setup_enviroment.prepare_files_strategy = strategy
automatic_correction = Mock()
return_value = TestSetupEnviroment.ORIGINAL_SCRIPT_FILE_PATH + TestSetupEnviroment.SCRIPT_FILE_NAME
automatic_correction.script = return_value
setup_enviroment.run(automatic_correction, TestSetupEnviroment.DESTINATION_PATH)
strategy.prepare_files.assert_called()
self.assertTrue(os.path.exists(TestSetupEnviroment.DESTINATION_PATH + TestSetupEnviroment.SCRIPT_FILE_NAME))
class AutomaticCorrectionRunner():
"""
This class is intended to be in charge of running the automated check on the
queued deliveries from the seal.model application model. It is thought to be
called in order to run the automatic correction script file for each delivery which has
not yet been checked, and for the Practices which has a script.
"""
TMP_DIR = managepath.get_instance().get_automatic_correction_tmp_dir()
SUCCESSFULL_RESULTS_KEY = "successfull"
FAILED_RESULTS_KEY = "failed"
def __init__(self):
self.selection_strategy = AutomaticCorrectionSelectionStrategyThroughRestApi(SERIALIZER_AUTH_USER, SERIALIZER_AUTH_PASS)
self.setup_enviroment = SetupEnviroment()
self.run_script_command = RunScriptCommand()
self.publish_result_visitors = (PublishResultsVisitorWeb(SERIALIZER_AUTH_USER, SERIALIZER_AUTH_PASS),
PublishResultsVisitorMail(SERIALIZER_AUTH_USER, SERIALIZER_AUTH_PASS),)
def clean_up_tmp_dir(self):
shutil.rmtree(AutomaticCorrectionRunner.TMP_DIR, ignore_errors=True)
def run(self):
"""Runs the corresponding script for every Delivery which has not yet been run."""
results = {AutomaticCorrectionRunner.SUCCESSFULL_RESULTS_KEY : 0, AutomaticCorrectionRunner.FAILED_RESULTS_KEY : 0}
pending_automatic_corrections = self.selection_strategy.get_automatic_corrections()
for pending_automatic_correction in pending_automatic_corrections:
try:
self.setup_enviroment.run(pending_automatic_correction, AutomaticCorrectionRunner.TMP_DIR)
self.run_script_command.set_script(os.path.join(AutomaticCorrectionRunner.TMP_DIR, os.path.basename(pending_automatic_correction.script)))
script_result = self.run_script_command.execute()
except Exception, e:
script_result = ScriptResult()
script_result.captured_stdout = "An error has occurred when running the automatic correction process. Error information: " + str(e)
script_result.exit_value = 2
script_result.automatic_correction = pending_automatic_correction
for visitor in self.publish_result_visitors:
script_result.accept(visitor)
self.clean_up_tmp_dir()
if(script_result.exit_value == 0):
results[AutomaticCorrectionRunner.SUCCESSFULL_RESULTS_KEY] += 1
else :
results[AutomaticCorrectionRunner.FAILED_RESULTS_KEY] += 1
return results;
class AutomaticCorrectionRunner():
"""
This class is intended to be in charge of running the automated check on the
queued deliveries from the seal.model application model. It is thought to be
called in order to run the automatic correction script file for each delivery which has
not yet been checked, and for the Practices which has a script.
"""
TMP_DIR = managepath.get_instance().get_automatic_correction_tmp_dir()
SUCCESSFULL_RESULTS_KEY = "successfull"
FAILED_RESULTS_KEY = "failed"
def __init__(self):
self.selection_strategy = AutomaticCorrectionSelectionStrategyThroughRestApi(
SERIALIZER_AUTH_USER, SERIALIZER_AUTH_PASS)
self.setup_enviroment = SetupEnviroment()
self.run_script_command = RunScriptCommand()
self.publish_result_visitors = (
PublishResultsVisitorWeb(SERIALIZER_AUTH_USER,
SERIALIZER_AUTH_PASS),
PublishResultsVisitorMail(SERIALIZER_AUTH_USER,
SERIALIZER_AUTH_PASS),
)
def clean_up_tmp_dir(self):
shutil.rmtree(AutomaticCorrectionRunner.TMP_DIR, ignore_errors=True)
def run(self):
"""Runs the corresponding script for every Delivery which has not yet been run."""
results = {
AutomaticCorrectionRunner.SUCCESSFULL_RESULTS_KEY: 0,
AutomaticCorrectionRunner.FAILED_RESULTS_KEY: 0
}
pending_automatic_corrections = self.selection_strategy.get_automatic_corrections(
)
for pending_automatic_correction in pending_automatic_corrections:
try:
self.setup_enviroment.run(pending_automatic_correction,
AutomaticCorrectionRunner.TMP_DIR)
self.run_script_command.set_script(
os.path.join(
AutomaticCorrectionRunner.TMP_DIR,
os.path.basename(pending_automatic_correction.script)))
script_result = self.run_script_command.execute()
except Exception, e:
script_result = ScriptResult()
script_result.captured_stdout = "An error has occurred when running the automatic correction process. Error information: " + str(
e)
script_result.exit_value = 2
script_result.automatic_correction = pending_automatic_correction
for visitor in self.publish_result_visitors:
script_result.accept(visitor)
self.clean_up_tmp_dir()
if (script_result.exit_value == 0):
results[AutomaticCorrectionRunner.SUCCESSFULL_RESULTS_KEY] += 1
else:
results[AutomaticCorrectionRunner.FAILED_RESULTS_KEY] += 1
return results