Skip to content

Commit 4ef6ba8

Browse files
fix: add "wait"s before changing token length & releasing limiter
1 parent bf1f270 commit 4ef6ba8

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/trio/_tests/test_sync.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,17 +242,20 @@ async def test_CapacityLimiter_zero_limit_tokens() -> None:
242242
await c.acquire_on_behalf_of(0) # total_tokens is 1
243243

244244
nursery.start_soon(c.acquire_on_behalf_of, 1)
245+
await wait_all_tasks_blocked()
245246
c.total_tokens = 0
246247

247248
assert c.statistics().borrowers == [0]
248249

249250
c.release_on_behalf_of(0)
251+
await wait_all_tasks_blocked()
250252
assert c.statistics().borrowers == []
251-
assert c.statistics().tasks_waiting == 0
253+
assert c.statistics().tasks_waiting == 1
252254

253255
c.total_tokens = 1
254256
await wait_all_tasks_blocked()
255257
assert c.statistics().borrowers == [1]
258+
assert c.statistics().tasks_waiting == 0
256259

257260
c.release_on_behalf_of(1)
258261

0 commit comments

Comments
 (0)