def test_double_patch(self):
patch()
with aiobotocore_client('sqs', self.tracer) as sqs:
yield from sqs.list_queues()
traces = self.tracer.writer.pop_traces()
eq_(len(traces), 1)
eq_(len(traces[0]), 1)
async def test_double_patch(tracer):
patch()
with aiobotocore_client("sqs", tracer) as sqs:
await sqs.list_queues()
traces = tracer.pop_traces()
assert len(traces) == 1
assert len(traces[0]) == 1
def test_double_patch(self):
patch()
with aiobotocore_client('sqs', self.tracer) as sqs:
yield from sqs.list_queues()
traces = self.tracer.writer.pop_traces()
self.assertEqual(len(traces), 1)
self.assertEqual(len(traces[0]), 1)
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 test_double_patch(self):
sqs = self.session.create_client('sqs', 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(sqs)
patch()
patch()
yield from sqs.list_queues()
spans = writer.pop()
assert spans
eq_(len(spans), 1)
finally:
sqs.close()
def patch_aiobotocore():
patch()
yield
unpatch()
def setUp(self):
super(AIOBotocoreTest, self).setUp()
patch()
self.tracer = get_dummy_tracer()
def setUp(self):
patch()
self.session = aiobotocore.session.get_session()
os.environ['AWS_ACCESS_KEY_ID'] = 'dummy'
os.environ['AWS_SECRET_ACCESS_KEY'] = 'dummy'
def setUp(self):
super(AIOBotocoreTest, self).setUp()
patch()
self.tracer = DummyTracer()
def setUp(self):
super(AIOBotocoreTest, self).setUp()
patch()
self.tracer = get_dummy_tracer()
