mirror of
https://github.com/django/django.git
synced 2024-12-24 18:16:19 +00:00
d818e0c9b2
This is the result of Christopher Medrela's 2013 Summer of Code project. Thanks also to Preston Holmes, Tim Graham, Anssi Kääriäinen, Florian Apolloner, and Alex Gaynor for review notes along the way. Also: Fixes #8579, fixes #3055, fixes #19844.
132 lines
4.3 KiB
Python
132 lines
4.3 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import warnings
|
|
|
|
from django.core import management
|
|
from django.db import transaction
|
|
from django.test import TestCase, TransactionTestCase
|
|
from django.test.utils import override_system_checks
|
|
from django.utils.six import StringIO
|
|
|
|
from .models import Article, Book
|
|
|
|
|
|
class SampleTestCase(TestCase):
|
|
fixtures = ['fixture1.json', 'fixture2.json']
|
|
|
|
def testClassFixtures(self):
|
|
"Test cases can load fixture objects into models defined in packages"
|
|
self.assertEqual(Article.objects.count(), 3)
|
|
self.assertQuerysetEqual(
|
|
Article.objects.all(), [
|
|
"Django conquers world!",
|
|
"Copyright is fine the way it is",
|
|
"Poker has no place on ESPN",
|
|
],
|
|
lambda a: a.headline
|
|
)
|
|
|
|
|
|
class TestNoInitialDataLoading(TransactionTestCase):
|
|
|
|
available_apps = ['fixtures_model_package']
|
|
|
|
@override_system_checks([])
|
|
def test_migrate(self):
|
|
with transaction.atomic():
|
|
Book.objects.all().delete()
|
|
|
|
management.call_command(
|
|
'migrate',
|
|
verbosity=0,
|
|
load_initial_data=False
|
|
)
|
|
self.assertQuerysetEqual(Book.objects.all(), [])
|
|
|
|
@override_system_checks([])
|
|
def test_flush(self):
|
|
# Test presence of fixture (flush called by TransactionTestCase)
|
|
self.assertQuerysetEqual(
|
|
Book.objects.all(), [
|
|
'Achieving self-awareness of Python programs'
|
|
],
|
|
lambda a: a.name
|
|
)
|
|
|
|
with transaction.atomic():
|
|
management.call_command(
|
|
'flush',
|
|
verbosity=0,
|
|
interactive=False,
|
|
load_initial_data=False
|
|
)
|
|
self.assertQuerysetEqual(Book.objects.all(), [])
|
|
|
|
|
|
class FixtureTestCase(TestCase):
|
|
def test_initial_data(self):
|
|
"Fixtures can load initial data into models defined in packages"
|
|
# migrate introduces 1 initial data object from initial_data.json
|
|
self.assertQuerysetEqual(
|
|
Book.objects.all(), [
|
|
'Achieving self-awareness of Python programs'
|
|
],
|
|
lambda a: a.name
|
|
)
|
|
|
|
def test_loaddata(self):
|
|
"Fixtures can load data into models defined in packages"
|
|
# Load fixture 1. Single JSON file, with two objects
|
|
management.call_command("loaddata", "fixture1.json", verbosity=0)
|
|
self.assertQuerysetEqual(
|
|
Article.objects.all(), [
|
|
"Time to reform copyright",
|
|
"Poker has no place on ESPN",
|
|
],
|
|
lambda a: a.headline,
|
|
)
|
|
|
|
# Load fixture 2. JSON file imported by default. Overwrites some
|
|
# existing objects
|
|
management.call_command("loaddata", "fixture2.json", verbosity=0)
|
|
self.assertQuerysetEqual(
|
|
Article.objects.all(), [
|
|
"Django conquers world!",
|
|
"Copyright is fine the way it is",
|
|
"Poker has no place on ESPN",
|
|
],
|
|
lambda a: a.headline,
|
|
)
|
|
|
|
# Load a fixture that doesn't exist
|
|
with warnings.catch_warnings(record=True) as w:
|
|
warnings.simplefilter("always")
|
|
management.call_command("loaddata", "unknown.json", verbosity=0)
|
|
self.assertEqual(len(w), 1)
|
|
self.assertTrue(w[0].message, "No fixture named 'unknown' found.")
|
|
|
|
self.assertQuerysetEqual(
|
|
Article.objects.all(), [
|
|
"Django conquers world!",
|
|
"Copyright is fine the way it is",
|
|
"Poker has no place on ESPN",
|
|
],
|
|
lambda a: a.headline,
|
|
)
|
|
|
|
|
|
class InitialSQLTests(TestCase):
|
|
|
|
def test_custom_sql(self):
|
|
"""
|
|
#14300 -- Verify that custom_sql_for_model searches `app/sql` and not
|
|
`app/models/sql` (the old location will work until Django 1.9)
|
|
"""
|
|
out = StringIO()
|
|
management.call_command("sqlcustom", "fixtures_model_package", stdout=out)
|
|
output = out.getvalue()
|
|
self.assertTrue("INSERT INTO fixtures_model_package_book (name) "
|
|
"VALUES ('My Book')" in output)
|
|
# value from deprecated search path models/sql (remove in Django 1.9)
|
|
self.assertTrue("Deprecated Book" in output)
|