def test_last_run_state_for_jobs(mock_status_last_run): mock_status_last_run.side_effect = [ ('faketimestamp', chronos_tools.LastRunState.Success), ('faketimestamp', chronos_tools.LastRunState.Fail), ('faketimestamp', chronos_tools.LastRunState.NotRun), ] assert check_chronos_jobs.last_run_state_for_jobs([{}, {}, {}]) == [ ({}, chronos_tools.LastRunState.Success), ({}, chronos_tools.LastRunState.Fail), ({}, chronos_tools.LastRunState.NotRun), ]
def test_last_run_state_for_jobs(mock_status_last_run): mock_status_last_run.side_effect = lambda x: { 'a': [('faketimestamp', chronos_tools.LastRunState.Success)], 'b': [('faketimestamp', chronos_tools.LastRunState.Fail)], 'c': [('faketimestamp', chronos_tools.LastRunState.NotRun)], }[x] assert check_chronos_jobs.last_run_state_for_jobs(['a', 'b', 'c']) == [ ('a', ('faketimestamp', chronos_tools.LastRunState.Success)), ('b', ('faketimestamp', chronos_tools.LastRunState.Fail)), ('c', ('faketimestamp', chronos_tools.LastRunState.NotRun)), ]