diff options
Diffstat (limited to 'tests/bash_completion')
| -rw-r--r-- | tests/bash_completion/tests.py | 49 |
1 files changed, 25 insertions, 24 deletions
diff --git a/tests/bash_completion/tests.py b/tests/bash_completion/tests.py index 1d35e1f28e..856e351dfa 100644 --- a/tests/bash_completion/tests.py +++ b/tests/bash_completion/tests.py @@ -18,14 +18,14 @@ class BashCompletionTests(unittest.TestCase): """ def setUp(self): - self.old_DJANGO_AUTO_COMPLETE = os.environ.get('DJANGO_AUTO_COMPLETE') - os.environ['DJANGO_AUTO_COMPLETE'] = '1' + self.old_DJANGO_AUTO_COMPLETE = os.environ.get("DJANGO_AUTO_COMPLETE") + os.environ["DJANGO_AUTO_COMPLETE"] = "1" def tearDown(self): if self.old_DJANGO_AUTO_COMPLETE: - os.environ['DJANGO_AUTO_COMPLETE'] = self.old_DJANGO_AUTO_COMPLETE + os.environ["DJANGO_AUTO_COMPLETE"] = self.old_DJANGO_AUTO_COMPLETE else: - del os.environ['DJANGO_AUTO_COMPLETE'] + del os.environ["DJANGO_AUTO_COMPLETE"] def _user_input(self, input_str): """ @@ -41,10 +41,10 @@ class BashCompletionTests(unittest.TestCase): * 'django-admin startproject' -> COMP_CWORD=1 * 'django-admin startproject ' -> COMP_CWORD=2 """ - os.environ['COMP_WORDS'] = input_str - idx = len(input_str.split(' ')) - 1 # Index of the last word - comp_cword = idx + 1 if input_str.endswith(' ') else idx - os.environ['COMP_CWORD'] = str(comp_cword) + os.environ["COMP_WORDS"] = input_str + idx = len(input_str.split(" ")) - 1 # Index of the last word + comp_cword = idx + 1 if input_str.endswith(" ") else idx + os.environ["COMP_CWORD"] = str(comp_cword) sys.argv = input_str.split() def _run_autocomplete(self): @@ -54,51 +54,52 @@ class BashCompletionTests(unittest.TestCase): util.autocomplete() except SystemExit: pass - return stdout.getvalue().strip().split('\n') + return stdout.getvalue().strip().split("\n") def test_django_admin_py(self): "django_admin.py will autocomplete option flags" - self._user_input('django-admin sqlmigrate --verb') + self._user_input("django-admin sqlmigrate --verb") output = self._run_autocomplete() - self.assertEqual(output, ['--verbosity=']) + self.assertEqual(output, ["--verbosity="]) def test_manage_py(self): "manage.py will autocomplete option flags" - self._user_input('manage.py sqlmigrate --verb') + self._user_input("manage.py sqlmigrate --verb") output = self._run_autocomplete() - self.assertEqual(output, ['--verbosity=']) + self.assertEqual(output, ["--verbosity="]) def test_custom_command(self): "A custom command can autocomplete option flags" - self._user_input('django-admin test_command --l') + self._user_input("django-admin test_command --l") output = self._run_autocomplete() - self.assertEqual(output, ['--list']) + self.assertEqual(output, ["--list"]) def test_subcommands(self): "Subcommands can be autocompleted" - self._user_input('django-admin sql') + self._user_input("django-admin sql") output = self._run_autocomplete() - self.assertEqual(output, ['sqlflush sqlmigrate sqlsequencereset']) + self.assertEqual(output, ["sqlflush sqlmigrate sqlsequencereset"]) def test_completed_subcommand(self): "Show option flags in case a subcommand is completed" - self._user_input('django-admin startproject ') # Trailing whitespace + self._user_input("django-admin startproject ") # Trailing whitespace output = self._run_autocomplete() for item in output: - self.assertTrue(item.startswith('--')) + self.assertTrue(item.startswith("--")) def test_help(self): "No errors, just an empty list if there are no autocomplete options" - self._user_input('django-admin help --') + self._user_input("django-admin help --") output = self._run_autocomplete() - self.assertEqual(output, ['']) + self.assertEqual(output, [""]) def test_app_completion(self): "Application names will be autocompleted for an AppCommand" - self._user_input('django-admin sqlmigrate a') + self._user_input("django-admin sqlmigrate a") output = self._run_autocomplete() a_labels = sorted( - app_config.label for app_config in apps.get_app_configs() - if app_config.label.startswith('a') + app_config.label + for app_config in apps.get_app_configs() + if app_config.label.startswith("a") ) self.assertEqual(output, a_labels) |