59 lines
2.6 KiB
Python
59 lines
2.6 KiB
Python
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 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])
|
|
self.assertTrue(output[0].startswith('CREATE TABLE'))
|
|
if connections[DEFAULT_DB_ALIAS].vendor == 'oracle':
|
|
self.assertEqual(len(output), 4) # Oracle creates a table, a sequence, a trigger and an index
|
|
self.assertIn('CREATE SEQUENCE', output[1])
|
|
self.assertIn('CREATE OR REPLACE TRIGGER', output[2])
|
|
self.assertTrue(output[3].startswith('CREATE INDEX'))
|
|
else:
|
|
# PostgreSQL creates two indexes
|
|
self.assertIn(len(output), [2, 3])
|
|
self.assertTrue(output[1].startswith('CREATE INDEX'))
|