Stabilize flaky Windows regression tests

Author: peter941221

distributed/tests/test_batched.py

@@ -172,7 +172,7 @@ async def send():
 
         async def recv():
             while True:
-                result = await wait_for(comm.read(), 1)
+                result = await wait_for(comm.read(), 5)
                 L.extend(result)
                 if result[-1] == 9999:
                     break

distributed/tests/test_scheduler.py

@@ -3642,6 +3642,7 @@ def print_memory_info(msg: str) -> None:
     # as fast as on Linux. Note that this behaviour is heavily impacted by OS tweaks,
     # meaning that what you observe on your local host may behave differently on CI.
     if not LINUX:
+        await asyncio.gather(*(n.close() for n in nannies))
         return
 
     print_memory_info("Before clearing memory leak")

distributed/tests/test_worker.py

@@ -3017,6 +3017,8 @@ async def test_log_remove_worker(c, s, a, b):
             if "stimulus_id" in ev:  # Strip timestamp
                 ev["stimulus_id"] = ev["stimulus_id"].rsplit("-", 1)[0]
 
+    worker_get_client_events = events.pop("worker-get-client")
+
     assert events == {
         a.address: [
             {
@@ -3091,9 +3093,10 @@ async def test_log_remove_worker(c, s, a, b):
                 "worker": b.address,
             },
         ],
-        "worker-get-client": [{"client": c.id, "timeout": 5, "worker": b.address}],
     }
 
+    assert {"client": c.id, "timeout": 5, "worker": b.address} in worker_get_client_events
+
 
 @gen_cluster(client=True)
 async def test_task_flight_compute_oserror(c, s, a, b):