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 setUp(self):
"""Prepare ES"""
super(ElasticsearchPatchTest, self).setUp()
es = elasticsearch.Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
Pin(service=self.TEST_SERVICE, tracer=self.tracer).onto(es.transport)
mapping = {
'mapping': {
'properties': {
'created': {
'type': 'date',
'format': 'yyyy-MM-dd'
}
}
}
}
es.indices.create(index=self.ES_INDEX, ignore=400, body=mapping)
patch()
self.es = es
def test_elasticsearch(self):
"""Test the elasticsearch integration
All in this for now. Will split it later.
"""
tracer = get_dummy_tracer()
writer = tracer.writer
transport_class = get_traced_transport(
opentelemetry_tracer=tracer,
opentelemetry_service=self.TEST_SERVICE)
es = elasticsearch.Elasticsearch(transport_class=transport_class,
port=ELASTICSEARCH_CONFIG['port'])
# Test index creation
mapping = {
'mapping': {
'properties': {
'created': {
'type': 'date',
'format': 'yyyy-MM-dd'
}
}
}
}
es.indices.create(index=self.ES_INDEX, ignore=400, body=mapping)
spans = writer.pop()
assert spans
assert len(spans) == 1
span = spans[0]
assert span.service == self.TEST_SERVICE
assert span.name == 'elasticsearch.query'
assert span.span_type == 'elasticsearch'
assert span.error == 0
assert span.get_tag('elasticsearch.method') == 'PUT'
assert span.get_tag('elasticsearch.url') == '/%s' % self.ES_INDEX
assert span.resource == 'PUT /%s' % self.ES_INDEX
# Put data
args = {'index': self.ES_INDEX, 'doc_type': self.ES_TYPE}
es.index(id=10,
body={
'name': 'ten',
'created': datetime.date(2016, 1, 1)
},
**args)
es.index(id=11,
body={
'name': 'eleven',
'created': datetime.date(2016, 2, 1)
},
**args)
es.index(id=12,
body={
'name': 'twelve',
'created': datetime.date(2016, 3, 1)
},
**args)
spans = writer.pop()
assert spans
assert len(spans) == 3
span = spans[0]
assert span.error == 0
assert span.get_tag('elasticsearch.method') == 'PUT'
assert span.get_tag('elasticsearch.url') == '/%s/%s/%s' % (
self.ES_INDEX, self.ES_TYPE, 10)
assert span.resource == 'PUT /%s/%s/?' % (self.ES_INDEX, self.ES_TYPE)
# Make the data available
es.indices.refresh(index=self.ES_INDEX)
spans = writer.pop()
assert spans, spans
assert len(spans) == 1
span = spans[0]
assert span.resource == 'POST /%s/_refresh' % self.ES_INDEX
assert span.get_tag('elasticsearch.method') == 'POST'
assert span.get_tag(
'elasticsearch.url') == '/%s/_refresh' % self.ES_INDEX
# Search data
result = es.search(sort=['name:desc'],
size=100,
body={'query': {
'match_all': {}
}},
**args)
assert len(result['hits']['hits']) == 3, result
spans = writer.pop()
assert spans
assert len(spans) == 1
span = spans[0]
assert span.resource == 'GET /%s/%s/_search' % (self.ES_INDEX,
self.ES_TYPE)
assert span.get_tag('elasticsearch.method') == 'GET'
assert span.get_tag('elasticsearch.url') == '/%s/%s/_search' % (
self.ES_INDEX, self.ES_TYPE)
assert span.get_tag('elasticsearch.body').replace(
' ', '') == '{"query":{"match_all":{}}}'
assert set(span.get_tag('elasticsearch.params').split('&')) == {
'sort=name%3Adesc', 'size=100'
}
assert http.QUERY_STRING not in span.meta
self.assertTrue(span.get_metric('elasticsearch.took') > 0)
# Search by type not supported by default json encoder
query = {'range': {'created': {'gte': datetime.date(2016, 2, 1)}}}
result = es.search(size=100, body={'query': query}, **args)
assert len(result['hits']['hits']) == 2, result
# Raise error 404 with a non existent index
writer.pop()
try:
es.get(index='non_existent_index', id=100, doc_type='_all')
assert 'error_not_raised' == 'elasticsearch.exceptions.TransportError'
except elasticsearch.exceptions.TransportError:
spans = writer.pop()
assert spans
span = spans[0]
assert span.get_tag(http.STATUS_CODE) == u'404'
# Raise error 400, the index 10 is created twice
try:
es.indices.create(index=10)
es.indices.create(index=10)
assert 'error_not_raised' == 'elasticsearch.exceptions.TransportError'
except elasticsearch.exceptions.TransportError:
spans = writer.pop()
assert spans
span = spans[-1]
assert span.get_tag(http.STATUS_CODE) == u'400'
# Drop the index, checking it won't raise exception on success or failure
es.indices.delete(index=self.ES_INDEX, ignore=[400, 404])
es.indices.delete(index=self.ES_INDEX, ignore=[400, 404])
def tearDown(self):
"""Clean ES"""
es = elasticsearch.Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
es.indices.delete(index=self.ES_INDEX, ignore=[400, 404])
def setUp(self):
"""Prepare ES"""
es = elasticsearch.Elasticsearch(port=ELASTICSEARCH_CONFIG['port'])
es.indices.delete(index=self.ES_INDEX, ignore=[400, 404])