mirror of
https://github.com/django/django.git
synced 2024-12-24 01:55:49 +00:00
75410228df
The sql_flush() positional argument sequences is replaced by the boolean keyword-only argument reset_sequences. This ensures that the old function signature can't be used by mistake when upgrading Django. When the new argument is True, the sequences of the truncated tables will reset. Using a single boolean value, rather than a list, allows making a binary yes/no choice as to whether to reset all sequences rather than a working on a completely different set.
140 lines
5.0 KiB
Python
140 lines
5.0 KiB
Python
import unittest
|
|
|
|
from django.core.management.color import no_style
|
|
from django.db import connection
|
|
|
|
from ..models import Person, Tag
|
|
|
|
|
|
@unittest.skipUnless(connection.vendor == 'oracle', 'Oracle tests')
|
|
class OperationsTests(unittest.TestCase):
|
|
|
|
def test_sequence_name_truncation(self):
|
|
seq_name = connection.ops._get_no_autofield_sequence_name('schema_authorwithevenlongee869')
|
|
self.assertEqual(seq_name, 'SCHEMA_AUTHORWITHEVENLOB0B8_SQ')
|
|
|
|
def test_bulk_batch_size(self):
|
|
# Oracle restricts the number of parameters in a query.
|
|
objects = range(2**16)
|
|
self.assertEqual(connection.ops.bulk_batch_size([], objects), len(objects))
|
|
# Each field is a parameter for each object.
|
|
self.assertEqual(
|
|
connection.ops.bulk_batch_size(['id'], objects),
|
|
connection.features.max_query_params,
|
|
)
|
|
self.assertEqual(
|
|
connection.ops.bulk_batch_size(['id', 'other'], objects),
|
|
connection.features.max_query_params // 2,
|
|
)
|
|
|
|
def test_sql_flush(self):
|
|
statements = connection.ops.sql_flush(
|
|
no_style(),
|
|
[Person._meta.db_table, Tag._meta.db_table],
|
|
)
|
|
# The tables and constraints are processed in an unordered set.
|
|
self.assertEqual(
|
|
statements[0],
|
|
'ALTER TABLE "BACKENDS_TAG" DISABLE CONSTRAINT '
|
|
'"BACKENDS__CONTENT_T_FD9D7A85_F" KEEP INDEX;',
|
|
)
|
|
self.assertEqual(
|
|
sorted(statements[1:-1]),
|
|
[
|
|
'TRUNCATE TABLE "BACKENDS_PERSON";',
|
|
'TRUNCATE TABLE "BACKENDS_TAG";',
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
statements[-1],
|
|
'ALTER TABLE "BACKENDS_TAG" ENABLE CONSTRAINT '
|
|
'"BACKENDS__CONTENT_T_FD9D7A85_F";',
|
|
)
|
|
|
|
def test_sql_flush_allow_cascade(self):
|
|
statements = connection.ops.sql_flush(
|
|
no_style(),
|
|
[Person._meta.db_table, Tag._meta.db_table],
|
|
allow_cascade=True,
|
|
)
|
|
# The tables and constraints are processed in an unordered set.
|
|
self.assertEqual(
|
|
statements[0],
|
|
'ALTER TABLE "BACKENDS_VERYLONGMODELNAME540F" DISABLE CONSTRAINT '
|
|
'"BACKENDS__PERSON_ID_1DD5E829_F" KEEP INDEX;',
|
|
)
|
|
self.assertEqual(
|
|
sorted(statements[1:-1]),
|
|
[
|
|
'TRUNCATE TABLE "BACKENDS_PERSON";',
|
|
'TRUNCATE TABLE "BACKENDS_TAG";',
|
|
'TRUNCATE TABLE "BACKENDS_VERYLONGMODELNAME540F";',
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
statements[-1],
|
|
'ALTER TABLE "BACKENDS_VERYLONGMODELNAME540F" ENABLE CONSTRAINT '
|
|
'"BACKENDS__PERSON_ID_1DD5E829_F";',
|
|
)
|
|
|
|
def test_sql_flush_sequences(self):
|
|
statements = connection.ops.sql_flush(
|
|
no_style(),
|
|
[Person._meta.db_table, Tag._meta.db_table],
|
|
reset_sequences=True,
|
|
)
|
|
# The tables and constraints are processed in an unordered set.
|
|
self.assertEqual(
|
|
statements[0],
|
|
'ALTER TABLE "BACKENDS_TAG" DISABLE CONSTRAINT '
|
|
'"BACKENDS__CONTENT_T_FD9D7A85_F" KEEP INDEX;',
|
|
)
|
|
self.assertEqual(
|
|
sorted(statements[1:3]),
|
|
[
|
|
'TRUNCATE TABLE "BACKENDS_PERSON";',
|
|
'TRUNCATE TABLE "BACKENDS_TAG";',
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
statements[3],
|
|
'ALTER TABLE "BACKENDS_TAG" ENABLE CONSTRAINT '
|
|
'"BACKENDS__CONTENT_T_FD9D7A85_F";',
|
|
)
|
|
# Sequences.
|
|
self.assertEqual(len(statements[4:]), 2)
|
|
self.assertIn('BACKENDS_PERSON_SQ', statements[4])
|
|
self.assertIn('BACKENDS_TAG_SQ', statements[5])
|
|
|
|
def test_sql_flush_sequences_allow_cascade(self):
|
|
statements = connection.ops.sql_flush(
|
|
no_style(),
|
|
[Person._meta.db_table, Tag._meta.db_table],
|
|
reset_sequences=True,
|
|
allow_cascade=True,
|
|
)
|
|
# The tables and constraints are processed in an unordered set.
|
|
self.assertEqual(
|
|
statements[0],
|
|
'ALTER TABLE "BACKENDS_VERYLONGMODELNAME540F" DISABLE CONSTRAINT '
|
|
'"BACKENDS__PERSON_ID_1DD5E829_F" KEEP INDEX;',
|
|
)
|
|
self.assertEqual(
|
|
sorted(statements[1:4]),
|
|
[
|
|
'TRUNCATE TABLE "BACKENDS_PERSON";',
|
|
'TRUNCATE TABLE "BACKENDS_TAG";',
|
|
'TRUNCATE TABLE "BACKENDS_VERYLONGMODELNAME540F";',
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
statements[4],
|
|
'ALTER TABLE "BACKENDS_VERYLONGMODELNAME540F" ENABLE CONSTRAINT '
|
|
'"BACKENDS__PERSON_ID_1DD5E829_F";',
|
|
)
|
|
# Sequences.
|
|
self.assertEqual(len(statements[5:]), 3)
|
|
self.assertIn('BACKENDS_PERSON_SQ', statements[5])
|
|
self.assertIn('BACKENDS_VERYLONGMODELN7BE2_SQ', statements[6])
|
|
self.assertIn('BACKENDS_TAG_SQ', statements[7])
|