44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
import asyncio
|
|
from uuid import UUID
|
|
import pytest
|
|
from unittest.mock import MagicMock
|
|
|
|
from app.core.tenant_context import (
|
|
set_current_account_id,
|
|
clear_current_account_id,
|
|
get_current_account_id,
|
|
)
|
|
|
|
|
|
def test_contextvar_is_none_by_default():
|
|
assert get_current_account_id() is None
|
|
|
|
|
|
def test_set_and_clear():
|
|
account_id = UUID("aaaaaaaa-0000-0000-0000-000000000001")
|
|
token = set_current_account_id(account_id)
|
|
assert get_current_account_id() == str(account_id)
|
|
clear_current_account_id(token)
|
|
assert get_current_account_id() is None
|
|
|
|
|
|
def test_tasks_are_isolated():
|
|
"""Each asyncio task has its own ContextVar value."""
|
|
results = {}
|
|
|
|
async def set_in_task(name: str, value: str):
|
|
token = set_current_account_id(UUID(value))
|
|
await asyncio.sleep(0)
|
|
results[name] = get_current_account_id()
|
|
clear_current_account_id(token)
|
|
|
|
async def run():
|
|
await asyncio.gather(
|
|
set_in_task("a", "aaaaaaaa-0000-0000-0000-000000000001"),
|
|
set_in_task("b", "bbbbbbbb-0000-0000-0000-000000000002"),
|
|
)
|
|
|
|
asyncio.run(run())
|
|
assert results["a"] == "aaaaaaaa-0000-0000-0000-000000000001"
|
|
assert results["b"] == "bbbbbbbb-0000-0000-0000-000000000002"
|