async def test_unpatch(tracer):
unpatch()
with aiobotocore_client("kinesis", tracer) as kinesis:
await kinesis.list_streams()
traces = tracer.pop_traces()
assert len(traces) == 0
def test_unpatch(self):
unpatch()
with aiobotocore_client('kinesis', self.tracer) as kinesis:
yield from kinesis.list_streams()
traces = self.tracer.writer.pop_traces()
self.assertEqual(len(traces), 0)
def test_unpatch(self):
unpatch()
with aiobotocore_client('kinesis', self.tracer) as kinesis:
yield from kinesis.list_streams()
traces = self.tracer.writer.pop_traces()
eq_(len(traces), 0)
def test_unpatch(self):
kinesis = self.session.create_client('kinesis', region_name='us-east-1', endpoint_url=MOTO_ENDPOINT_URL)
try:
tracer = get_dummy_tracer()
writer = tracer.writer
Pin(service=self.TEST_SERVICE, tracer=tracer).onto(kinesis)
unpatch()
yield from kinesis.list_streams()
spans = writer.pop()
assert not spans, spans
finally:
kinesis.close()
async def test_user_specified_service(tracer):
"""
When a service name is specified by the user
The aiobotocore integration should use it as the service name
"""
with override_global_config(dict(service="mysvc")):
# Repatch to take config into account
unpatch()
patch()
with aiobotocore_client("ec2", tracer) as ec2:
await ec2.describe_instances()
traces = tracer.pop_traces()
assert len(traces) == 1
assert len(traces[0]) == 1
span = traces[0][0]
assert span.service == "mysvc"
def patch_aiobotocore():
patch()
yield
unpatch()
def tearDown(self):
super(AIOBotocoreTest, self).tearDown()
unpatch()
self.tracer = None
def tearDown(self):
unpatch()
self.session = None
del os.environ['AWS_ACCESS_KEY_ID']
del os.environ['AWS_SECRET_ACCESS_KEY']
def tearDown(self):
super(AIOBotocoreTest, self).tearDown()
unpatch()
self.tracer = None
