def test_skip_serve_logs_on_worker_start(self):
with patch(
'airflow.cli.commands.worker_command.Process') as mock_popen:
args = self.parser.parse_args(['worker', '-c', '-1', '-s'])
with patch('celery.platforms.check_privileges') as mock_privil:
mock_privil.return_value = 0
worker_command.worker(args)
mock_popen.assert_not_called()
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)
worker_command.worker(mock_args)
self.assertEqual(cm.exception.code, 1)
def test_skip_serve_logs_on_worker_start(self):
with patch('airflow.cli.commands.worker_command.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
worker_command.worker(args)
mock_popen.assert_not_called()