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']