code: def test_flight_cancelled_error(ws):
      """Test flight -> cancelled -> error transition loop.
      This can be caused by an issue while (un)pickling or a bug in the network stack.
  
      See https://github.com/dask/distributed/issues/6877
      """
      ws2 = "127.0.0.1:2"
      instructions = ws.handle_stimulus(
          ComputeTaskEvent.dummy(
              "y", who_has={"x": [ws2]}, stimulus_id="s1"
          ),
          FreeKeysEvent(keys=["y", "x"], stimulus_id="s2"),
          GatherDepFailureEvent.from_exception(
              Exception(), worker=ws2, total_nbytes=1, stimulus_id="s3"
          ),
      )
      assert instructions == [
          GatherDep(
              worker=ws2,
              to_gather={"x"},
              total_nbytes=1,
              stimulus_id="s1",
          )
      ]
      assert not ws.tasks
  
imports: from __future__ import annotations
  import asyncio
  import pytest
  import distributed
  from distributed import Event, Lock, Worker
  from distributed.client import wait
  from distributed.utils_test import BlockedExecute, BlockedGatherDep, BlockedGetData, _LockedCommPool, assert_story, async_poll_for, freeze_batched_send, gen_cluster, inc, lock_inc, wait_for_state, wait_for_stimulus
  from distributed.worker_state_machine import AddKeysMsg, ComputeTaskEvent, Execute, ExecuteFailureEvent, ExecuteSuccessEvent, FreeKeysEvent, GatherDep, GatherDepFailureEvent, GatherDepNetworkFailureEvent, GatherDepSuccessEvent, LongRunningMsg, ReleaseWorkerDataMsg, RemoveReplicasEvent, RescheduleEvent, SecedeEvent, TaskFinishedMsg, UpdateDataEvent
label_desc: Use image node_id 1 for calling a global function with example usage: test_flight_cancelled_error(ws) without return types