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

10
tests/test_swf/models/test_activity_task.py
View file

@ -8,7 +8,11 @@ from moto.swf.models import (
Timeout,
)
from ..utils import make_workflow_execution, ACTIVITY_TASK_TIMEOUTS
from ..utils import (
ACTIVITY_TASK_TIMEOUTS,
make_workflow_execution,
process_first_timeout,
)
def test_activity_task_creation():
@ -101,7 +105,7 @@ def test_activity_task_first_timeout():
# activity task timeout is 300s == 5mins
with freeze_time("2015-01-01 12:06:00"):
task.first_timeout().should.be.a(Timeout)
task.process_timeouts()
process_first_timeout(task)
task.state.should.equal("TIMED_OUT")
task.timeout_type.should.equal("HEARTBEAT")
@ -123,7 +127,7 @@ def test_activity_task_cannot_timeout_on_closed_workflow_execution():
with freeze_time("2015-01-01 14:10:00"):
task.first_timeout().should.be.a(Timeout)
wfe.first_timeout().should.be.a(Timeout)
wfe.process_timeouts()
process_first_timeout(wfe)
task.first_timeout().should.be.none
def test_activity_task_cannot_change_state_on_closed_workflow_execution():