def test_set_duration(self): task = DummyOperator(task_id='op', email='*****@*****.**') ti = TI( task=task, execution_date=datetime.datetime.now(), ) ti.start_date = datetime.datetime(2018, 10, 1, 1) ti.end_date = datetime.datetime(2018, 10, 1, 2) ti.set_duration() self.assertEqual(ti.duration, 3600)
def test_set_duration(self): task = DummyOperator(task_id='op', email='*****@*****.**') ti = TI( task=task, execution_date=datetime.datetime.now(), ) ti.start_date = datetime.datetime(2018, 10, 1, 1) ti.end_date = datetime.datetime(2018, 10, 1, 2) ti.set_duration() self.assertEqual(ti.duration, 3600)
def test_set_duration_empty_dates(self): task = DummyOperator(task_id='op', email='*****@*****.**') ti = TI(task=task, execution_date=datetime.datetime.now()) ti.set_duration() self.assertIsNone(ti.duration)
def test_set_duration_empty_dates(self): task = DummyOperator(task_id='op', email='*****@*****.**') ti = TI(task=task, execution_date=datetime.datetime.now()) ti.set_duration() self.assertIsNone(ti.duration)