Congratulations on your first Pull Request and welcome to the Apache Airflow community! If you have any issues or are unsure about any anything please check our Contribution Guide (https://github.com/apache/airflow/blob/main/CONTRIBUTING.rst)
Here are some useful points:

• Pay attention to the quality of your code (ruff, mypy and type annotations). Our pre-commits will help you with that.
• In case of a new feature add useful documentation (in docstrings or in docs/ directory). Adding a new operator? Check this short guide Consider adding an example DAG that shows how users should use it.
• Consider using Breeze environment for testing locally, it's a heavy docker but it ships with a working Airflow and a lot of integrations.
• Be patient and persistent. It might take some time to get a review or get the final approval from Committers.
• Please follow ASF Code of Conduct for all communication including (but not limited to) comments on Pull Requests, Mailing list and Slack.
• Be sure to read the Airflow Coding style.
  Apache Airflow is a community-driven project and together we are making it better .
  In case of doubts contact the developers at:
  Mailing List: dev@airflow.apache.org
  Slack: https://s.apache.org/airflow-slack

@jedcunningham Thank you for the review, good points, I should have applied all of the suggested changes on my last commit, please let me know in case I missed something.
I also consolidated the tests in a single file, if it the structure is fine I can try adding more test cases.

Rebased to see if everything works fine with latest main. @jedcunningham ?

@jedcunningham thanks again for the review, I should have applied both of the suggested changes on my last commit, please let me know in case I missed something.

Awesome work, congrats on your first merged pull request! You are invited to check our Issue Tracker for additional contributions.

Thanks @rzjfr! Congrats on your first commit