async def test_task_creation_patching(): # noqa: E999
var = ContextVar('test_task_creation_patching') # type: ContextVar[Union[str, int]]
var.set(12)
test_context = {'var': None, 'new_var': None} # type: Dict[str, Union[str, int, None]]
# noinspection PyCompatibility
async def coroutine():
test_context['var'] = var.get('default')
var.set(15)
test_context['new_var'] = var.get('default')
loop = asyncio.get_event_loop()
task = loop.create_task(coroutine())
assert var._cv_variable is not None
if sys.version_info < (3, 7):
assert var._cv_variable in task.context
assert task.context[var._cv_variable] == 12
await task
assert test_context['var'] == 12
assert test_context['new_var'] == 15
assert var.get(12)
async def test_task_creation_patching(): # noqa: E999
var = ContextVar('test_task_creation_patching')
var.set(12)
test_context = {'var': None, 'new_var': None}
# noinspection PyCompatibility
async def coroutine():
test_context['var'] = var.get('default')
var.set(15)
test_context['new_var'] = var.get('default')
loop = asyncio.get_event_loop()
task = loop.create_task(coroutine())
if sys.version_info < (3, 7):
# noinspection PyUnresolvedReferences
assert var.variable in task.context
# noinspection PyUnresolvedReferences
assert task.context[var.variable] == 12
await task
assert test_context['var'] == 12
assert test_context['new_var'] == 15
assert var.get(12)
def test_one_thread():
var = ContextVar('test_one_thread1')
with pytest.raises(LookupError) as error_context:
var.get()
if six.PY2:
assert error_context.value.args[0] is var
assert "pysoa.common.compatibility.ContextVar name='test_one_thread1' at " in repr(var)
assert isinstance(var.variable, getattr(threading, 'local'))
else:
assert "ContextVar name='test_one_thread1' at " in repr(var)
assert isinstance(var.variable, contextvars.ContextVar)
assert var.get('default1') == 'default1'
assert var.get(default='default2') == 'default2'
var.set('set1')
assert var.get() == 'set1'
var = ContextVar('test_one_thread2', 'default3')
assert var.get() == 'default3'
assert var.get('default4') == 'default4'
var.set('set2')
assert var.get() == 'set2'
var = ContextVar('test_one_thread3', default='default5')
assert var.get() == 'default5'
assert var.get('default6') == 'default6'
var.set('set3')
assert var.get() == 'set3'
def test_one_thread():
var = ContextVar('test_one_thread1') # type: ContextVar[six.text_type]
assert "<ContextVar name='test_one_thread1' at " in repr(var)
assert 'default=' not in repr(var)
with pytest.raises(LookupError) as error_context:
var.get()
if six.PY2:
assert error_context.value.args[0] is var
assert var._tl_variable is not None
assert isinstance(var._tl_variable, getattr(threading, 'local'))
else:
assert var._cv_variable is not None
assert contextvars is not None
assert isinstance(var._cv_variable, contextvars.ContextVar)
assert var.get('default1') == 'default1'
assert var.get(default='default2') == 'default2'
var.set('set1')
assert var.get() == 'set1'
var = ContextVar('test_one_thread2', 'default3')
if six.PY2:
assert "<ContextVar name='test_one_thread2' default=u'default3' at " in repr(
var)
else:
assert "<ContextVar name='test_one_thread2' default='default3' at " in repr(
var)
assert var.get() == 'default3'
assert var.get('default4') == 'default4'
var.set('set2')
assert var.get() == 'set2'
var = ContextVar('test_one_thread3', default='default5')
assert var.get() == 'default5'
assert var.get('default6') == 'default6'
var.set('set3')
assert var.get() == 'set3'
async def test_multiple_coroutines_async():
var1 = ContextVar('test_multiple_threads1') # type: ContextVar[str]
var2 = ContextVar('test_multiple_threads1') # type: ContextVar[str]
test_context = {
'c1_called': False,
'c1_complete': False,
'c2_called': False,
'c2_complete': False,
'var1_thread1_start': None,
'var2_thread1_start': None,
'var1_thread1_mid': None,
'var2_thread1_mid': None,
'var1_thread1_end': None,
'var2_thread1_end': None,
'var1_thread2_start': None,
'var2_thread2_start': None,
'var1_thread2_mid': None,
'var2_thread2_mid': None,
'var1_thread2_end': None,
'var2_thread2_end': None,
} # type: Dict[str, Union[bool, Optional[str]]]
# noinspection PyCompatibility
async def c1():
test_context['c1_called'] = True
test_context['var1_thread1_start'] = var1.get('default1_thread1')
test_context['var2_thread1_start'] = var2.get('default2_thread1')
var1.set('value1')
var2.set('value2')
await asyncio.sleep(0.1)
test_context['var1_thread1_mid'] = var1.get()
test_context['var2_thread1_mid'] = var2.get()
await asyncio.sleep(0.3)
test_context['var1_thread1_end'] = var1.get()
test_context['var2_thread1_end'] = var2.get()
test_context['c1_complete'] = True
# noinspection PyCompatibility
async def c2():
test_context['c2_called'] = True
test_context['var1_thread2_start'] = var1.get('default1_thread2')
test_context['var2_thread2_start'] = var2.get('default2_thread2')
var1.set('value3')
var2.set('value4')
await asyncio.sleep(0.3)
test_context['var1_thread2_mid'] = var1.get()
test_context['var2_thread2_mid'] = var2.get()
await asyncio.sleep(0.1)
test_context['var1_thread2_end'] = var1.get()
test_context['var2_thread2_end'] = var2.get()
test_context['c2_complete'] = True
loop_thread = SimpleLoopThread()
loop_thread.start()
var1.set('super1')
var2.set('super2')
asyncio.run_coroutine_threadsafe(c1(), loop_thread.loop)
asyncio.run_coroutine_threadsafe(c2(), loop_thread.loop)
# It's _extremely_ unlikely that _both_ have already started and completed, and we want to ensure that our patched
# run_coroutine_threadsafe hasn't run them both to completion before returning.
assert (
test_context['c1_called'] is False or
test_context['c1_complete'] is False or
test_context['c2_called'] is False or
test_context['c2_complete'] is False
)
try:
assert var1.get('default1_main') == 'super1'
assert var2.get('default2_main') == 'super2'
var1.set('value5')
var2.set('value6')
await asyncio.sleep(0.2)
assert var1.get() == 'value5'
assert var2.get() == 'value6'
await asyncio.sleep(0.2)
assert var1.get() == 'value5'
assert var2.get() == 'value6'
finally:
loop_thread.join()
assert test_context['c1_called'] is True
assert test_context['c1_complete'] is True
assert test_context['var1_thread1_start'] == 'super1'
assert test_context['var2_thread1_start'] == 'super2'
assert test_context['var1_thread1_mid'] == 'value1'
assert test_context['var2_thread1_mid'] == 'value2'
assert test_context['var1_thread1_end'] == 'value1'
assert test_context['var2_thread1_end'] == 'value2'
assert test_context['c2_called'] is True
assert test_context['c2_complete'] is True
assert test_context['var1_thread2_start'] == 'super1'
assert test_context['var2_thread2_start'] == 'super2'
assert test_context['var1_thread2_mid'] == 'value3'
assert test_context['var2_thread2_mid'] == 'value4'
assert test_context['var1_thread2_end'] == 'value3'
assert test_context['var2_thread2_end'] == 'value4'
def test_reset_tokens(with_context_var):
if with_context_var:
context = _fake_context_manager() # type: ignore
else:
context = mock.patch('pysoa.common.compatibility.contextvars',
new=None) # type: ignore
with context:
var1 = ContextVar('test_reset_tokens1',
default='foo') # type: ContextVar[six.text_type]
var2 = ContextVar('test_reset_tokens2',
default='bar') # type: ContextVar[six.text_type]
var3 = ContextVar(
'test_reset_tokens3') # type: ContextVar[six.text_type]
token1 = var1.set('hello')
token2 = var2.set('goodbye')
assert var1.get() == 'hello'
assert var2.get() == 'goodbye'
with pytest.raises(ValueError):
var1.reset(token2)
with pytest.raises(ValueError):
var2.reset(token1)
assert var1.get() == 'hello'
assert var2.get() == 'goodbye'
if not with_context_var:
bad_token1 = _ContextVarToken(var1, None) # type: ignore
bad_token2 = _ContextVarToken(var2, None) # type: ignore
else:
bad_token1 = _ThreadLocalToken(var1, None) # type: ignore
bad_token2 = _ThreadLocalToken(var2, None) # type: ignore
with pytest.raises(TypeError):
var1.reset(bad_token1)
with pytest.raises(TypeError):
var2.reset(bad_token2)
assert var1.get() == 'hello'
assert var2.get() == 'goodbye'
var1.reset(token1)
assert var1.get() == 'foo'
assert var2.get() == 'goodbye'
var2.reset(token2)
assert var1.get() == 'foo'
assert var2.get() == 'bar'
token1a = var1.set('hello')
token2a = var2.set('goodbye')
assert var1.get() == 'hello'
assert var2.get() == 'goodbye'
token1b = var1.set('world')
token2b = var2.set('universe')
assert var1.get() == 'world'
assert var2.get() == 'universe'
var2.reset(token2b)
assert var1.get() == 'world'
assert var2.get() == 'goodbye'
var1.reset(token1b)
assert var1.get() == 'hello'
assert var2.get() == 'goodbye'
var1.reset(token1a)
var2.reset(token2a)
assert var1.get() == 'foo'
assert var2.get() == 'bar'
token3 = var3.set('baz')
assert var3.get() == 'baz'
var3.reset(token3)
assert var3.get(default='qux') == 'qux'
with pytest.raises(LookupError):
var3.get()
def test_multiple_threads():
var1 = ContextVar(
'test_multiple_threads1') # type: ContextVar[six.text_type]
var2 = ContextVar(
'test_multiple_threads1') # type: ContextVar[six.text_type]
test_context = {
'var1_thread1_start': None,
'var2_thread1_start': None,
'var1_thread1_mid': None,
'var2_thread1_mid': None,
'var1_thread1_end': None,
'var2_thread1_end': None,
'var1_thread2_start': None,
'var2_thread2_start': None,
'var1_thread2_mid': None,
'var2_thread2_mid': None,
'var1_thread2_end': None,
'var2_thread2_end': None,
} # type: Dict[six.text_type, Optional[six.text_type]]
def t1():
test_context['var1_thread1_start'] = var1.get('default1_thread1')
test_context['var2_thread1_start'] = var2.get('default2_thread1')
var1.set('value1')
var2.set('value2')
time.sleep(0.1)
test_context['var1_thread1_mid'] = var1.get()
test_context['var2_thread1_mid'] = var2.get()
time.sleep(0.3)
test_context['var1_thread1_end'] = var1.get()
test_context['var2_thread1_end'] = var2.get()
def t2():
test_context['var1_thread2_start'] = var1.get('default1_thread2')
test_context['var2_thread2_start'] = var2.get('default2_thread2')
var1.set('value3')
var2.set('value4')
time.sleep(0.3)
test_context['var1_thread2_mid'] = var1.get()
test_context['var2_thread2_mid'] = var2.get()
time.sleep(0.1)
test_context['var1_thread2_end'] = var1.get()
test_context['var2_thread2_end'] = var2.get()
thread1 = threading.Thread(target=t1)
thread2 = threading.Thread(target=t2)
thread1.start()
thread2.start()
try:
assert var1.get('default1_main') == 'default1_main'
assert var2.get('default2_main') == 'default2_main'
var1.set('value5')
var2.set('value6')
time.sleep(0.2)
assert var1.get() == 'value5'
assert var2.get() == 'value6'
time.sleep(0.2)
assert var1.get() == 'value5'
assert var2.get() == 'value6'
finally:
thread1.join()
thread2.join()
assert test_context['var1_thread1_start'] == 'default1_thread1'
assert test_context['var2_thread1_start'] == 'default2_thread1'
assert test_context['var1_thread1_mid'] == 'value1'
assert test_context['var2_thread1_mid'] == 'value2'
assert test_context['var1_thread1_end'] == 'value1'
assert test_context['var2_thread1_end'] == 'value2'
assert test_context['var1_thread2_start'] == 'default1_thread2'
assert test_context['var2_thread2_start'] == 'default2_thread2'
assert test_context['var1_thread2_mid'] == 'value3'
assert test_context['var2_thread2_mid'] == 'value4'
assert test_context['var1_thread2_end'] == 'value3'
assert test_context['var2_thread2_end'] == 'value4'