catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Remove process_timeouts() method in favor of timeout(<Timeout>) and a helper for tests

Browse source
This commit is contained in:
Jean-Baptiste Barth 2015-11-05 02:47:05 +01:00
commit d007dfe3ff
4 changed files with 21 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

15
moto/swf/models/workflow_execution.py
View file

@ -154,8 +154,8 @@ class WorkflowExecution(object):
# workflow execution timeout
_timeout = self.first_timeout()
if _timeout:
self.execute_timeout(_timeout)
# TODO: process child policy on child workflows here or in process_timeouts()
self.timeout(_timeout)
# TODO: process child policy on child workflows here or in timeout()
self._add_event(
"WorkflowExecutionTimedOut",
child_policy=self.child_policy,
@ -168,7 +168,7 @@ class WorkflowExecution(object):
_timeout = task.first_timeout()
if task.state == "STARTED" and _timeout:
self.should_schedule_decision_next = True
task.process_timeouts()
task.timeout(_timeout)
self._add_event(
"DecisionTaskTimedOut",
event_timestamp=_timeout.timestamp,
@ -182,7 +182,7 @@ class WorkflowExecution(object):
_timeout = task.first_timeout()
if task.open and _timeout:
self.should_schedule_decision_next = True
task.process_timeouts()
task.timeout(_timeout)
self._add_event(
"ActivityTaskTimedOut",
details=task.details,
@ -538,16 +538,11 @@ class WorkflowExecution(object):
if _timeout.reached:
return _timeout
def execute_timeout(self, timeout):
def timeout(self, timeout):
self.execution_status = "CLOSED"
self.close_status = "TIMED_OUT"
self.timeout_type = timeout.kind
def process_timeouts(self):
_timeout = self.first_timeout()
if _timeout:
self.execute_timeout(_timeout)
@property
def open(self):
return self.execution_status == "OPEN"