Reused a duplicated class in admin_scripts tests.
@@ -1459,6 +1459,13 @@ class ManageTestserver(AdminScriptTestCase):
# user-space commands are correctly handled - in particular, arguments to
# the commands are correctly parsed and processed.
##########################################################################
+class ColorCommand(BaseCommand):
+ requires_system_checks = False
+
+ def handle(self, *args, **options):
+ self.stdout.write('Hello, world!', self.style.ERROR)
+ self.stderr.write('Hello, world!', self.style.ERROR)
+
class CommandTypes(AdminScriptTestCase):
"Tests for the various types of base command types that can be defined."
@@ -1542,16 +1549,9 @@ class CommandTypes(AdminScriptTestCase):
self.assertNotEqual(style.ERROR('Hello, world!'), 'Hello, world!')
def test_command_color(self):
- class Command(BaseCommand):
- requires_system_checks = False
-
- def handle(self, *args, **options):
- self.stdout.write('Hello, world!', self.style.ERROR)
- self.stderr.write('Hello, world!', self.style.ERROR)
-
out = StringIO()
err = StringIO()
- command = Command(stdout=out, stderr=err)
+ command = ColorCommand(stdout=out, stderr=err)
call_command(command)
if color.supports_color():
self.assertIn('Hello, world!\n', out.getvalue())
@@ -1564,23 +1564,16 @@ class CommandTypes(AdminScriptTestCase):
def test_command_no_color(self):
"--no-color prevent colorization of the output"
- class Command(BaseCommand):
- requires_system_checks = False
-
- def handle(self, *args, **options):
- self.stdout.write('Hello, world!', self.style.ERROR)
- self.stderr.write('Hello, world!', self.style.ERROR)
-
out = StringIO()
err = StringIO()
- command = Command(stdout=out, stderr=err, no_color=True)
+ command = ColorCommand(stdout=out, stderr=err, no_color=True)
call_command(command)
self.assertEqual(out.getvalue(), 'Hello, world!\n')
self.assertEqual(err.getvalue(), 'Hello, world!\n')
out = StringIO()
err = StringIO()
- command = Command(stdout=out, stderr=err)
+ command = ColorCommand(stdout=out, stderr=err)
call_command(command, no_color=True)
self.assertEqual(out.getvalue(), 'Hello, world!\n')
self.assertEqual(err.getvalue(), 'Hello, world!\n')
