ipdate

tests/unit/services/devii/tasks/context.py

@@ -0,0 +1,104 @@
+# 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