diff --git a/tests/ext/argcomplete_tests.py b/tests/ext/argcomplete_tests.py index d508c4d8..764cd1cd 100644 --- a/tests/ext/argcomplete_tests.py +++ b/tests/ext/argcomplete_tests.py @@ -2,17 +2,31 @@ import os from cement.ext import ext_argcomplete +from cement.ext.ext_argparse import ArgparseController, expose from cement.utils import test from cement.utils.misc import rando APP = rando()[:12] +class MyBaseController(ArgparseController): + class Meta: + label = 'base' + + @expose() + def default(self): + pass class ArgcompleteExtTestCase(test.CementExtTestCase): def setUp(self): super(ArgcompleteExtTestCase, self).setUp() - self.app = self.make_app(APP, extensions=['argcomplete']) + self.app = self.make_app(APP, + base_controller=MyBaseController, + extensions=[ + 'argparse', + 'argcomplete' + ], + ) def test_argcomplete(self): # not really sure how to test this for reals... but let's atleast get