def test_custom_request_gets_instantiated(self):
_MyRequest = Mock(name='MyRequest')
class MyRequest(Request):
def __init__(self, *args, **kwargs):
Request.__init__(self, *args, **kwargs)
_MyRequest()
class MyTask(Task):
Request = MyRequest
@self.app.task(base=MyTask)
def failed():
raise AssertionError
sig = failed.s()
with self._context(sig) as C:
task_message_handler = default_strategy(failed, self.app,
C.consumer)
task_message_handler(C.message, None, None, None, None)
_MyRequest.assert_called()
def test_custom_request_gets_instantiated(self):
_MyRequest = Mock(name='MyRequest')
class MyRequest(Request):
def __init__(self, *args, **kwargs):
Request.__init__(self, *args, **kwargs)
_MyRequest()
class MyTask(Task):
Request = MyRequest
@self.app.task(base=MyTask)
def failed():
raise AssertionError
sig = failed.s()
with self._context(sig) as C:
task_message_handler = default_strategy(
failed,
self.app,
C.consumer
)
task_message_handler(C.message, None, None, None, None)
_MyRequest.assert_called()
