def handle(self, *args, **kwargs): queues = [queue for queue in args] concurrency = kwargs['concurrency'] _main(queue_prefixes=queues, concurrency=concurrency)
def test_main_method(ManagerWorker): _main(["email1", "email2"], concurrency=2) ManagerWorker.assert_called_once_with(['email1', 'email2'], 2) ManagerWorker.return_value.start.assert_called_once_with()