opensource/opentelemetry-python-contrib
1
0
Fork 0
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git synced 2025-07-31 22:23:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add uninstrument test for boto3sqs (#1550)

Browse Source
This commit is contained in:
Shalev Roda
2023-01-01 23:06:56 +02:00
committed by GitHub
parent 346160467f
commit 6d1c740eff
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
instrumentation/opentelemetry-instrumentation-boto3sqs/tests/test_boto3sqs_instrumentation.py
View File

@ -311,3 +311,27 @@ class TestBoto3SQSInstrumentation(TestBase):
self.assertEqual(attrs["span_id"], link.context.span_id) self.assertEqual(attrs["span_id"], link.context.span_id)
self.memory_exporter.clear() self.memory_exporter.clear()
def test_uninstrument(self):
mock_response = {
"MessageId": "123456789",
}
with self._mocked_endpoint(mock_response):
self._client.send_message(
QueueUrl=self._queue_url,
MessageBody="test",
)
spans = self.get_finished_spans()
self.assertEqual(1, len(spans))
self.memory_exporter.clear()
Boto3SQSInstrumentor().uninstrument()
self._client.send_message(
QueueUrl=self._queue_url,
MessageBody="test",
)
spans = self.get_finished_spans()
self.assertEqual(0, len(spans))