add more robust powershell install/remove script

Author: bckohan

django_typer/management/commands/shellcompletion.py

@@ -498,6 +498,10 @@ def get_completion() -> None:
             if not args:
                 call_fallback(fallback)
             else:
+                # we first need to figure out which command is being invoked
+                # but we cant be sure which index the command name is at so
+                # we try to fetch each in order and assume the first arg
+                # that resolves to a command is the command
                 cmd = None
                 cmd_idx = -1
                 try:
@@ -519,6 +523,7 @@ def get_completion() -> None:
                     raise  # otherwise nowhere to go - just error out
 
                 if isinstance(cmd, TyperCommand):  # type: ignore[unreachable]
+                    # this will exit out so no return is needed here
                     cmd.typer_app(
                         args=args[cmd_idx + 1 :],
                         standalone_mode=True,

django_typer/tests/completion_tests.py

@@ -318,6 +318,22 @@ def verify_remove(self, script=None):
             self.assertTrue(contents)  # should have been deleted if it were empty
 
 
+class PowerShellInstallRemoveTests(_InstalledScriptTestCase, PowerShellTests):
+
+    def test_shell_complete(self):
+        # the power shell completion script registration is all in one file
+        # so install/remove is more complicated when other scripts are in that
+        # file - this tests that
+        self.install(script="./manage.py")
+        self.verify_install(script="./manage.py")
+        self.install(script=self.manage_script)
+        self.verify_install(script=self.manage_script)
+        self.remove(script=self.manage_script)
+        self.verify_remove(script=self.manage_script)
+        self.remove(script="./manage.py")
+        self.verify_remove(script="./manage.py")
+
+
 @pytest.mark.skipif(default_shell is None, reason="shellingham failed to detect shell")
 class DefaultCompleteTestCase(_DefaultCompleteTestCase, TestCase):
     pass