mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-09-05 02:24:21 -05:00
Kyle AltendorfandGitHub
c63324abe9
* remove event_loop() fixtures * flake8 * flake8 * remove sys.exit() from daemon shutdown * bump full node test timeout. a lot... to see. * fixup some tests * back to module scope event loop fixture for test_full_node.py * Update test_full_node.py * Iterator... * for the whole directory * some fixtures back to module scope for reduced runtime * back to 40 minute workflow timeouts * these are being addressed separately
76 lines
2.0 KiB
Python
76 lines
2.0 KiB
Python
import asyncio
|
|
import logging
|
|
import time
|
|
from asyncio import CancelledError
|
|
|
|
import pytest
|
|
|
|
from chia.full_node.lock_queue import LockQueue, LockClient
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class TestLockQueue:
|
|
@pytest.mark.asyncio
|
|
async def test_lock_queue(self):
|
|
lock = asyncio.Lock()
|
|
|
|
queue = LockQueue(lock)
|
|
|
|
low_priority_client = LockClient(1, queue)
|
|
high_priority_client = LockClient(0, queue)
|
|
|
|
async def very_slow_func():
|
|
await asyncio.sleep(2)
|
|
raise CancelledError()
|
|
|
|
async def slow_func():
|
|
for i in range(100):
|
|
await asyncio.sleep(0.01)
|
|
|
|
async def kind_of_slow_func():
|
|
for i in range(100):
|
|
await asyncio.sleep(0.001)
|
|
|
|
async def do_high():
|
|
nonlocal high_priority_client
|
|
for i in range(10):
|
|
log.warning("Starting high")
|
|
t1 = time.time()
|
|
async with high_priority_client:
|
|
log.warning(f"Spend {time.time() - t1} waiting for high")
|
|
await slow_func()
|
|
|
|
async def do_low(i: int):
|
|
nonlocal low_priority_client
|
|
log.warning(f"Starting low {i}")
|
|
t1 = time.time()
|
|
async with low_priority_client:
|
|
log.warning(f"Spend {time.time() - t1} waiting for low {i}")
|
|
await kind_of_slow_func()
|
|
|
|
h = asyncio.create_task(do_high())
|
|
l_tasks = []
|
|
for i in range(50):
|
|
l_tasks.append(asyncio.create_task(do_low(i)))
|
|
|
|
winner = None
|
|
|
|
while True:
|
|
if h.done():
|
|
if winner is None:
|
|
winner = "h"
|
|
|
|
l_finished = True
|
|
for t in l_tasks:
|
|
if not t.done():
|
|
l_finished = False
|
|
if l_finished and winner is None:
|
|
winner = "l"
|
|
if l_finished and h.done():
|
|
break
|
|
await asyncio.sleep(1)
|
|
assert winner == "h"
|
|
|
|
queue.close()
|