def test_cli_list_dag_runs(self):
dag_command.dag_trigger(
self.parser.parse_args([
'dags',
'trigger',
'example_bash_operator',
]))
args = self.parser.parse_args(
['dags', 'list_runs', 'example_bash_operator', '--no_backfill'])
dag_command.dag_list_dag_runs(args)
def test_trigger_dag(self):
dag_command.dag_trigger(
self.parser.parse_args([
'dags', 'trigger', 'example_bash_operator', '--conf',
'{"foo": "bar"}'
]))
self.assertRaises(
ValueError, dag_command.dag_trigger,
self.parser.parse_args([
'dags', 'trigger', 'example_bash_operator', '--run-id',
'trigger_dag_xxx', '--conf', 'NOT JSON'
]))
def test_cli_list_dag_runs(self):
dag_command.dag_trigger(self.parser.parse_args([
'dags', 'trigger', 'example_bash_operator', ]))
args = self.parser.parse_args(['dags',
'list_runs',
'--dag-id',
'example_bash_operator',
'--no-backfill',
'--start-date',
DEFAULT_DATE.isoformat(),
'--end-date',
timezone.make_aware(datetime.max).isoformat()])
dag_command.dag_list_dag_runs(args)