Fixed #23303 -- Added BEGIN and COMMIT statements to the output of sqlmigrate.

This commit is contained in:
Baptiste Mispelon 2014-08-16 17:21:14 +02:00
parent b6aa60f425
commit 5853c87a45
7 changed files with 38 additions and 2 deletions

View file

@ -0,0 +1,10 @@
from django.core.management.base import BaseCommand
class Command(BaseCommand):
help = "Say hello."
args = ''
output_transaction = True
def handle(self, *args, **options):
return 'Hello!'

View file

@ -134,6 +134,11 @@ class CommandTests(SimpleTestCase):
with self.assertRaises(CommandError):
management.call_command('hal', stdout=out)
def test_output_transaction(self):
out = StringIO()
management.call_command('transaction', stdout=out, no_color=True)
self.assertEqual(out.getvalue(), 'BEGIN;\nHello!\n\nCOMMIT;\n')
class UtilsTests(SimpleTestCase):