mirror of
https://github.com/django/django.git
synced 2025-10-24 14:16:09 +00:00
Refs #33476 -- Reformatted code with Black.
This commit is contained in:
committed by
Mariusz Felisiak
parent
f68fa8b45d
commit
9c19aff7c7
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user