[GH 59] Enable/disable console logging

Add an API on storops module to enabled and disable the console logging

storops/__init__.py

@@ -15,9 +15,29 @@
 #    under the License.
 from __future__ import unicode_literals
 
+import sys
+import logging
+
 from storops.vnx.resource.system import VNXSystem  # noqa
 from storops.unity.resource.system import UnitySystem  # noqa
 from storops.vnx.enums import *  # noqa
 from storops.unity.enums import *  # noqa
 
 __author__ = 'Cedric Zhuang'
+
+
+def enable_log(level=logging.DEBUG):
+    """Enable console logging.
+
+    This is a utils method for try run with storops.
+    """
+    log = logging.getLogger(__name__)
+    log.setLevel(level)
+    if not log.handlers:
+        _handler = logging.StreamHandler(sys.stdout)
+        log.addHandler(_handler)
+
+
+def disable_log():
+    log = logging.getLogger(__name__)
+    log.handlers = []

test/test_storops.py

@@ -0,0 +1,44 @@
+# coding=utf-8
+# Copyright (c) 2015 EMC Corporation.
+# All Rights Reserved.
+#
+#    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.
+from __future__ import unicode_literals
+
+import logging
+from hamcrest import assert_that, equal_to
+
+import storops
+
+
+def test_enabled_logger():
+    storops.enable_log()
+    storops.enable_log()
+    log = logging.getLogger('storops')
+    assert_that(len(log.handlers), equal_to(1))
+    assert_that(log.getEffectiveLevel(), equal_to(logging.DEBUG))
+
+
+def test_enable_logger_called_twice():
+    storops.enable_log()
+    storops.enable_log(logging.INFO)
+    log = logging.getLogger('storops')
+    assert_that(len(log.handlers), equal_to(1))
+    assert_that(log.getEffectiveLevel(), equal_to(logging.INFO))
+
+
+def test_disable_logger():
+    storops.enable_log()
+    storops.disable_log()
+    log = logging.getLogger('storops')
+    assert_that(len(log.handlers), equal_to(0))