Merge "Re-enable stacktracing when --debug is used"

glanceclient/tests/unit/test_shell.py

@@ -23,6 +23,7 @@ import hashlib
 import logging
 import os
 import sys
+import traceback
 import uuid
 
 import fixtures
@@ -151,6 +152,44 @@ class ShellTest(testutils.TestCase):
         argstr = '--os-image-api-version 2 help foofoo'
         self.assertRaises(exc.CommandError, shell.main, argstr.split())
 
+    @mock.patch('sys.stdout', six.StringIO())
+    @mock.patch('sys.stderr', six.StringIO())
+    @mock.patch('sys.argv', ['glance', 'help', 'foofoo'])
+    def test_no_stacktrace_when_debug_disabled(self):
+        with mock.patch.object(traceback, 'print_exc') as mock_print_exc:
+            try:
+                openstack_shell.main()
+            except SystemExit:
+                pass
+            self.assertFalse(mock_print_exc.called)
+
+    @mock.patch('sys.stdout', six.StringIO())
+    @mock.patch('sys.stderr', six.StringIO())
+    @mock.patch('sys.argv', ['glance', 'help', 'foofoo'])
+    def test_stacktrace_when_debug_enabled_by_env(self):
+        old_environment = os.environ.copy()
+        os.environ = {'GLANCECLIENT_DEBUG': '1'}
+        try:
+            with mock.patch.object(traceback, 'print_exc') as mock_print_exc:
+                try:
+                    openstack_shell.main()
+                except SystemExit:
+                    pass
+                self.assertTrue(mock_print_exc.called)
+        finally:
+            os.environ = old_environment
+
+    @mock.patch('sys.stdout', six.StringIO())
+    @mock.patch('sys.stderr', six.StringIO())
+    @mock.patch('sys.argv', ['glance', '--debug', 'help', 'foofoo'])
+    def test_stacktrace_when_debug_enabled(self):
+        with mock.patch.object(traceback, 'print_exc') as mock_print_exc:
+            try:
+                openstack_shell.main()
+            except SystemExit:
+                pass
+            self.assertTrue(mock_print_exc.called)
+
     def test_help(self):
         shell = openstack_shell.OpenStackImagesShell()
         argstr = '--os-image-api-version 2 help'