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