seanghaeli opened a new pull request, #66976: URL: https://github.com/apache/airflow/pull/66976
## Summary Fix the `example_sagemaker_unified_studio_notebook` system test by assuming the DataZone environment's native user role before calling `StartNotebookRun`. DataZone requires the caller to be the environment's authorized role — a custom test execution role with IAM permissions alone is not sufficient. Changes: - Add `DATAZONE_ROLE_ARN` variable via `SystemTestContextBuilder` - Add `@task setup_datazone_connection` to create an Airflow connection that assumes the role - Pass `aws_conn_id` to all operators/sensors - Skip deferrable path (operator does not propagate `aws_conn_id` to trigger — tracked separately) ## Test plan - [x] Ran system test — passes (build `eea9ac90`) - [x] Confirmed deferrable skip is consistent with `example_sagemaker_unified_studio` (also `deferrable=False`) --- ##### Was generative AI tooling used to co-author this PR? - [X] Yes — Claude Code (Opus 4.6) Generated-by: Claude Code (Opus 4.6) following [the guidelines](https://github.com/apache/airflow/blob/main/contributing-docs/05_pull_requests.rst#gen-ai-assisted-contributions) -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
