|
# retoor <retoor@molodetz.nl>
|
|
|
|
import asyncio
|
|
|
|
from devplacepy.database import invalidate_admins_cache
|
|
from devplacepy.services.devii.tasks.context import in_task_run, task_run_scope
|
|
from devplacepy.services.devii.tasks.guards import nesting_allowed
|
|
from devplacepy.services.devii.tasks.schedule import now_utc, to_iso
|
|
from devplacepy.utils import generate_uid
|
|
from tests.conftest import run_async
|
|
|
|
|
|
def _account(local_db, role):
|
|
uid = generate_uid()
|
|
local_db["users"].insert(
|
|
{
|
|
"uid": uid,
|
|
"username": f"ctx-{uid[-10:]}",
|
|
"role": role,
|
|
"deleted_at": None,
|
|
"created_at": to_iso(now_utc()),
|
|
}
|
|
)
|
|
invalidate_admins_cache()
|
|
return uid
|
|
|
|
|
|
def test_outside_a_task_run_the_flag_is_off():
|
|
assert in_task_run() is False
|
|
|
|
|
|
def test_scope_sets_and_restores_the_flag():
|
|
assert in_task_run() is False
|
|
with task_run_scope():
|
|
assert in_task_run() is True
|
|
assert in_task_run() is False
|
|
|
|
|
|
def test_scope_restores_the_flag_after_an_exception():
|
|
try:
|
|
with task_run_scope():
|
|
raise RuntimeError("boom")
|
|
except RuntimeError:
|
|
pass
|
|
assert in_task_run() is False
|
|
|
|
|
|
def test_nested_scopes_restore_correctly():
|
|
with task_run_scope():
|
|
with task_run_scope():
|
|
assert in_task_run() is True
|
|
assert in_task_run() is True
|
|
assert in_task_run() is False
|
|
|
|
|
|
def test_the_flag_reaches_work_spawned_inside_the_run():
|
|
seen = []
|
|
|
|
async def inner():
|
|
seen.append(in_task_run())
|
|
|
|
async def run():
|
|
with task_run_scope():
|
|
await inner()
|
|
task = asyncio.create_task(inner())
|
|
await task
|
|
|
|
run_async(run())
|
|
assert seen == [True, True]
|
|
|
|
|
|
def test_a_concurrent_turn_outside_the_run_never_sees_the_flag():
|
|
seen = {}
|
|
|
|
async def scheduled(gate):
|
|
with task_run_scope():
|
|
seen["inside"] = in_task_run()
|
|
gate.set()
|
|
await asyncio.sleep(0.05)
|
|
seen["inside_after"] = in_task_run()
|
|
|
|
async def interactive(gate):
|
|
await gate.wait()
|
|
seen["outside"] = in_task_run()
|
|
|
|
async def run():
|
|
gate = asyncio.Event()
|
|
await asyncio.gather(scheduled(gate), interactive(gate))
|
|
|
|
run_async(run())
|
|
assert seen["inside"] is True
|
|
assert seen["inside_after"] is True
|
|
assert seen["outside"] is False
|
|
|
|
|
|
def test_members_may_not_nest_but_administrators_may(local_db):
|
|
member = _account(local_db, "Member")
|
|
admin = _account(local_db, "Admin")
|
|
assert nesting_allowed(member) is True
|
|
assert nesting_allowed(admin) is True
|
|
with task_run_scope():
|
|
assert nesting_allowed(member) is False
|
|
assert nesting_allowed(admin) is True
|
|
assert nesting_allowed(member) is True
|