示例#1
0
 def create_worker(self, filters=None, api_class=DummyAPI, enable_stats=False):
     with self.override_global_config(dict(health_metrics_enabled=enable_stats)):
         self.dogstatsd = mock.Mock()
         worker = AgentWriter(dogstatsd=self.dogstatsd, filters=filters)
         worker._STATS_EVERY_INTERVAL = 1
         self.api = api_class()
         worker.api = self.api
         for i in range(self.N_TRACES):
             worker.write(
                 [Span(tracer=None, name="name", trace_id=i, span_id=j, parent_id=j - 1 or None) for j in range(7)]
             )
         worker.stop()
         worker.join()
         return worker
示例#2
0
 def create_worker(self, filters=None, api_class=DummyAPI, enable_stats=False):
     self.dogstatsd = mock.Mock()
     worker = AgentWriter(dogstatsd=self.dogstatsd, filters=filters)
     worker._ENABLE_STATS = enable_stats
     worker._STATS_EVERY_INTERVAL = 1
     self.api = api_class()
     worker.api = self.api
     for i in range(self.N_TRACES):
         worker.write([
             Span(tracer=None, name='name', trace_id=i, span_id=j, parent_id=j - 1 or None)
             for j in range(7)
         ])
     worker.stop()
     worker.join()
     return worker