mirror of
https://github.com/django/django.git
synced 2024-11-19 16:04:13 +00:00
423244bc6b
In particular, allow the '--' sequence to be present in string values without being interpreted as comment marker. Thanks Tim Chase for the report and shaleh for the initial patch.
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from django.test import TestCase
|
|
|
|
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):
|
|
from django.core.management.sql import custom_sql_for_model
|
|
from django.core.management.color import no_style
|
|
from django.db import connections, DEFAULT_DB_ALIAS
|
|
|
|
# Simulate the custom SQL loading by syncdb
|
|
connection = connections[DEFAULT_DB_ALIAS]
|
|
custom_sql = custom_sql_for_model(Simple, no_style(), connection)
|
|
self.assertEqual(len(custom_sql), 8)
|
|
cursor = connection.cursor()
|
|
for sql in custom_sql:
|
|
cursor.execute(sql)
|
|
self.assertEqual(Simple.objects.count(), 8)
|