improve test coverage

contrib/opencensus-ext-azure/opencensus/ext/azure/trace_exporter/__init__.py

@@ -237,7 +237,7 @@ def emit(self, batch, event=None):
             if len(batch) < self.options.max_batch_size:
                 self._transmit_from_storage()
         except Exception as ex:
-            logger.error('Transmission exception: %s.', ex)
+            logger.exception('Transmission exception: %s.', ex)
 
     def _stop(self, timeout=None):
         self.storage.close()

contrib/opencensus-ext-azure/tests/test_azure_trace_exporter.py

@@ -57,6 +57,16 @@ def test_emit_empty(self, request_mock):
         self.assertEqual(len(os.listdir(exporter.storage.path)), 0)
         exporter._stop()
 
+    @mock.patch('opencensus.ext.azure.trace_exporter.logger')
+    def test_emit_exception(self, mock_logger):
+        exporter = trace_exporter.AzureExporter(
+            instrumentation_key='12345678-1234-5678-abcd-12345678abcd',
+            storage_path=os.path.join(TEST_FOLDER, self.id()),
+        )
+        exporter.emit([1, 2, 3])
+        mock_logger.exception.assert_called()
+        exporter._stop()
+
     @mock.patch('opencensus.ext.azure.trace_exporter.AzureExporter.span_data_to_envelope')  # noqa: E501
     def test_emit_failure(self, span_data_to_envelope_mock):
         span_data_to_envelope_mock.return_value = ['bar']

tests/unit/common/test_schedule.py

@@ -20,6 +20,13 @@
 TIMEOUT = .1
 
 
+class TestQueueEvent(unittest.TestCase):
+    def test_basic(self):
+        evt = QueueEvent('foobar')
+        self.assertFalse(evt.wait(timeout=TIMEOUT))
+        evt.set()
+        self.assertTrue(evt.wait(timeout=TIMEOUT))
+
 class TestQueue(unittest.TestCase):
     def test_gets(self):
         queue = Queue(capacity=100)