def test_cli_test_with_params(self):
task_command.task_test(self.parser.parse_args([
'tasks', 'test', 'example_passing_params_via_test_command', 'run_this',
'--task-params', '{"foo":"bar"}', DEFAULT_DATE.isoformat()]))
task_command.task_test(self.parser.parse_args([
'tasks', 'test', 'example_passing_params_via_test_command', 'also_run_this',
'--task-params', '{"foo":"bar"}', DEFAULT_DATE.isoformat()]))
def test_cli_test(self):
task_command.task_test(self.parser.parse_args([
'tasks', 'test', 'example_bash_operator', 'runme_0',
DEFAULT_DATE.isoformat()]))
task_command.task_test(self.parser.parse_args([
'tasks', 'test', 'example_bash_operator', 'runme_0', '--dry-run',
DEFAULT_DATE.isoformat()]))
def test_cli_test_with_env_vars(self):
with redirect_stdout(io.StringIO()) as stdout:
task_command.task_test(self.parser.parse_args([
'tasks', 'test', 'example_passing_params_via_test_command', 'env_var_test_task',
'--env-vars', '{"foo":"bar"}', DEFAULT_DATE.isoformat()]))
output = stdout.getvalue()
self.assertIn('foo=bar', output)
self.assertIn('AIRFLOW_TEST_MODE=True', output)
def test_test(self):
"""Test the `airflow test` command"""
args = self.parser.parse_args([
"tasks", "test", "example_python_operator", 'print_the_context', '2018-01-01'
])
with redirect_stdout(io.StringIO()) as stdout:
task_command.task_test(args)
# Check that prints, and log messages, are shown
self.assertIn("'example_python_operator__print_the_context__20180101'", stdout.getvalue())
def test_test(self):
"""Test the `airflow test` command"""
args = create_mock_args(task_id='print_the_context',
dag_id='example_python_operator',
subdir=None,
execution_date=timezone.parse('2018-01-01'))
with redirect_stdout(io.StringIO()) as stdout:
task_command.task_test(args)
# Check that prints, and log messages, are shown
self.assertIn("'example_python_operator__print_the_context__20180101'",
stdout.getvalue())
def test_test(self, mock_run_mini_scheduler):
"""Test the `airflow test` command"""
args = self.parser.parse_args([
"tasks", "test", "example_python_operator", 'print_the_context',
'2018-01-01'
])
with redirect_stdout(io.StringIO()) as stdout:
task_command.task_test(args)
mock_run_mini_scheduler.assert_not_called()
# Check that prints, and log messages, are shown
assert "'example_python_operator__print_the_context__20180101'" in stdout.getvalue(
)
def test_test(self):
"""Test the `airflow test` command"""
args = create_mock_args(task_id='print_the_context',
dag_id='example_python_operator',
subdir=None,
execution_date=timezone.parse('2018-01-01'))
saved_stdout = sys.stdout
try:
sys.stdout = out = io.StringIO()
task_command.task_test(args)
output = out.getvalue()
# Check that prints, and log messages, are shown
self.assertIn(
"'example_python_operator__print_the_context__20180101'",
output)
finally:
sys.stdout = saved_stdout
