33from ddtrace .span import Span
44from ddtrace .writer import AsyncWorker , Q
55
6- class RemoveAllProcessor ():
6+ class RemoveAllFilter ():
77 def __init__ (self ):
8- self .processed_traces = 0
8+ self .filtered_traces = 0
99
1010 def process_trace (self , trace ):
11- self .processed_traces += 1
11+ self .filtered_traces += 1
1212 return None
1313
14- class KeepAllProcessor ():
14+ class KeepAllFilter ():
1515 def __init__ (self ):
16- self .processed_traces = 0
16+ self .filtered_traces = 0
1717
1818 def process_trace (self , trace ):
19- self .processed_traces += 1
19+ self .filtered_traces += 1
2020 return trace
2121
22- class AddTagProcessor ():
22+ class AddTagFilter ():
2323 def __init__ (self , tag_name ):
2424 self .tag_name = tag_name
25- self .processed_traces = 0
25+ self .filtered_traces = 0
2626
2727 def process_trace (self , trace ):
28- self .processed_traces += 1
28+ self .filtered_traces += 1
2929 for span in trace :
3030 span .set_tag (self .tag_name , "A value" )
3131 return trace
@@ -48,42 +48,42 @@ def setUp(self):
4848 for i in range (N_TRACES ):
4949 self .traces .add ([Span (tracer = None , name = "name" , trace_id = i , span_id = j , parent_id = j - 1 or None ) for j in range (7 )])
5050
51- def test_processing_pipeline_keep_all (self ):
52- processor = KeepAllProcessor ()
53- processing_pipeline = [processor ]
54- worker = AsyncWorker (self .api , self .traces , self .services , processing_pipeline = processing_pipeline )
51+ def test_filters_keep_all (self ):
52+ filtr = KeepAllFilter ()
53+ filters = [filtr ]
54+ worker = AsyncWorker (self .api , self .traces , self .services , filters = filters )
5555 worker .stop ()
5656 worker .join ()
5757 self .assertEqual (len (self .api .traces ), N_TRACES )
58- self .assertEqual (processor . processed_traces , N_TRACES )
58+ self .assertEqual (filtr . filtered_traces , N_TRACES )
5959
60- def test_processing_pipeline_remove_all (self ):
61- processor = RemoveAllProcessor ()
62- processing_pipeline = [processor ]
63- worker = AsyncWorker (self .api , self .traces , self .services , processing_pipeline = processing_pipeline )
60+ def test_filters_remove_all (self ):
61+ filtr = RemoveAllFilter ()
62+ filters = [filtr ]
63+ worker = AsyncWorker (self .api , self .traces , self .services , filters = filters )
6464 worker .stop ()
6565 worker .join ()
6666 self .assertEqual (len (self .api .traces ), 0 )
67- self .assertEqual (processor . processed_traces , N_TRACES )
67+ self .assertEqual (filtr . filtered_traces , N_TRACES )
6868
69- def test_processing_pipeline_add_tag (self ):
69+ def test_filters_add_tag (self ):
7070 tag_name = "Tag"
71- processor = AddTagProcessor (tag_name )
72- processing_pipeline = [processor ]
73- worker = AsyncWorker (self .api , self .traces , self .services , processing_pipeline = processing_pipeline )
71+ filtr = AddTagFilter (tag_name )
72+ filters = [filtr ]
73+ worker = AsyncWorker (self .api , self .traces , self .services , filters = filters )
7474 worker .stop ()
7575 worker .join ()
7676 self .assertEqual (len (self .api .traces ), N_TRACES )
77- self .assertEqual (processor . processed_traces , N_TRACES )
77+ self .assertEqual (filtr . filtered_traces , N_TRACES )
7878 for trace in self .api .traces :
7979 for span in trace :
8080 self .assertIsNotNone (span .get_tag (tag_name ))
8181
82- def test_processing_pipeline_short_circuit (self ):
83- processor = KeepAllProcessor ()
84- processing_pipeline = [RemoveAllProcessor (), processor ]
85- worker = AsyncWorker (self .api , self .traces , self .services , processing_pipeline = processing_pipeline )
82+ def test_filters_short_circuit (self ):
83+ filtr = KeepAllFilter ()
84+ filters = [RemoveAllFilter (), filtr ]
85+ worker = AsyncWorker (self .api , self .traces , self .services , filters = filters )
8686 worker .stop ()
8787 worker .join ()
8888 self .assertEqual (len (self .api .traces ), 0 )
89- self .assertEqual (processor . processed_traces , 0 )
89+ self .assertEqual (filtr . filtered_traces , 0 )
0 commit comments