def test_schedule_with_completed_activity(monkeypatch): """Test the scheduling of an activity. """ mock(monkeypatch) from tests.fixtures.flows import example monkeypatch.setattr(decider, 'schedule_activity_task', MagicMock()) decisions = MagicMock() schedule_context = decider.ScheduleContext() instance_state = activity.ActivityState('activity_1') instance_state.add_state(activity.ACTIVITY_COMPLETED) current_activity = example.activity_1 history = { current_activity.name: { 'workflow_name_activity_1-1-schedule_id': instance_state } } resp = decider.schedule(decisions, schedule_context, history, {}, 'schedule_id', current_activity) assert not decider.schedule_activity_task.called assert resp.get_last_state() == activity.ACTIVITY_COMPLETED assert schedule_context.completed resp.result.get('foo')
def test_schedule_with_completed_activity(monkeypatch): """Test the scheduling of an activity. """ mock(monkeypatch) from tests.fixtures.flows import example monkeypatch.setattr(decider, 'schedule_activity_task', MagicMock()) decisions = MagicMock() schedule_context = decider.ScheduleContext() instance_state = activity.ActivityState('activity_1') instance_state.add_state(activity.ACTIVITY_COMPLETED) current_activity = example.activity_1 history = { current_activity.name: { 'workflow_name_activity_1-1-schedule_id': instance_state } } resp = decider.schedule( decisions, schedule_context, history, {}, 'schedule_id', current_activity) assert not decider.schedule_activity_task.called assert resp.get_last_state() == activity.ACTIVITY_COMPLETED assert schedule_context.completed resp.result.get('foo')
def test_schedule_with_unscheduled_activity(monkeypatch): """Test the scheduling of an activity. """ from tests.fixtures.flows import example monkeypatch.setattr(decider, 'schedule_activity_task', MagicMock()) decisions = MagicMock() schedule_context = decider.ScheduleContext() history = {} current_activity = example.activity_1 decider.schedule(decisions, schedule_context, history, {}, 'schedule_id', current_activity) assert decider.schedule_activity_task.called assert not schedule_context.completed
def test_schedule_with_unscheduled_activity(monkeypatch): """Test the scheduling of an activity. """ mock(monkeypatch) from tests.fixtures.flows import example monkeypatch.setattr(decider, 'schedule_activity_task', MagicMock()) decisions = MagicMock() schedule_context = decider.ScheduleContext() history = {} current_activity = example.activity_1 decider.schedule( decisions, schedule_context, history, {}, 'schedule_id', current_activity) assert decider.schedule_activity_task.called assert not schedule_context.completed