1616
1717import mock
1818
19- from opencensus .common .transports import async
19+ from opencensus .common .transports import async_
2020
2121class Test_Worker (unittest .TestCase ):
2222
@@ -34,7 +34,7 @@ def test_constructor(self):
3434 grace_period = 10
3535 max_batch_size = 20
3636
37- worker = async ._Worker (exporter , grace_period = grace_period ,
37+ worker = async_ ._Worker (exporter , grace_period = grace_period ,
3838 max_batch_size = max_batch_size )
3939
4040 self .assertEqual (worker .exporter , exporter )
@@ -45,7 +45,7 @@ def test_constructor(self):
4545
4646 def test_start (self ):
4747 exporter = mock .Mock ()
48- worker = async ._Worker (exporter )
48+ worker = async_ ._Worker (exporter )
4949
5050 mock_thread , mock_atexit = self ._start_worker (worker )
5151
@@ -54,7 +54,7 @@ def test_start(self):
5454 self .assertTrue (worker ._thread .daemon )
5555 self .assertEqual (worker ._thread ._target , worker ._thread_main )
5656 self .assertEqual (
57- worker ._thread ._name , async ._WORKER_THREAD_NAME )
57+ worker ._thread ._name , async_ ._WORKER_THREAD_NAME )
5858 mock_atexit .assert_called_once_with (worker ._export_pending_data )
5959
6060 cur_thread = worker ._thread
@@ -63,7 +63,7 @@ def test_start(self):
6363
6464 def test_stop (self ):
6565 exporter = mock .Mock ()
66- worker = async ._Worker (exporter )
66+ worker = async_ ._Worker (exporter )
6767
6868 mock_thread , mock_atexit = self ._start_worker (worker )
6969
@@ -73,7 +73,7 @@ def test_stop(self):
7373
7474 self .assertEqual (worker ._queue .qsize (), 1 )
7575 self .assertEqual (
76- worker ._queue .get (), async ._WORKER_TERMINATOR )
76+ worker ._queue .get (), async_ ._WORKER_TERMINATOR )
7777 self .assertFalse (worker .is_alive )
7878 self .assertIsNone (worker ._thread )
7979
@@ -82,7 +82,7 @@ def test_stop(self):
8282
8383 def test__export_pending_data (self ):
8484 exporter = mock .Mock ()
85- worker = async ._Worker (exporter )
85+ worker = async_ ._Worker (exporter )
8686
8787 self ._start_worker (worker )
8888 worker ._export_pending_data ()
@@ -93,7 +93,7 @@ def test__export_pending_data(self):
9393
9494 def test__export_pending_data_non_empty_queue (self ):
9595 exporter = mock .Mock ()
96- worker = async ._Worker (exporter )
96+ worker = async_ ._Worker (exporter )
9797
9898 self ._start_worker (worker )
9999 worker .enqueue (mock .Mock ())
@@ -103,7 +103,7 @@ def test__export_pending_data_non_empty_queue(self):
103103
104104 def test__export_pending_data_did_not_join (self ):
105105 exporter = mock .Mock ()
106- worker = async ._Worker (exporter )
106+ worker = async_ ._Worker (exporter )
107107
108108 self ._start_worker (worker )
109109 worker ._thread ._terminate_on_join = False
@@ -114,7 +114,7 @@ def test__export_pending_data_did_not_join(self):
114114
115115 def test__thread_main (self ):
116116 exporter = mock .Mock ()
117- worker = async ._Worker (exporter )
117+ worker = async_ ._Worker (exporter )
118118
119119 trace1 = {
120120 'traceId' : 'test1' ,
@@ -127,7 +127,7 @@ def test__thread_main(self):
127127
128128 worker .enqueue (trace1 )
129129 worker .enqueue (trace2 )
130- worker ._queue .put_nowait (async ._WORKER_TERMINATOR )
130+ worker ._queue .put_nowait (async_ ._WORKER_TERMINATOR )
131131
132132 worker ._thread_main ()
133133
@@ -136,7 +136,7 @@ def test__thread_main(self):
136136
137137 def test__thread_main_batches (self ):
138138 exporter = mock .Mock ()
139- worker = async ._Worker (exporter , max_batch_size = 2 )
139+ worker = async_ ._Worker (exporter , max_batch_size = 2 )
140140
141141 # Enqueue three records and the termination signal. This should be
142142 # enough to perform two separate batches and a third loop with just
@@ -162,7 +162,7 @@ def test__thread_main_batches(self):
162162 worker .enqueue (trace3 )
163163 worker .enqueue (trace4 )
164164
165- worker ._queue .put_nowait (async ._WORKER_TERMINATOR )
165+ worker ._queue .put_nowait (async_ ._WORKER_TERMINATOR )
166166
167167 worker ._thread_main ()
168168
@@ -178,12 +178,12 @@ def emit(self, span):
178178 self .exported .append (span )
179179
180180 exporter = Exporter ()
181- worker = async ._Worker (exporter , max_batch_size = 2 )
181+ worker = async_ ._Worker (exporter , max_batch_size = 2 )
182182
183183 # Enqueue three records and the termination signal. This should be
184184 # enough to perform two separate batches and a third loop with just
185185 # the exit.
186- worker ._queue .put_nowait (async ._WORKER_TERMINATOR )
186+ worker ._queue .put_nowait (async_ ._WORKER_TERMINATOR )
187187
188188 # Worker should be terminated after sending span_data1, and
189189 # span_data2 won't be exported.
@@ -214,7 +214,7 @@ def emit(self, span):
214214 raise Exception ("This exporter is broken !" )
215215
216216 exporter = Exporter ()
217- worker = async ._Worker (exporter , max_batch_size = 2 )
217+ worker = async_ ._Worker (exporter , max_batch_size = 2 )
218218
219219 span_data0 = [mock .Mock ()]
220220 span_data1 = [mock .Mock ()]
@@ -223,7 +223,7 @@ def emit(self, span):
223223 worker .enqueue (span_data0 )
224224 worker .enqueue (span_data1 )
225225 worker .enqueue (span_data2 )
226- worker .enqueue (async ._WORKER_TERMINATOR )
226+ worker .enqueue (async_ ._WORKER_TERMINATOR )
227227
228228 worker ._thread_main ()
229229
@@ -243,7 +243,7 @@ def test_flush(self):
243243 from six .moves import queue
244244
245245 exporter = mock .Mock ()
246- worker = async ._Worker (exporter )
246+ worker = async_ ._Worker (exporter )
247247 worker ._queue = mock .Mock (spec = queue .Queue )
248248
249249 # Queue is empty, should not block.
@@ -255,24 +255,24 @@ class TestAsyncTransport(unittest.TestCase):
255255
256256 def test_constructor (self ):
257257 patch_worker = mock .patch (
258- 'opencensus.common.transports.async ._Worker' ,
258+ 'opencensus.common.transports.async_ ._Worker' ,
259259 autospec = True )
260260 exporter = mock .Mock ()
261261
262262 with patch_worker as mock_worker :
263- transport = async .AsyncTransport (exporter )
263+ transport = async_ .AsyncTransport (exporter )
264264
265265 self .assertTrue (transport .worker .start .called )
266266 self .assertEqual (transport .exporter , exporter )
267267
268268 def test_export (self ):
269269 patch_worker = mock .patch (
270- 'opencensus.common.transports.async ._Worker' ,
270+ 'opencensus.common.transports.async_ ._Worker' ,
271271 autospec = True )
272272 exporter = mock .Mock ()
273273
274274 with patch_worker as mock_worker :
275- transport = async .AsyncTransport (exporter )
275+ transport = async_ .AsyncTransport (exporter )
276276
277277 trace = {
278278 'traceId' : 'test' ,
@@ -285,12 +285,12 @@ def test_export(self):
285285
286286 def test_flush (self ):
287287 patch_worker = mock .patch (
288- 'opencensus.common.transports.async ._Worker' ,
288+ 'opencensus.common.transports.async_ ._Worker' ,
289289 autospec = True )
290290 exporter = mock .Mock ()
291291
292292 with patch_worker as mock_worker :
293- transport = async .AsyncTransport (exporter )
293+ transport = async_ .AsyncTransport (exporter )
294294
295295 transport .flush ()
296296
0 commit comments