start to tests for OpenPathSamplingCLI class

Author: dwhswenson

paths_cli/cli.py

@@ -60,7 +60,7 @@ def _deregister_plugin(self, plugin):
         self._sections[plugin.section].remove(plugin.name)
 
     def plugin_for_command(self, command_name):
-        return {p.name: p for p in self.plugins}[name]
+        return {p.name: p for p in self.plugins}[command_name]
 
     @staticmethod
     def _list_plugin_files(plugin_folders):

paths_cli/tests/test_cli.py

@@ -1,4 +1,5 @@
 import pytest
+from unittest.mock import patch, MagicMock
 from click.testing import CliRunner
 
 import logging
@@ -7,25 +8,51 @@
 from .null_command import NullCommandContext
 
 class TestOpenPathSamplingCLI(object):
+    # TODO: more thorough testing of private methods to find/register
+    # plugins might be nice; so far we mainly focus on testing the API.
+    # (Still have smoke tests covering everything, though.)
     def setup(self):
-        # TODO: patch out the directory to fake the plugins
-        self.cli = OpenPathSamplingCLI()
+        self.plugin_dict = {
+            'foo': OPSPlugin(name='foo',
+                             filename='foo.py',
+                             func=lambda: 'foo',
+                             section='Simulation'),
+            'foo-bar': OPSPlugin(name='foo-bar',
+                                 filename='foo_bar.py',
+                                 func=lambda: 'foobar',
+                                 section='Miscellaneous')
+        }
+        self.fake_plugins = list(self.plugin_dict.values())
+        mock_plugins = MagicMock(return_value=self.fake_plugins)
+        with patch.object(OpenPathSamplingCLI, '_load_plugin_files',
+                          mock_plugins):
+            self.cli = OpenPathSamplingCLI()
 
     def test_plugins(self):
-        pytest.skip()
-        pass
+        assert self.cli.plugins == self.fake_plugins
+        assert self.cli._sections['Simulation'] == ['foo']
+        assert self.cli._sections['Miscellaneous'] == ['foo-bar']
 
-    def test_get_command(self):
-        # test renamings
-        pytest.skip()
-        pass
+    @pytest.mark.parametrize('name', ['foo', 'foo-bar'])
+    def test_plugin_for_command(self, name):
+        assert self.cli.plugin_for_command(name) == self.plugin_dict[name]
+
+    def test_list_commands(self):
+        assert self.cli.list_commands(ctx=None) == ['foo', 'foo-bar']
+
+    @pytest.mark.parametrize('command', ['foo-bar', 'foo_bar'])
+    def test_get_command(self, command):
+        # this tests that renamings work
+        cmd = self.cli.get_command(ctx=None, name=command)
+        assert cmd() == 'foobar'
 
     def test_format_commands(self):
         pytest.skip()
         # use a mock to get the formatter
         # test that it skips a section if it is empty
         pass
 
+
 @pytest.mark.parametrize('with_log', [True, False])
 def test_main_log(with_log):
     logged_stdout = "About to run command\n"