mirror of
https://github.com/django/django.git
synced 2025-07-22 20:55:17 +00:00
Added tests for some sql_* management commands
This commit is contained in:
parent
d93edffa89
commit
718afcafc2
3 changed files with 45 additions and 0 deletions
38
tests/regressiontests/commands_sql/tests.py
Normal file
38
tests/regressiontests/commands_sql/tests.py
Normal file
|
@ -0,0 +1,38 @@
|
|||
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_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])
|
||||
six.assertRegex(self, output[0], 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])
|
||||
six.assertRegex(self, output[0], 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_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'))
|
Loading…
Add table
Add a link
Reference in a new issue