Fix test runner

python-manage · Jun 14, 2016 · 0f86ef2 · 0f86ef2
1 parent 9369c68
commit 0f86ef2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_manage.py b/tests/test_manage.py
@@ -13,7 +13,7 @@
 # from contextlib import contextmanager
 from click.testing import CliRunner
 
-from manage import cli
+from manage.cli import cli
 
 
 class TestManage(object):
@@ -27,12 +27,12 @@ def test_something(self):
 
     def test_command_line_interface(self):
         runner = CliRunner()
-        result = runner.invoke(cli.main)
+        result = runner.invoke(cli, ['--help'])
         assert result.exit_code == 0
         assert 'shell' in result.output
-        help_result = runner.invoke(cli.main, ['--help'])
+        help_result = runner.invoke(cli, ['--help'])
         assert help_result.exit_code == 0
-        assert '--help  Show this message and exit.' in help_result.output
+        assert 'Show this message and exit.' in help_result.output
 
     @classmethod
     def teardown_class(cls):