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()