coverage

Author: bckohan

django_typer/tests/test_app/management/commands/completion.py

@@ -25,8 +25,23 @@ def handle(
                 shell_complete=completers.complete_app_label,
             ),
         ],
+        option: Annotated[
+            t.Optional[AppConfig],
+            typer.Option(
+                parser=parsers.parse_app_label,
+                help=_("An app given as an option."),
+                shell_complete=completers.complete_app_label,
+            ),
+        ] = None,
     ):
         assert self.__class__ == Command
         for app in django_apps:
             assert isinstance(app, AppConfig)
+        if option:
+            return json.dumps(
+                {
+                    "django_apps": [app.label for app in django_apps],
+                    "option": option.label,
+                }
+            )
         return json.dumps([app.label for app in django_apps])

django_typer/tests/tests.py

@@ -1855,6 +1855,7 @@ def test_app_label_parser_completers(self):
             call_command("completion", "django_typer_tests.polls")
 
         poll_app = apps.get_app_config("django_typer_tests_polls")
+        test_app = apps.get_app_config("django_typer_tests_test_app")
         cmd = get_command("completion")
         self.assertEqual(
             json.loads(cmd([poll_app])),
@@ -1865,6 +1866,52 @@ def test_app_label_parser_completers(self):
             json.loads(cmd(django_apps=[poll_app])), ["django_typer_tests_polls"]
         )
 
+        self.assertEqual(
+            json.loads(cmd(django_apps=[poll_app], option=test_app)),
+            {
+                "django_apps": ["django_typer_tests_polls"],
+                "option": "django_typer_tests_test_app",
+            },
+        )
+
+        self.assertEqual(
+            json.loads(
+                call_command("completion", "django_typer_tests_polls", option=test_app)
+            ),
+            {
+                "django_apps": ["django_typer_tests_polls"],
+                "option": "django_typer_tests_test_app",
+            },
+        )
+
+        self.assertEqual(
+            json.loads(
+                call_command(
+                    "completion",
+                    "django_typer_tests_polls",
+                    option="django_typer_tests_test_app",
+                )
+            ),
+            {
+                "django_apps": ["django_typer_tests_polls"],
+                "option": "django_typer_tests_test_app",
+            },
+        )
+
+        self.assertEqual(
+            json.loads(
+                call_command(
+                    "completion",
+                    "django_typer_tests_polls",
+                    "--option=django_typer_tests_test_app",
+                )
+            ),
+            {
+                "django_apps": ["django_typer_tests_polls"],
+                "option": "django_typer_tests_test_app",
+            },
+        )
+
     def test_char_field(self):
         result = StringIO()
         with contextlib.redirect_stdout(result):