Merge pull request #646 from maykinmedia/feature/573-Add-tests-for-generate_token.py

✅ [#573] Add tests for generate_token.py

Author: OlhaZahoruiko

pyproject.toml

@@ -147,4 +147,9 @@ section-order = [
 line-ending = "lf"
 
 [tool.ruff.lint.flake8-tidy-imports.banned-api]
-"logging".msg = "Use `structlog.stdlib.get_logger(__name__)` instead of the stdlib logging module."
+"logging".msg = "Use `structlog.stdlib.get_logger(__name__)` instead of the stdlib logging module."
+
+[tool.coverage.run]
+branch = true
+source = ["src"]
+omit = ["*/test_*.py"]

requirements/ci.txt

@@ -117,7 +117,7 @@ commonground-api-common==2.7.0
     #   -r requirements/base.txt
 commonmark==0.9.1
     # via recommonmark
-coverage==4.5.4
+coverage==7.10.1
     # via
     #   -r requirements/test-tools.in
     #   codecov

requirements/dev.txt

@@ -148,7 +148,7 @@ commonmark==0.9.1
     #   -c requirements/ci.txt
     #   -r requirements/ci.txt
     #   recommonmark
-coverage==4.5.4
+coverage==7.10.1
     # via
     #   -c requirements/ci.txt
     #   -r requirements/ci.txt

requirements/test-tools.in

@@ -5,7 +5,7 @@
 codecov
 pytest
 pytest-benchmark
-coverage < 5.0
+coverage
 django-webtest
 factory-boy
 freezegun
@@ -15,4 +15,4 @@ tblib
 vcrpy
 
 # Code formatting
-ruff
+ruff

src/objects/token/tests/test_generate_token.py

@@ -0,0 +1,69 @@
+from io import StringIO
+
+from django.core.management import CommandError, call_command
+from django.test import TestCase
+
+from objects.token.models import TokenAuth
+
+
+class GenerateTokenCommandTests(TestCase):
+    def test_token_created_with_required_args(self):
+        out = StringIO()
+        contact_person = "John Doe"
+        email = "[email protected]"
+
+        call_command(
+            "generate_token",
+            contact_person,
+            email,
+            stdout=out,
+        )
+
+        self.assertEqual(TokenAuth.objects.count(), 1)
+        token = TokenAuth.objects.first()
+        self.assertEqual(token.contact_person, contact_person)
+        self.assertEqual(token.email, email)
+        self.assertEqual(token.organization, "")
+        self.assertIn("Token", out.getvalue())
+
+    def test_token_created_with_all_optional_args(self):
+        out = StringIO()
+        call_command(
+            "generate_token",
+            "Alice Smith",
+            "[email protected]",
+            "--organization",
+            "Acme Inc",
+            "--application",
+            "WeatherApp",
+            "--administration",
+            "City Council",
+            stdout=out,
+        )
+
+        token = TokenAuth.objects.get(email="[email protected]")
+        self.assertEqual(token.contact_person, "Alice Smith")
+        self.assertEqual(token.organization, "Acme Inc")
+        self.assertEqual(token.application, "WeatherApp")
+        self.assertEqual(token.administration, "City Council")
+        self.assertIn("Token", out.getvalue())
+
+    def test_missing_required_arguments(self):
+        with self.assertRaises(CommandError):
+            call_command("generate_token")
+
+    def test_stdout_contains_token(self):
+        out = StringIO()
+        call_command("generate_token", "Bob", "[email protected]", stdout=out)
+        output = out.getvalue()
+        self.assertIn("Token", output)
+        self.assertIn(TokenAuth.objects.first().token, output)
+
+    def test_duplicate_email_allowed_with_unique_identifier(self):
+        TokenAuth.objects.create(
+            contact_person="Bob", email="[email protected]", identifier="unique-1"
+        )
+        out = StringIO()
+        call_command("generate_token", "Bob", "[email protected]", stdout=out)
+
+        self.assertEqual(TokenAuth.objects.filter(email="[email protected]").count(), 2)