Fix hanging async tests (#382)

Author: c24t

tests/unit/common/transports/test_async.py

@@ -18,6 +18,12 @@
 
 from opencensus.common.transports import async_
 
+
+# Don't let workers wait between exports in testing
+wait_period_patch = mock.patch(
+    'opencensus.common.transports.async_._WAIT_PERIOD', 0)
+
+
 class Test_Worker(unittest.TestCase):
 
     def _start_worker(self, worker):
@@ -35,7 +41,7 @@ def test_constructor(self):
         max_batch_size = 20
 
         worker = async_._Worker(exporter, grace_period=grace_period,
-                                           max_batch_size=max_batch_size)
+                                max_batch_size=max_batch_size)
 
         self.assertEqual(worker.exporter, exporter)
         self.assertEqual(worker._grace_period, grace_period)
@@ -67,8 +73,6 @@ def test_stop(self):
 
         mock_thread, mock_atexit = self._start_worker(worker)
 
-        thread = worker._thread
-
         worker.stop()
 
         self.assertEqual(worker._queue.qsize(), 1)
@@ -129,7 +133,8 @@ def test__thread_main(self):
         worker.enqueue(trace2)
         worker._queue.put_nowait(async_._WORKER_TERMINATOR)
 
-        worker._thread_main()
+        with wait_period_patch:
+            worker._thread_main()
 
         self.assertTrue(worker.exporter.emit.called)
         self.assertEqual(worker._queue.qsize(), 0)
@@ -164,7 +169,8 @@ def test__thread_main_batches(self):
 
         worker._queue.put_nowait(async_._WORKER_TERMINATOR)
 
-        worker._thread_main()
+        with wait_period_patch:
+            worker._thread_main()
 
         self.assertEqual(worker._queue.qsize(), 0)
 
@@ -193,7 +199,8 @@ def emit(self, span):
         worker.enqueue(span_data1)
         worker.enqueue(span_data2)
 
-        worker._thread_main()
+        with wait_period_patch:
+            worker._thread_main()
 
         self.assertEqual(exporter.exported, [span_data1])
 
@@ -225,7 +232,8 @@ def emit(self, span):
         worker.enqueue(span_data2)
         worker.enqueue(async_._WORKER_TERMINATOR)
 
-        worker._thread_main()
+        with wait_period_patch:
+            worker._thread_main()
 
         # Span 2 should throw an exception, only span 0 and 1 are left
         self.assertEqual(exporter.exported, span_data0 + span_data1)
@@ -238,7 +246,6 @@ def emit(self, span):
         # and the data was dropped.
         self.assertEqual(worker._queue.qsize(), 0)
 
-
     def test_flush(self):
         from six.moves import queue
 
@@ -259,7 +266,7 @@ def test_constructor(self):
             autospec=True)
         exporter = mock.Mock()
 
-        with patch_worker as mock_worker:
+        with patch_worker:
             transport = async_.AsyncTransport(exporter)
 
         self.assertTrue(transport.worker.start.called)
@@ -271,7 +278,7 @@ def test_export(self):
             autospec=True)
         exporter = mock.Mock()
 
-        with patch_worker as mock_worker:
+        with patch_worker:
             transport = async_.AsyncTransport(exporter)
 
         trace = {
@@ -289,7 +296,7 @@ def test_flush(self):
             autospec=True)
         exporter = mock.Mock()
 
-        with patch_worker as mock_worker:
+        with patch_worker:
             transport = async_.AsyncTransport(exporter)
 
             transport.flush()