Ticket #23551: test_subcommand_completed.patch

.py

@@ -32 +32 @@
             del os.environ['DJANGO_AUTO_COMPLETE']
 
     def _user_input(self, input_str):
+        """Set the environment and the list of command line arguments.
+
+        This method sets the bash variables $COMP_WORDS and $COMP_CWORD.
+        The former is an array consisting of the individual words in the 
+        current command line, the latter is the index of the current 
+        cursor position, so in case a word is completed and the cursor is
+        placed after a whitespace, $COMP_CWORD must be incremented by 1:
+
+          * 'django-admin start' -> COMP_CWORD=1 
+          * 'django-admin startproject' -> COMP_CWORD=1
+          * 'django-admin startproject ' -> COMP_CWORD=2 
+        """
         os.environ['COMP_WORDS'] = input_str
-        os.environ['COMP_CWORD'] = str(len(input_str.split()) - 1)
-        sys.argv = input_str.split(' ')
+        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):
         util = ManagementUtility(argv=sys.argv)
@@ -68 +82 @@
         output = self._run_autocomplete()
         self.assertEqual(output, ['sql sqlall sqlclear sqlcustom sqldropindexes sqlflush sqlindexes sqlmigrate sqlsequencereset'])
 
+    def test_completed_subcommand(self):
+        """Show option flags in case a subcommand is completed"""
+        self._user_input('django-admin startproject ') # Trailing whitespace
+        output = self._run_autocomplete()
+        for item in output:
+            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 --')
@@ -88 +109 @@
             for app_config in apps.get_app_configs()
             if app_config.label.startswith('a'))
         self.assertEqual(output, a_labels)
+