fix v5 tests

django-commons · Dec 26, 2023 · 9a430bd · 9a430bd
1 parent e9c0520
commit 9a430bd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/django_typer/tests/tests.py b/django_typer/tests/tests.py
@@ -78,7 +78,7 @@ def test_call_command_stdout(self):
 
     def test_get_version(self):
         self.assertEqual(
-            run_command("basic", "--version").strip(), django.get_version()
+            str(run_command("basic", "--version")).strip(), django.get_version()
         )
 
     def test_call_direct(self):
@@ -183,16 +183,16 @@ def test_call_command_stdout(self):
 
     def test_get_version(self):
         self.assertEqual(
-            run_command("multi", "--version").strip(), django.get_version()
+            str(run_command("multi", "--version")).strip(), django.get_version()
         )
         self.assertEqual(
-            run_command("multi", "cmd1", "--version").strip(), django.get_version()
+            str(run_command("multi", "cmd1", "--version")).strip(), django.get_version()
         )
         self.assertEqual(
-            run_command("multi", "sum", "--version").strip(), django.get_version()
+            str(run_command("multi", "sum", "--version")).strip(), django.get_version()
         )
         self.assertEqual(
-            run_command("multi", "cmd3", "--version").strip(), django.get_version()
+            str(run_command("multi", "cmd3", "--version")).strip(), django.get_version()
         )
 
     def test_call_direct(self):
@@ -439,10 +439,10 @@ def test_call_command_stdout(self):
 
     def test_get_version(self):
         self.assertEqual(
-            run_command(self.cmd_name, "--version").strip(), django.get_version()
+            str(run_command(self.cmd_name, "--version")).strip(), django.get_version()
         )
         self.assertEqual(
-            run_command(self.cmd_name, "6", self.cmd_name, "--version").strip(),
+            str(run_command(self.cmd_name, "6", self.cmd_name, "--version")).strip(),
             django.get_version(),
         )
 

diff --git a/django_typer/types.py b/django_typer/types.py
@@ -17,7 +17,9 @@ def print_version(context, _, value):
     command when --version is specified.
     """
     if value:
-        echo(context.django_command.get_version())
+        context.django_command.stdout.write(
+            context.django_command.get_version()
+        )
         sys.exit()