close client in sync test_actor tests

graingert · graingert · commit 411dc7957d36 · 2022-05-26T13:50:59.000+01:00
diff --git a/distributed/tests/test_actor.py b/distributed/tests/test_actor.py
@@ -635,8 +635,7 @@ class UsesCounter:
         def do_inc(self, ac):
             return ac.increment().result()
 
-    with cluster(nworkers=1) as (cl, _):
-        client = Client(cl["address"])
+    with cluster(nworkers=1) as (cl, _), Client(cl["address"]) as client:
         ac = client.submit(Counter, actor=True).result()
         ac2 = client.submit(UsesCounter, actor=True).result()
 
@@ -652,8 +651,7 @@ def do_inc(self, ac):
             # cannot expire
             return ac.increment().result(timeout=0.001)
 
-    with cluster(nworkers=1) as (cl, _):
-        client = Client(cl["address"])
+    with cluster(nworkers=1) as (cl, _), Client(cl["address"]) as client:
         ac = client.submit(Counter, actor=True).result()
         ac2 = client.submit(UsesCounter, actor=True).result()
 
@@ -667,8 +665,7 @@ class UsesCounter:
         def do_inc(self, ac):
             return get_client().sync(ac.increment)
 
-    with cluster(nworkers=1) as (cl, _):
-        client = Client(cl["address"])
+    with cluster(nworkers=1) as (cl, _), Client(cl["address"]) as client:
         ac = client.submit(Counter, actor=True).result()
         ac2 = client.submit(UsesCounter, actor=True).result()
 
@@ -701,8 +698,7 @@ def method(self):
         def prop(self):
             raise MyException
 
-    with cluster(nworkers=2) as (cl, w):
-        client = Client(cl["address"])
+    with cluster(nworkers=2) as (cl, w), Client(cl["address"]) as client:
         ac = client.submit(Broken, actor=True).result()
         acfut = ac.method()
         with pytest.raises(MyException):
