houqp commented on a change in pull request #10917:
URL: https://github.com/apache/airflow/pull/10917#discussion_r511694599



##########
File path: tests/jobs/test_local_task_job.py
##########
@@ -438,5 +521,5 @@ def test_number_of_queries_single_loop(self, 
mock_get_task_runner, return_codes)
         mock_get_task_runner.return_value.return_code.side_effects = 
return_codes
 
         job = LocalTaskJob(task_instance=ti, executor=MockExecutor())
-        with assert_queries_count(12):
+        with assert_queries_count(13):

Review comment:
       added a new DB query in local_task_job to query task state after task 
finished execution so we can decide to which callback to invoke




----------------------------------------------------------------
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.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to