Push producer's task completion status to XCOM

[pankajkoti]

This PR pushes a task_status XCOM for the DbtProducerWatcherOperator task after its completion -- be it failed or succeeded.

The idea is that sensors will pull this XCOM while checking for the result of the model. If a sensor sees that the DbtProducerWatcherOperator task has completed (either succeeded or failed), but the DbtProducerWatcherOperator has not pushed the model status to XCOM—whether via the node_finished... XCOM or the run_results XCOM—it means the operator has already finished processing and will not push any further status updates.

In that case, the sensor should immediately mark the model as failed, since there is no chance of receiving a run status for it in the future, and the sensor task itself should also fail.

related: https://github.com/astronomer/oss-integrations-private/issues/239
related: #1960

[netlify]

✅ Deploy Preview for sunny-pastelito-5ecb04 canceled.

Name	Link
🔨 Latest commit	429aed2
🔍 Latest deploy log	https://app.netlify.com/projects/sunny-pastelito-5ecb04/deploys/68de387266b7e90008a34532

[Copilot]

Pull Request Overview

This PR adds task completion status tracking to the DbtProducerWatcherOperator by pushing a "completed" XCOM value after task execution. This helps downstream sensors determine when the producer task has finished processing and won't send any further updates.

• Adds try-catch wrapper around execute method to ensure completion status is always pushed
• Pushes "completed" status to XCOM in both success and failure scenarios
• Adds comprehensive test coverage for both execution paths

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 1 comment.

File	Description
cosmos/operators/watcher.py	Wraps execute method with try-catch to push "completed" status to XCOM in all scenarios
tests/operators/test_watcher.py	Adds test coverage for completion status XCOM functionality

---

_{Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.}

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 97.81%. Comparing base (ba9b869) to head (429aed2).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #2000   +/-   ##
=======================================
  Coverage   97.81%   97.81%           
=======================================
  Files          87       87           
  Lines        5526     5532    +6     
=======================================
+ Hits         5405     5411    +6     
  Misses        121      121           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[tatiana]

Looks great, thank you very much @pankajkoti . This is one of the few scenarios I think it justifies for us to use the broad exception.