Use scopped logger in transports.async_ (census-instrumentation#691)

angelini · reyang · commit e3858808eefc · 2019-06-19T10:10:22.000-07:00
diff --git a/opencensus/common/transports/async_.py b/opencensus/common/transports/async_.py
@@ -27,6 +27,8 @@
 _WORKER_THREAD_NAME = 'opencensus.common.Worker'
 _WORKER_TERMINATOR = object()
 
+logger = logging.getLogger(__name__)
+
 
 class _Worker(object):
     """A background thread that exports batches of data.
@@ -108,7 +110,7 @@ def _thread_main(self):
                 try:
                     self.exporter.emit(data)
                 except Exception:
-                    logging.exception(
+                    logger.exception(
                         '%s failed to emit data.'
                         'Dropping %s objects from queue.',
                         self.exporter.__class__.__name__,
diff --git a/tests/unit/common/transports/test_async.py b/tests/unit/common/transports/test_async.py
@@ -199,7 +199,7 @@ def emit(self, span):
         # trace2 should be left in the queue because worker is terminated.
         self.assertEqual(worker._queue.qsize(), 1)
 
-    @mock.patch('logging.exception')
+    @mock.patch('opencensus.common.transports.async_.logger.exception')
     def test__thread_main_alive_on_emit_failed(self, mock):
 
         class Exporter(object):
