Skip to content
Snippets Groups Projects
Commit d3684dc6 authored by Jedrzej Rybicki's avatar Jedrzej Rybicki
Browse files

one more to fix

parent 70dbc09a
No related branches found
No related tags found
No related merge requests found
Pipeline #77457 failed
import unittest import unittest
from airflow.utils.state import State from airflow.utils.state import State
from airflow.utils.dates import days_ago
from dags.b2shareoperator import B2ShareOperator from dags.b2shareoperator import B2ShareOperator
from airflow import DAG from airflow import DAG
from airflow.models.taskinstance import TaskInstance from airflow.models.taskinstance import TaskInstance
...@@ -9,13 +10,13 @@ TEST_DAG_ID = 'test_my_custom_operator' ...@@ -9,13 +10,13 @@ TEST_DAG_ID = 'test_my_custom_operator'
class B2ShareOperatorTest(unittest.TestCase): class B2ShareOperatorTest(unittest.TestCase):
def setUp(self): def setUp(self):
self.dag = DAG(TEST_DAG_ID, schedule_interval='@daily', default_args={'start_date' : DEFAULT_DATE}) self.dag = DAG(TEST_DAG_ID, schedule_interval='@daily', default_args={'start_date' : days_ago(2)})
self.op = B2ShareOperator( self.op = B2ShareOperator(
dag=self.dag, dag=self.dag,
task_id='test', task_id='test',
name='test_name' name='test_name'
) )
self.ti = TaskInstance(task=self.op, execution_date=DEFAULT_DATE) self.ti = TaskInstance(task=self.op, execution_date=days_ago(1))
def test_execute_no_trigger(self): def test_execute_no_trigger(self):
self.ti.run(ignore_ti_state=True) self.ti.run(ignore_ti_state=True)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment