def test_patch_unpatch(self):
# Test patch idempotence
patch()
patch()
es = elasticsearch.Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
Pin(service=self.TEST_SERVICE, tracer=self.tracer).onto(es.transport)
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = self.get_spans()
self.reset()
assert spans, spans
assert len(spans) == 1
# Test unpatch
self.reset()
unpatch()
es = elasticsearch.Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = self.get_spans()
self.reset()
assert not spans, spans
# Test patch again
self.reset()
patch()
es = elasticsearch.Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
Pin(service=self.TEST_SERVICE, tracer=self.tracer).onto(es.transport)
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = self.get_spans()
self.reset()
assert spans, spans
assert len(spans) == 1
def test_patch_unpatch(self):
tracer = get_dummy_tracer()
writer = tracer.writer
# Test patch idempotence
patch()
patch()
es = Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
Pin(service=self.TEST_SERVICE, tracer=tracer).onto(es.transport)
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
# Test unpatch
unpatch()
es = Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = writer.pop()
assert not spans, spans
# Test patch again
patch()
es = Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
Pin(service=self.TEST_SERVICE, tracer=tracer).onto(es.transport)
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
def test_patch_unpatch(self):
tracer = get_dummy_tracer()
writer = tracer.writer
# Test patch idempotence
patch()
patch()
es = Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
Pin(service=self.TEST_SERVICE, tracer=tracer).onto(es.transport)
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
# Test unpatch
unpatch()
es = Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = writer.pop()
assert not spans, spans
# Test patch again
patch()
es = Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
Pin(service=self.TEST_SERVICE, tracer=tracer).onto(es.transport)
# Test index creation
es.indices.create(index=self.ES_INDEX, ignore=400)
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
def tearDown(self):
"""Clean ES"""
unpatch()
es = Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
es.indices.delete(index=self.ES_INDEX, ignore=[400, 404])
def tearDown(self):
"""Clean ES"""
super(ElasticsearchPatchTest, self).tearDown()
unpatch()
self.es.indices.delete(index=self.ES_INDEX, ignore=[400, 404])
def tearDown(self):
"""Clean ES"""
unpatch()
es = Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
es.indices.delete(index=self.ES_INDEX, ignore=[400, 404])
