def test_result_mapper_str(self):
m = Mock()
m.return_value = {'returned': 'bar'}
ctx = DefaultContext()
ctx.baz = None
with patch('inspect.getargspec') as argspec:
argspec.return_value = [[]]
Step(m, result_map={'baz': 'returned'})(ctx)
assert m.called
assert ctx.baz == 'bar'
def test_result_mapper_callable(self):
m = Mock()
m.return_value = {'returned': ['abc', 'bar']}
ctx = DefaultContext()
ctx.baz = None
def reverse_and_join(result, context): #@UnusedVariable
return "".join(result['returned'])[::-1]
with patch('inspect.getargspec') as argspec:
argspec.return_value = [[]]
Step(m, result_map={'baz': reverse_and_join})(ctx)
assert m.called
assert ctx.baz == 'rabcba'
