@@ -1559,32 +1559,52 @@ def test_register_process_target_object_attributes(self):
15591559 def test_change_run_status_complete (self ):
15601560 """
15611561 Testing that when changing the run status from 'running' to 'complete' the run record updates successfully.
1562+ Process record is also updated for last_completed_date_time.
15621563 :return:
15631564 """
1564- self .process_tracker .change_run_status (new_status = "completed" )
1565+ end_date = datetime .now ()
1566+ self .process_tracker .change_run_status (
1567+ new_status = "completed" , end_date = end_date
1568+ )
15651569
1566- run_record = self .session .query (ProcessTracking ).filter (
1567- ProcessTracking .process_id == self .process_id
1570+ run_record = (
1571+ self .session .query (
1572+ ProcessTracking .process_status_id , Process .last_completed_run_date_time
1573+ )
1574+ .join (Process )
1575+ .filter (ProcessTracking .process_id == self .process_id )
15681576 )
15691577
1570- given_result = run_record [0 ].process_status_id
1571- expected_result = self .process_tracker .process_status_complete
1578+ given_result = [
1579+ run_record [0 ].process_status_id ,
1580+ run_record [0 ].last_completed_run_date_time ,
1581+ ]
1582+ expected_result = [self .process_tracker .process_status_complete , end_date ]
15721583
15731584 self .assertEqual (expected_result , given_result )
15741585
15751586 def test_change_run_status_failed (self ):
15761587 """
15771588 Testing that when changing the run status from 'running' to 'failed' the run record updates successfully.
1589+ Process record is also updated for last_failed_date_time.
15781590 :return:
15791591 """
1580- self .process_tracker .change_run_status (new_status = "failed" )
1592+ end_date = datetime .now ()
1593+ self .process_tracker .change_run_status (new_status = "failed" , end_date = end_date )
15811594
1582- run_record = self .session .query (ProcessTracking ).filter (
1583- ProcessTracking .process_id == self .process_id
1595+ run_record = (
1596+ self .session .query (
1597+ ProcessTracking .process_status_id , Process .last_failed_run_date_time
1598+ )
1599+ .join (Process )
1600+ .filter (ProcessTracking .process_id == self .process_id )
15841601 )
15851602
1586- given_result = run_record [0 ].process_status_id
1587- expected_result = self .process_tracker .process_status_failed
1603+ given_result = [
1604+ run_record [0 ].process_status_id ,
1605+ run_record [0 ].last_failed_run_date_time ,
1606+ ]
1607+ expected_result = [self .process_tracker .process_status_failed , end_date ]
15881608
15891609 self .assertEqual (expected_result , given_result )
15901610
0 commit comments