def test_patch_unpatch(self):
self.patch_algoliasearch()
# Test patch idempotence
patch()
patch()
self.perform_search('test search')
spans = self.get_spans()
self.reset()
assert spans, spans
assert len(spans) == 1
# Test unpatch
unpatch()
self.index.search('test search')
spans = self.get_spans()
self.reset()
assert not spans, spans
# Test patch again
self.reset()
patch()
self.index.search('test search')
spans = self.get_spans()
assert spans, spans
assert len(spans) == 1
def test_user_specified_service(self):
"""
When a service name is specified by the user
The algoliasearch integration should use it as the service name
"""
patch_all()
Pin.override(self.index, tracer=self.tracer)
self.perform_search("test search")
spans = self.get_spans()
self.reset()
assert spans, spans
assert len(spans) == 1
assert spans[0].service == "mysvc"
unpatch()
def test_patch_all_auto_enable(self):
patch_all()
Pin.override(self.index, tracer=self.tracer)
self.perform_search('test search')
spans = self.get_spans()
self.reset()
assert spans, spans
assert len(spans) == 1
unpatch()
self.perform_search('test search')
spans = self.get_spans()
assert not spans, spans
def tearDown(self):
super(AlgoliasearchTest, self).tearDown()
unpatch()
if hasattr(self, 'tracer'):
self.reset()
