def test_worker_started_with_required_arguments(self, mock_worker,
setup_logging_mock):
concurrency = '1'
celery_hostname = "celery_hostname"
queues = "queue"
autoscale = "2,5"
args = self.parser.parse_args([
'worker', '--autoscale', autoscale, '--concurrency', concurrency,
'--celery_hostname', celery_hostname, '--queues', queues
])
with mock.patch('celery.platforms.check_privileges') as mock_privil:
mock_privil.return_value = 0
cli.worker(args)
mock_worker.return_value.run.assert_called_once_with(
pool='prefork',
optimization='fair',
O='fair', # noqa
queues=queues,
concurrency=int(concurrency),
autoscale=autoscale,
hostname=celery_hostname,
loglevel=mock.ANY,
)
def test_error(self, mock_validate_session):
"""
Test to verify the exit mechanism of airflow-worker cli
by mocking validate_session method
"""
mock_validate_session.return_value = False
with self.assertRaises(SystemExit) as cm:
# airflow.bin.cli.worker(mock_args)
cli.worker(mock_args)
self.assertEqual(cm.exception.code, 1)
def test_error(self, mock_validate_session):
"""
Test to verify the exit mechanism of airflow-worker cli
by mocking validate_session method
"""
mock_validate_session.return_value = False
with self.assertRaises(SystemExit) as cm:
# airflow.bin.cli.worker(mock_args)
cli.worker(mock_args)
self.assertEqual(cm.exception.code, 1)
def test_skip_serve_logs_on_worker_start(self, celery_mock):
with patch('airflow.bin.cli.subprocess.Popen') as mock_popen:
mock_popen.return_value.communicate.return_value = (b'output', b'error')
mock_popen.return_value.returncode = 0
args = self.parser.parse_args(['worker', '-c', '-1', '-s'])
with patch('celery.platforms.check_privileges') as mock_privil:
mock_privil.return_value = 0
cli.worker(args)
mock_popen.assert_not_called()
