def test_get_friendly_name_returns_default_name(self, mock_logger):
expected = "Batch job"
actual = batch_job_handler.get_friendly_name(
OTHER_JOB_QUEUE,
JOB_NAME,
FAILED_JOB_STATUS,
)
self.assertEqual(expected, actual)
def test_get_friendly_name_returns_kafka_matched_name(self, mock_logger):
expected = "Kafka reconciliation batch"
actual = batch_job_handler.get_friendly_name(
KAFKA_JOB_QUEUE,
"kafka_job_name",
FAILED_JOB_STATUS,
)
self.assertEqual(expected, actual)
def test_get_friendly_name_returns_a_matched_name(self, mock_logger):
expected = "Object tagger"
actual = batch_job_handler.get_friendly_name(
PDM_JOB_QUEUE,
"job-name",
FAILED_JOB_STATUS,
)
self.assertEqual(expected, actual)
def test_get_friendly_name_returns_pt_1_matched_name_with_pt_job_queue(
self, mock_logger):
expected = "PT-1 object tagger"
actual = batch_job_handler.get_friendly_name(
PT_JOB_QUEUE,
"pdm_Pt_1_job-name",
FAILED_JOB_STATUS,
)
self.assertEqual(expected, actual)
