1 files changed, 80 insertions, 0 deletions
diff --git a/src/python/grpcio_tests/tests/unit/_logging_test.py b/src/python/grpcio_tests/tests/unit/_logging_test.py
new file mode 100644
index 0000000000..631b9de9db
--- /dev/null
+++ b/src/python/grpcio_tests/tests/unit/_logging_test.py
@@ -0,0 +1,80 @@
+# Copyright 2018 gRPC authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Test of gRPC Python's interaction with the python logging module"""
+
+import unittest
+import six
+from six.moves import reload_module
+import logging
+import grpc
+import functools
+import sys
+
+
+def patch_stderr(f):
+
+    @functools.wraps(f)
+    def _impl(*args, **kwargs):
+        old_stderr = sys.stderr
+        sys.stderr = six.StringIO()
+        try:
+            f(*args, **kwargs)
+        finally:
+            sys.stderr = old_stderr
+
+    return _impl
+
+
+def isolated_logging(f):
+
+    @functools.wraps(f)
+    def _impl(*args, **kwargs):
+        reload_module(logging)
+        reload_module(grpc)
+        try:
+            f(*args, **kwargs)
+        finally:
+            reload_module(logging)
+
+    return _impl
+
+
+class LoggingTest(unittest.TestCase):
+
+    @isolated_logging
+    def test_logger_not_occupied(self):
+        self.assertEqual(0, len(logging.getLogger().handlers))
+
+    @patch_stderr
+    @isolated_logging
+    def test_handler_found(self):
+        self.assertEqual(0, len(sys.stderr.getvalue()))
+
+    @isolated_logging
+    def test_can_configure_logger(self):
+        intended_stream = six.StringIO()
+        logging.basicConfig(stream=intended_stream)
+        self.assertEqual(1, len(logging.getLogger().handlers))
+        self.assertIs(logging.getLogger().handlers[0].stream, intended_stream)
+
+    @isolated_logging
+    def test_grpc_logger(self):
+        self.assertIn("grpc", logging.Logger.manager.loggerDict)
+        root_logger = logging.getLogger("grpc")
+        self.assertEqual(1, len(root_logger.handlers))
+        self.assertIsInstance(root_logger.handlers[0], logging.NullHandler)
+
+
+if __name__ == '__main__':
+    unittest.main(verbosity=2)