def test_run(mock_time):
# so the worker exits immediately
worker = Main(1, 'test name', ['default'])
def set_shutdown(_):
# stop the worker after one run
worker.state = State.shutdown
mock_time.side_effect = set_shutdown
worker.run()
def handle(self, *args, **options):
log_level = logging.INFO
if options['verbosity'] == 0:
log_level = logging.ERROR
if options['verbosity'] > 1:
log_level = logging.DEBUG
logging.basicConfig(level=log_level, format='%(message)s')
worker = Main(
options['concurrency'],
options['name'],
options['queues'],
)
worker.run()