def test_sleep(m_get_context):
m_decider = MagicMock(spec=Decider)
m_decider.handle_execute_timer.return_value = 'Works'
m_get_context.return_value = MagicMock(spec=DecisionContext, decider=m_decider)
assert 'Works' == workflow_time.sleep(10)
assert m_decider.handle_execute_timer.mock_calls == [call(10)]
def test_sleep(m_get_context):
m_decider = MagicMock(spec=Decider)
m_decider.handle_execute_timer.return_value = 'Works'
m_get_context.return_value = MagicMock(spec=DecisionContext,
decider=m_decider)
assert 'Works' == workflow_time.sleep(10)
assert m_decider.handle_execute_timer.mock_calls == [call(10)]
def do_try_except():
arg_sum = 0
try:
arg_sum += yield BunchOfActivities.sum(arg1, arg2)
yield BunchOfActivities.throw()
except ActivityTaskFailedError as err:
if isinstance(err.cause, ValueError) \
and str(err.cause) == 'Hello-Error':
if err.event_id != 13 or err.activity_id != '2':
raise RuntimeError("Test Failed")
arg_sum += yield BunchOfActivities.sum(arg1, arg2)
yield workflow_time.sleep(1)
return_(arg_sum)
def execute(self):
yield workflow_time.sleep(4)
return_(self.msg)
def test_sleep_no_context(m_get_context):
m_get_context.side_effect = AttributeError()
with pytest.raises(TypeError):
workflow_time.sleep()
def test_sleep_wrong_context():
with pytest.raises(TypeError):
workflow_time.sleep(10)
def test_sleep_no_context(m_get_context):
m_get_context.side_effect = AttributeError()
with pytest.raises(TypeError):
workflow_time.sleep()
def test_sleep_wrong_context():
with pytest.raises(TypeError):
workflow_time.sleep(10)
def execute(self, arg1, arg2):
yield workflow_time.sleep(2)
arg_sum = yield self.activities_client.sum(arg1, arg2)
return_(arg_sum)
def execute(self):
yield workflow_time.sleep(4)
return_(self.msg)
def execute(self):
activity_future = self.activities_client.wrong_tasklist_activity()
yield workflow_time.sleep(2)
yield activity_future.cancel()
yield activity_future
return_(False)
def execute(self):
self.activities_client.heartbeating_activity(5)
yield workflow_time.sleep(1)
self.cancel()
return_(True)
def execute(self):
activity_future = self.activities_client.heartbeating_activity(5)
yield workflow_time.sleep(2)
activity_future._activity_id = '100' # set invalid ID
yield activity_future.cancel()
return_(False)
def execute(self):
activity_future = self.activities_client.heartbeating_activity(10)
yield workflow_time.sleep(2)
yield activity_future.cancel()
yield activity_future
return_(False)
