Python get_and_validate_job_detailsの例

コード例 #1

    def test_job_is_invalid_with_no_job_queue_field(self, mock_logger):
        message = {
            "detail": {
                JOB_NAME_KEY: JOB_NAME,
                JOB_STATUS_KEY: SUCCEEDED_JOB_STATUS,
            }
        }
        event = {"Records": [{"Sns": {"Message": f"{json.dumps(message)}"}}]}

        with pytest.raises(KeyError):
            actual = batch_job_handler.get_and_validate_job_details(
                event,
                SNS_TOPIC_ARN,
            )

コード例 #2

    def test_job_is_valid_with_valid_input(self, mock_logger):
        message = {
            "detail": {
                JOB_NAME_KEY: JOB_NAME,
                JOB_STATUS_KEY: SUCCEEDED_JOB_STATUS,
                JOB_QUEUE_KEY: PDM_JOB_QUEUE,
            }
        }
        event = {"Records": [{"Sns": {"Message": f"{json.dumps(message)}"}}]}

        expected = {
            JOB_NAME_KEY: JOB_NAME,
            JOB_STATUS_KEY: SUCCEEDED_JOB_STATUS,
            JOB_QUEUE_KEY: PDM_JOB_QUEUE,
        }
        actual = batch_job_handler.get_and_validate_job_details(
            event,
            SNS_TOPIC_ARN,
        )

        self.assertEqual(expected, actual)