Make the Result available on an error

Author: jedie

manageprojects/test_utils/click_cli_utils.py

@@ -30,6 +30,11 @@ class ForceRichTerminalWidth(MassContextManager):
     )
 
 
+class ClickInvokeCliException(Exception):
+    def __init__(self, result: Result):
+        self.result = result
+
+
 def invoke_click(cli, *args, expected_stderr='', expected_exit_code=0, strip=True, **kwargs):
     assert isinstance(cli, click.Command)
 
@@ -40,7 +45,7 @@ def invoke_click(cli, *args, expected_stderr='', expected_exit_code=0, strip=Tru
         result: Result = runner.invoke(cli=cli, args=args, **kwargs, color=False)
 
     if result.exception:
-        raise result.exception
+        raise ClickInvokeCliException(result) from result.exception
 
     stdout = result.stdout
     stderr = result.stderr

manageprojects/tests/test_click_cli_utils.py

@@ -0,0 +1,20 @@
+from unittest.mock import patch
+
+from click.testing import Result
+
+from manageprojects.cli import cli_app
+from manageprojects.cli.cli_app import cli
+from manageprojects.test_utils.click_cli_utils import ClickInvokeCliException, invoke_click
+from manageprojects.tests.base import BaseTestCase
+
+
+class CliTestCase(BaseTestCase):
+    def test_invoke_click(self):
+        with patch.object(cli_app, 'verbose_check_call', side_effect=RuntimeError('Bam!')):
+            with self.assertRaises(ClickInvokeCliException) as cm:
+                invoke_click(cli, 'install')
+
+        self.assertIsInstance(cm.exception, ClickInvokeCliException)
+        self.assertIsInstance(cm.exception.result, Result)
+        self.assertIsInstance(cm.exception.result.exception, RuntimeError)
+        self.assertEqual(cm.exception.result.exception.args, ('Bam!',))