Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix dynamic allocation specs handling for custom launcher #38223

Merged
merged 4 commits into from
Mar 18, 2024

Conversation

sudohainguyen
Copy link
Contributor

@sudohainguyen sudohainguyen commented Mar 17, 2024

Primarily about a fix to handle dynamicAllocation specs gracefully and adding unit test for custom object launcher.

Context:
When defining dynamic allocation specs, if users accidentally miss any required fields, currently we throw a KeyError exception which looks confused. So that I've put a minor change to make the exception thrown should be AirflowException with a more meaningful error message.

On the other hand, we also don't have unit test to cover some helper classes in custom_object_launcher.py, this PR also includes a few simple test cases.


^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in newsfragments.

@sudohainguyen sudohainguyen changed the title Add unit test for Spark Kubernetes custom launcher Fix dynamic allocation specs handling for custom launcher Mar 17, 2024
Copy link
Collaborator

@dirrao dirrao left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@Taragolis Taragolis merged commit d4350a6 into apache:main Mar 18, 2024
55 checks passed
@sudohainguyen sudohainguyen deleted the cncf/unittest branch March 18, 2024 23:01
utkarsharma2 pushed a commit to astronomer/airflow that referenced this pull request Apr 22, 2024
* Add unit test for Spark Kubernetes custom launcher

Signed-off-by: Hai Nguyen <[email protected]>

* Test on cloned entries correctly

Signed-off-by: Hai Nguyen <[email protected]>

* Fix lint issue

Signed-off-by: Hai Nguyen <[email protected]>

* Move cloned entries declaration out of raise assertion scope

Signed-off-by: Hai Nguyen <[email protected]>

---------

Signed-off-by: Hai Nguyen <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
area:providers provider:cncf-kubernetes Kubernetes provider related issues
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants