From dffda2ba4e55d4567b3a829d410dae794636ebdd Mon Sep 17 00:00:00 2001 From: Tim Graham Date: Tue, 30 Jul 2013 11:30:20 -0400 Subject: Fixed a test that depended on the DB backend; refs #19877. Thanks Loic. --- tests/admin_scripts/tests.py | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) (limited to 'tests/admin_scripts/tests.py') diff --git a/tests/admin_scripts/tests.py b/tests/admin_scripts/tests.py index 68615b787e..e9863ebd94 100644 --- a/tests/admin_scripts/tests.py +++ b/tests/admin_scripts/tests.py @@ -1282,17 +1282,13 @@ class CommandTypes(AdminScriptTestCase): 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" -- cgit v1.3