open-telemetry
/
opentelemetry-python-contrib
mirror of https://github.com/open-telemetry/opentelemetry-python-contrib.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add uninstrument test for boto (#1488)

This commit is contained in:
Shalev Roda 2022-12-18 11:52:55 +02:00 committed by GitHub
parent 0f2451be26
commit 41592015a3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
instrumentation/opentelemetry-instrumentation-boto/tests/test_boto_instrumentation.py
View File

@ -210,6 +210,22 @@ class TestBotoInstrumentor(TestBase):
assert spans assert spans
self.assertEqual(len(spans), 1) self.assertEqual(len(spans), 1)
@mock_s3_deprecated
def test_uninstrument(self):
s3 = boto.s3.connect_to_region("us-east-1")
# Get the created bucket
s3.create_bucket("cheese")
spans = self.memory_exporter.get_finished_spans()
assert spans
self.assertEqual(len(spans), 1)
self.memory_exporter.clear()
BotoInstrumentor().uninstrument()
s3.get_bucket("cheese")
spans = self.memory_exporter.get_finished_spans()
self.assertEqual(len(spans), 0)
@mock_lambda_deprecated @mock_lambda_deprecated
def test_lambda_client(self): def test_lambda_client(self):
lamb = boto.awslambda.connect_to_region("us-east-2") lamb = boto.awslambda.connect_to_region("us-east-2")