def test_get_call_context(self):
# it should return a context even if not attached to the Task
ctx = self.tracer.get_call_context()
assert ctx is not None
# test that it behaves the wrong way
task = asyncio_current_task()
assert task
task_ctx = getattr(task, "__datadog_context", None)
assert task_ctx is None
async def test_get_call_context(tracer):
tracer.configure(context_provider=DefaultContextProvider())
ctx = tracer.current_trace_context()
assert ctx is None
# test that it behaves the wrong way
task = asyncio_current_task()
assert task
task_ctx = getattr(task, "__datadog_context", None)
assert task_ctx is None
def test_context_task_none(self):
# it should handle the case where a Task is not available
# Note: the @mark_asyncio is missing to simulate an execution
# without a Task
task = asyncio_current_task()
# the task is not available
assert task is None
# but a new Context is still created making the operation safe
ctx = self.tracer.get_call_context()
assert ctx is not None
async def test_get_call_context(tracer):
tracer.configure(context_provider=DefaultContextProvider())
# it should return a context even if not attached to the Task
ctx = tracer.get_call_context()
assert ctx is not None
# test that it behaves the wrong way
task = asyncio_current_task()
assert task
task_ctx = getattr(task, "__datadog_context", None)
assert task_ctx is None
def test_get_call_context(self):
# it should return the context attached to the current Task
# or create a new one
task = asyncio_current_task()
ctx = getattr(task, "__datadog_context", None)
assert ctx is None
# get the context from the loop creates a new one that
# is attached to the Task object
ctx = self.tracer.get_call_context()
assert ctx == getattr(task, "__datadog_context", None)
def test_context_task_none(tracer):
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
# it should handle the case where a Task is not available
# Note: the @pytest.mark.asyncio is missing to simulate an execution
# without a Task
task = asyncio_current_task()
# the task is not available
assert task is None
ctx = tracer.current_trace_context()
assert ctx is None
