def testCancelTimerShouldInvokeCancelMethodTimer(self):
process = Mock()
timer = Mock()
process_timeout = ProcessTimeout(process, 1000)
process_timeout.timer = timer
process_timeout.cancel_timer()
timer.cancel.assert_called()
def testStartTimerShouldInvokeStartMethodTimer(self):
process = Mock()
timer = Mock()
process_timeout = ProcessTimeout(process, 1000)
process_timeout.timer = timer
process_timeout.start_timer()
timer.start.assert_called()
def execute(self):
if (self.script is None):
self.log.error(
"attempt to run the correction process but the script to be run is not defined."
)
raise IllegalStateException(
reason="In order to execute the script, you must set it first."
)
# now we may call the script
self.log.debug("launching correction process...")
# FIXME: this is plain shit. We should try to use 'with'
automatic_correction_tmp_dir = os.path.dirname(
self.script
) # managepath.get_instance().get_automatic_correction_tmp_dir()
current_dir = os.getcwd()
self.log.debug("current dir: %s", current_dir)
self.log.debug("script dir : %s", automatic_correction_tmp_dir)
os.chdir(automatic_correction_tmp_dir)
process = subprocess.Popen([self.script],
shell=False,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
os.chdir(current_dir)
# fin FIXME!!!
# must ensure that the process won't run forever
process_timer = ProcessTimeout(process, RunScriptCommand.TIME_OUT)
process_timer.start_timer()
self.log.debug("Process timeout timer launched")
# finally, we must capture all results so the can be published
script_result = ScriptResult()
self.log.debug("waiting for process to finish...")
script_result.exit_value = process.wait()
self.log.debug("process finished with exit value %d",
script_result.exit_value)
#if the result has been obtained, the is no point on keeping the timer alive
if process_timer.ran:
self.log.info(
"Execution has been terminated for exceding the timeout limit."
)
else:
process_timer.cancel_timer()
self.log.debug(
"Process finished correctly without exceding timeout limit.")
output = process.communicate()
script_result.captured_stdout = output[0]
self.log.debug("stdout captured.")
self.log.debug("excecution completed.")
return script_result
