46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
from django.core.management.color import no_style
|
|
from django.core.management.sql import custom_sql_for_model
|
|
from django.db import connections, DEFAULT_DB_ALIAS
|
|
from django.test import TestCase, override_settings
|
|
|
|
from .models import Simple
|
|
|
|
|
|
class InitialSQLTests(TestCase):
|
|
"""
|
|
The format of the included SQL file for this test suite is important.
|
|
It must end with a trailing newline in order to test the fix for #2161.
|
|
"""
|
|
|
|
def test_initial_sql(self):
|
|
"""
|
|
As pointed out by #14661, test data loaded by custom SQL
|
|
can't be relied upon; as a result, the test framework flushes the
|
|
data contents before every test. This test validates that this has
|
|
occurred.
|
|
"""
|
|
self.assertEqual(Simple.objects.count(), 0)
|
|
|
|
def test_custom_sql(self):
|
|
"""
|
|
Simulate the custom SQL loading by migrate.
|
|
"""
|
|
connection = connections[DEFAULT_DB_ALIAS]
|
|
custom_sql = custom_sql_for_model(Simple, no_style(), connection)
|
|
with connection.cursor() as cursor:
|
|
for sql in custom_sql:
|
|
cursor.execute(sql)
|
|
self.assertEqual(Simple.objects.count(), 9)
|
|
self.assertEqual(
|
|
Simple.objects.get(name__contains='placeholders').name,
|
|
'"100%" of % are not placeholders'
|
|
)
|
|
|
|
@override_settings(DEBUG=True)
|
|
def test_custom_sql_debug(self):
|
|
"""
|
|
Same test, ensure that CursorDebugWrapper doesn't alter sql loading
|
|
(#3485).
|
|
"""
|
|
self.test_custom_sql()
|