From 2987cdfbb29fb740479c06eecb9c1e1ea5f58452 Mon Sep 17 00:00:00 2001 From: Matthias Mair Date: Tue, 8 Apr 2025 23:37:28 +0200 Subject: [PATCH] feat(backend): test custom command (#9479) --- .../InvenTree/management/commands/schema.py | 2 ++ src/backend/InvenTree/InvenTree/test_commands.py | 13 +++++++++++++ 2 files changed, 15 insertions(+) create mode 100644 src/backend/InvenTree/InvenTree/test_commands.py diff --git a/src/backend/InvenTree/InvenTree/management/commands/schema.py b/src/backend/InvenTree/InvenTree/management/commands/schema.py index ceff5c4f40..25c9cc38c2 100644 --- a/src/backend/InvenTree/InvenTree/management/commands/schema.py +++ b/src/backend/InvenTree/InvenTree/management/commands/schema.py @@ -108,3 +108,5 @@ class Command(spectacular.Command): settings.SPECTACULAR_SETTINGS['APPEND_COMPONENTS'] = components super().handle(*args, **kwargs) + + return 'done' diff --git a/src/backend/InvenTree/InvenTree/test_commands.py b/src/backend/InvenTree/InvenTree/test_commands.py new file mode 100644 index 0000000000..955a70c062 --- /dev/null +++ b/src/backend/InvenTree/InvenTree/test_commands.py @@ -0,0 +1,13 @@ +"""Testts for custom InvenTree management commands.""" + +from django.core.management import call_command +from django.test import TestCase + + +class CommandTestCase(TestCase): + """Test case for custom management commands.""" + + def test_schema(self): + """Test the schema generation command.""" + output = call_command('schema', verbosity=0) + self.assertEqual(output, 'done')