def test_prepare_output_config_with_optional_params(): kms_key = 'key' assemble_with = 'Line' accept = 'text/csv' expected_config = { 'S3OutputPath': OUTPUT_PATH, 'KmsKeyId': kms_key, 'AssembleWith': assemble_with, 'Accept': accept, } actual_config = _TransformJob._prepare_output_config(OUTPUT_PATH, kms_key, assemble_with, accept) assert actual_config == expected_config
def test_prepare_output_config_with_optional_params(): kms_key = "key" assemble_with = "Line" accept = "text/csv" expected_config = { "S3OutputPath": OUTPUT_PATH, "KmsKeyId": kms_key, "AssembleWith": assemble_with, "Accept": accept, } actual_config = _TransformJob._prepare_output_config( OUTPUT_PATH, kms_key, assemble_with, accept) assert actual_config == expected_config
def test_prepare_output_config(): config = _TransformJob._prepare_output_config(OUTPUT_PATH, None, None, None) assert config == {'S3OutputPath': OUTPUT_PATH}