def test_trace_task_ret__no_trace(self): try: delattr(mytask, '__trace__') except AttributeError: pass res = _trace_task_ret(mytask.name, uuid(), [4], {}) self.assertEqual(res, 4 ** 4)
def test_trace_task_ret(self): mytask.__trace__ = build_tracer(mytask.name, mytask, self.app.loader, 'test') res = _trace_task_ret(mytask.name, uuid(), [4], {}) self.assertEqual(res, 4 ** 4)