def test_sentinel_exited_complete_root_exception():
"""
This test forces a transaction to exit while it still has an active trace
this causes an exception to be raised in TraceCache complete_root(). It
verifies that the sentinel.exited property is set to true if an exception
is raised in complete_root()
"""
expected_error = "not the current trace"
try:
txn = None
sentinel = None
txn = BackgroundTask(application_instance(), "Parent")
txn.__enter__()
sentinel = txn.root_span
trace = FunctionTrace("trace")
trace.__enter__()
txn.__exit__(None, None, None)
assert False, "Did not raise exception"
except RuntimeError as e:
assert str(e) == expected_error
finally:
assert sentinel.exited
# Make sure to exit properly so cleanup is performed
trace.__exit__(None, None, None)
txn.__exit__(None, None, None)
def test_dead_transaction_ends(circular):
if circular and six.PY2:
pytest.skip("Circular references in py2 result in a memory leak. "
"There is no way to remove transactions from the weakref "
"cache in this case.")
transaction = BackgroundTask(application_instance(),
"test_dead_transaction_ends")
if circular:
transaction._self = transaction
transaction.__enter__()
del transaction
gc.collect()
def test_sentinel_exited_complete_root_exception():
"""
This test forces a transaction to exit while it still has an active trace
this causes an exception to be raised in TraceCache complete_root(). It
verifies that the sentinel.exited property is set to true if an exception
is raised in complete_root(), and that the exception is caught.
"""
txn = None
sentinel = None
txn = BackgroundTask(application_instance(), "Parent")
txn.__enter__()
sentinel = txn.root_span
trace = FunctionTrace("trace")
trace.__enter__()
txn.__exit__(None, None, None)
assert sentinel.exited
# Make sure to exit properly so cleanup is performed
trace.__exit__(None, None, None)
txn.__exit__(None, None, None)