From 3f91ad3554359f985dc4c8f0dbfc095fdf18dda5 Mon Sep 17 00:00:00 2001
From: Gabe Joseph <gjoseph92@gmail.com>
Date: Fri, 5 Aug 2022 16:03:43 -0600
Subject: [PATCH] Dump `has_what`, `missing_dep_flight` (#6830)

---
 distributed/tests/test_worker.py               | 2 ++
 distributed/tests/test_worker_state_machine.py | 2 ++
 distributed/worker_state_machine.py            | 2 ++
 3 files changed, 6 insertions(+)

diff --git a/distributed/tests/test_worker.py b/distributed/tests/test_worker.py
index 75985e8df7..3d7d2c0105 100644
--- a/distributed/tests/test_worker.py
+++ b/distributed/tests/test_worker.py
@@ -3327,9 +3327,11 @@ async def test_Worker__to_dict(c, s, a):
         "nthreads",
         "running",
         "ready",
+        "has_what",
         "constrained",
         "executing",
         "long_running",
+        "missing_dep_flight",
         "in_flight_tasks",
         "in_flight_workers",
         "busy_workers",
diff --git a/distributed/tests/test_worker_state_machine.py b/distributed/tests/test_worker_state_machine.py
index e0a46a79be..ae96835605 100644
--- a/distributed/tests/test_worker_state_machine.py
+++ b/distributed/tests/test_worker_state_machine.py
@@ -155,6 +155,7 @@ def test_WorkerState__to_dict(ws):
         "data": {"y": None},
         "data_needed": {},
         "executing": [],
+        "has_what": {"127.0.0.1:1235": ["x"]},
         "in_flight_tasks": ["x"],
         "in_flight_workers": {"127.0.0.1:1235": ["x"]},
         "log": [
@@ -166,6 +167,7 @@ def test_WorkerState__to_dict(ws):
             ["y", "receive-from-scatter", "s2"],
         ],
         "long_running": [],
+        "missing_dep_flight": [],
         "nthreads": 1,
         "ready": [],
         "running": True,
diff --git a/distributed/worker_state_machine.py b/distributed/worker_state_machine.py
index 62dbe84af1..7c0466e25a 100644
--- a/distributed/worker_state_machine.py
+++ b/distributed/worker_state_machine.py
@@ -3070,9 +3070,11 @@ def _to_dict(self, *, exclude: Container[str] = ()) -> dict:
                 for w, tss in self.data_needed.items()
             },
             "executing": {ts.key for ts in self.executing},
+            "has_what": dict(self.has_what),
             "long_running": {ts.key for ts in self.long_running},
             "in_flight_tasks": {ts.key for ts in self.in_flight_tasks},
             "in_flight_workers": self.in_flight_workers,
+            "missing_dep_flight": [ts.key for ts in self.missing_dep_flight],
             "busy_workers": self.busy_workers,
             "log": self.log,
             "stimulus_log": self.stimulus_log,