Simplify

Author: dandavison

tests/nexus/test_worker.py

@@ -95,29 +95,17 @@ async def op(
 
 
 class Barrier:
-    """Minimal implementation of asyncio.Barrier for Python 3.9+ compatibility.
-
-    This is a simplified version that only implements the wait() method needed
-    for this test. All tasks block until exactly 'parties' tasks have called
-    wait(), then all are released simultaneously.
-    """
-
     def __init__(self, parties: int):
         """Create a barrier for 'parties' tasks."""
         if parties < 1:
             raise ValueError("parties must be > 0")
         self._parties = parties
         self._count = 0
-        self._lock = asyncio.Lock()
         self._event = asyncio.Event()
 
     async def wait(self) -> None:
         """Wait for all parties to reach the barrier."""
-        async with self._lock:
-            self._count += 1
-            if self._count == self._parties:
-                # Last one in, release everyone
-                self._event.set()
-
-        # Wait for all parties to arrive
+        self._count += 1
+        if self._count == self._parties:
+            self._event.set()
         await self._event.wait()