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