1 files changed, 52 insertions, 0 deletions
diff --git a/tests/commands_sql/tests.py b/tests/commands_sql/tests.py
new file mode 100644
index 0000000000..949032ae6f
--- /dev/null
+++ b/tests/commands_sql/tests.py
@@ -0,0 +1,52 @@
+from __future__ import unicode_literals
+
+from django.core.management.color import no_style
+from django.core.management.sql import (sql_create, sql_delete, sql_indexes,
+    sql_destroy_indexes, sql_all)
+from django.db import connections, DEFAULT_DB_ALIAS, models
+from django.test import TestCase
+from django.utils import six
+
+# See also regressiontests/initial_sql_regress for 'custom_sql_for_model' tests
+
+
+class SQLCommandsTestCase(TestCase):
+    """Tests for several functions in django/core/management/sql.py"""
+    def test_sql_create(self):
+        app = models.get_app('commands_sql')
+        output = sql_create(app, no_style(), connections[DEFAULT_DB_ALIAS])
+        # Lower so that Oracle's upper case tbl names wont break
+        sql = output[0].lower()
+        six.assertRegex(self, sql, r'^create table .commands_sql_book.*')
+
+    def test_sql_delete(self):
+        app = models.get_app('commands_sql')
+        output = sql_delete(app, no_style(), connections[DEFAULT_DB_ALIAS])
+        # Oracle produces DROP SEQUENCE and DROP TABLE for this command.
+        if connections[DEFAULT_DB_ALIAS].vendor == 'oracle':
+            sql = output[1].lower()
+        else:
+            sql = output[0].lower()
+        six.assertRegex(self, sql, r'^drop table .commands_sql_book.*')
+
+    def test_sql_indexes(self):
+        app = models.get_app('commands_sql')
+        output = sql_indexes(app, no_style(), connections[DEFAULT_DB_ALIAS])
+        # PostgreSQL creates two indexes
+        self.assertIn(len(output), [1, 2])
+        self.assertTrue(output[0].startswith("CREATE INDEX"))
+
+    def test_sql_destroy_indexes(self):
+        app = models.get_app('commands_sql')
+        output = sql_destroy_indexes(app, no_style(), connections[DEFAULT_DB_ALIAS])
+        # PostgreSQL creates two indexes
+        self.assertIn(len(output), [1, 2])
+        self.assertTrue(output[0].startswith("DROP INDEX"))
+
+    def test_sql_all(self):
+        app = models.get_app('commands_sql')
+        output = sql_all(app, no_style(), connections[DEFAULT_DB_ALIAS])
+        # PostgreSQL creates two indexes
+        self.assertIn(len(output), [2, 3])
+        self.assertTrue(output[0].startswith('CREATE TABLE'))
+        self.assertTrue(output[1].startswith('CREATE INDEX'))