Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Fixed a test that depended on the DB backend; refs #19877. Thanks Loic.

  • Loading branch information...
commit dffda2ba4e55d4567b3a829d410dae794636ebdd 1 parent 5b47a9c
@timgraham timgraham authored
View
9 tests/admin_scripts/management/commands/color_command.py
@@ -0,0 +1,9 @@
+from django.core.management.base import NoArgsCommand
+
+
+class Command(NoArgsCommand):
+ help = "Test color output"
+ requires_model_validation = False
+
+ def handle_noargs(self, **options):
+ return self.style.SQL_KEYWORD('BEGIN')
View
18 tests/admin_scripts/tests.py
@@ -1282,17 +1282,13 @@ def test_specific_help(self):
def test_no_color(self):
"--no-color prevent colorization of the output"
out = StringIO()
- call_command("sqlall", "admin_scripts", no_color=True, stdout=out)
- self.assertEqual(out.getvalue(), """BEGIN;
-CREATE TABLE "admin_scripts_article" (
- "id" integer NOT NULL PRIMARY KEY,
- "headline" varchar(100) NOT NULL,
- "pub_date" datetime NOT NULL
-)
-;
-
-COMMIT;
-""")
+
+ call_command('color_command', no_color=True, stdout=out)
+ self.assertEqual(out.getvalue(), 'BEGIN\n')
+
+ out = StringIO()
+ call_command('color_command', stdout=out)
+ self.assertEqual(out.getvalue(), '\x1b[33mBEGIN\x1b[0m\n')
def test_base_command(self):
"User BaseCommands can execute when a label is provided"
Please sign in to comment.
Something went wrong with that request. Please try again.