Update system test example_emr to have logs (#30715)

Author: vincbeck

tests/system/providers/amazon/aws/example_emr.py

@@ -33,6 +33,7 @@
     EmrModifyClusterOperator,
     EmrTerminateJobFlowOperator,
 )
+from airflow.providers.amazon.aws.operators.s3 import S3CreateBucketOperator, S3DeleteBucketOperator
 from airflow.providers.amazon.aws.sensors.emr import EmrJobFlowSensor
 from airflow.utils.trigger_rule import TriggerRule
 from tests.system.providers.amazon.aws.utils import ENV_ID_KEY, SystemTestContextBuilder
@@ -126,12 +127,18 @@ def delete_security_config(config_name: str):
     tags=["example"],
 ) as dag:
     test_context = sys_test_context_task()
+
     env_id = test_context[ENV_ID_KEY]
     config_name = f"{CONFIG_NAME}-{env_id}"
     execution_role_arn = test_context[EXECUTION_ROLE_ARN_KEY]
+    s3_bucket = f"{env_id}-emr-bucket"
+
+    JOB_FLOW_OVERRIDES["LogUri"] = f"s3://{s3_bucket}/"
     JOB_FLOW_OVERRIDES["SecurityConfiguration"] = config_name
     JOB_FLOW_OVERRIDES["Instances"]["InstanceGroups"][0]["CustomAmiId"] = get_ami_id()
 
+    create_s3_bucket = S3CreateBucketOperator(task_id="create_s3_bucket", bucket_name=s3_bucket)
+
     create_security_configuration = configure_security_config(config_name)
 
     # [START howto_operator_emr_create_job_flow]
@@ -172,9 +179,17 @@ def delete_security_config(config_name: str):
 
     delete_security_configuration = delete_security_config(config_name)
 
+    delete_s3_bucket = S3DeleteBucketOperator(
+        task_id="delete_s3_bucket",
+        bucket_name=s3_bucket,
+        force_delete=True,
+        trigger_rule=TriggerRule.ALL_DONE,
+    )
+
     chain(
         # TEST SETUP
         test_context,
+        create_s3_bucket,
         create_security_configuration,
         # TEST BODY
         create_job_flow,
@@ -184,6 +199,7 @@ def delete_security_config(config_name: str):
         remove_cluster,
         check_job_flow,
         delete_security_configuration,
+        delete_s3_bucket,
     )
 
     from tests.system.utils.watcher import watcher