Python get_notification_type Examples

Example #1

    def test_get_notification_type_returns_information_for_non_failed_other_job(
            self, mock_logger):
        expected = INFORMATION_NOTIFICATION_TYPE
        actual = batch_job_handler.get_notification_type(
            OTHER_JOB_QUEUE,
            SUCCEEDED_JOB_STATUS,
            JOB_NAME,
        )

        self.assertEqual(expected, actual)

Example #2

    def test_get_notification_type_returns_error_for_failed_other_job(
            self, mock_logger):
        expected = WARNING_NOTIFICATION_TYPE
        actual = batch_job_handler.get_notification_type(
            OTHER_JOB_QUEUE,
            FAILED_JOB_STATUS,
            JOB_NAME,
        )

        self.assertEqual(expected, actual)