예제 #1
0
    def handle(self, *args, **kwargs):
        queues = [queue for queue in args]
        concurrency = kwargs['concurrency']

        _main(queue_prefixes=queues, concurrency=concurrency)
예제 #2
0
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()