-
Notifications
You must be signed in to change notification settings - Fork 14.5k
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 Neo4jHook to get the query response #18007
Conversation
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)
|
Could you add/modify a test (or a couple of them since there are two branches) for this change? |
@minu7 I pushed ttests. Can you look at it? |
def test_neo4j_operator_test(self, mock_hook): | ||
|
||
sql = """ | ||
MATCH (tom {name: "Tom Hanks"}) RETURN tom | ||
""" | ||
op = Neo4jOperator(task_id='basic_neo4j', sql=sql, dag=self.dag) | ||
op.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This test took an unnecessarily long time - 6 seconds. We need a full run, but running the execute method is enough. This test also lacked assertion.
Sorry for the late reply, the tests look good, thanks 👍 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
One small nit otherwise lgtm.
The PR is likely OK to be merged with just subset of tests for default Python and Database versions without running the full matrix of tests, because it does not modify the core of Airflow. If the committers decide that the full tests matrix is needed, they will add the label 'full tests needed'. Then you should rebase to the latest main or amend the last commit of the PR, and push it with --force-with-lease. |
Co-authored-by: Tzu-ping Chung <[email protected]>
Awesome work, congrats on your first merged pull request! |
It allows to get the answer of the query from the Neo4jHook.
closes: #18005
related: #18005