tests for engines and errors

Author: dwhswenson

.coveragerc

@@ -7,3 +7,4 @@ exclude_lines =
     no-cov
     def __repr__
     raise NotImplementedError
+    __name__ == "__main__":

paths_cli/tests/wizard/test_engines.py

@@ -0,0 +1,14 @@
+import pytest
+import mock
+
+from paths_cli.tests.wizard.mock_wizard import mock_wizard
+
+from paths_cli.wizard.engines import engines, SUPPORTED_ENGINES
+
+def test_engines():
+    wizard = mock_wizard(['foo'])
+    with mock.patch.dict(SUPPORTED_ENGINES,
+                         {'foo': mock.Mock(return_value='ran foo')}):
+        foo = engines(wizard)
+    assert foo == 'ran foo'
+

paths_cli/tests/wizard/test_errors.py

@@ -0,0 +1,21 @@
+import pytest
+import mock
+
+from paths_cli.tests.wizard.mock_wizard import mock_wizard
+
+from paths_cli.wizard.errors import *
+
+def test_impossible_error_default():
+    with pytest.raises(ImpossibleError, match="You should never see this."):
+        raise ImpossibleError()
+
+@pytest.mark.parametrize('inputs', ['r', 'q'])
+def test_not_installed(inputs):
+    expected = {'r': RestartObjectException, 'q': SystemExit}[inputs]
+    wizard = mock_wizard([inputs])
+    with pytest.raises(expected):
+        not_installed(wizard, 'foo', 'widget')
+    log = wizard.console.log_text
+    assert 'foo installed' in log
+    assert 'different widget' in log
+

paths_cli/wizard/errors.py

@@ -7,16 +7,16 @@ def __init__(self, msg=None):
 class RestartObjectException(BaseException):
     pass
 
-def not_installed(package, obj_type):
-    retry = wizard.ask("Hey, it looks like you don't have {package} "
+def not_installed(wizard, package, obj_type):
+    retry = wizard.ask(f"Hey, it looks like you don't have {package} "
                        "installed. Do you want to try a different "
-                       "{obj_type}, or do you want to quit?",
+                       f"{obj_type}, or do you want to quit?",
                        options=["[R]etry", "[Q]uit"])
     if retry == 'r':
-        return
+        raise RestartObjectException()
     elif retry == 'q':
         exit()
-    else:
+    else:  # no-cov
         raise ImpossibleError()