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'