def test_schedule_activity_task(monkeypatch): """Test scheduling an activity task. """ mock(monkeypatch) from tests.fixtures.flows import example instance = list(example.activity_1.instances({}))[0] decisions = MagicMock() decider.schedule_activity_task(decisions, instance) decisions.schedule_activity_task.assert_called_with( instance.id, instance.activity_name, '1.0', task_list=instance.activity_worker.task_list, input=json.dumps(instance.create_execution_input()), heartbeat_timeout=str(instance.heartbeat_timeout), start_to_close_timeout=str(instance.timeout), schedule_to_start_timeout=str(instance.schedule_to_start), schedule_to_close_timeout=str(instance.schedule_to_close))
def test_schedule_activity_task(monkeypatch): """Test scheduling an activity task. """ from tests.fixtures.flows import example instance = list(example.activity_1.instances({}))[0] decisions = [] decider.schedule_activity_task(decisions, instance) expects = dict(decisionType='ScheduleActivityTask', scheduleActivityTaskDecisionAttributes=dict( activityId=instance.id, activityType=dict(name=instance.activity_name, version='1.0'), taskList=dict(name=instance.activity_worker.task_list), input=json.dumps(instance.create_execution_input()), heartbeatTimeout=str(instance.heartbeat_timeout), startToCloseTimeout=str(instance.timeout), scheduleToStartTimeout=str(instance.schedule_to_start), scheduleToCloseTimeout=str(instance.schedule_to_close))) assert expects in decisions