mirror of
https://github.com/django/django.git
synced 2024-12-25 18:46:22 +00:00
742ed9878e
Got rid of AppConfig._stub. As a side effect, app_cache.app_configs now only contains entries for applications that are in INSTALLED_APPS, which is a good thing and will allow dramatic simplifications (which I will perform in the next commit). That required adjusting all methods that iterate on app_configs without checking the "installed" flag, hence the large changes in get_model[s]. Introduced AppCache.all_models to store models: - while the app cache is being populated and a suitable app config object to register models isn't available yet; - for applications that aren't in INSTALLED_APPS since they don't have an app config any longer. Replaced get_model(seed_cache=False) by registered_model() which can be kept simple and safe to call at any time, and removed the seed_cache argument to get_model[s]. There's no replacement for that private API. Allowed non-master app caches to go through populate() as it is now safe to do so. They were introduced in 1.7 so backwards compatibility isn't a concern as long as the migrations framework keeps working.
127 lines
5.3 KiB
Python
127 lines
5.3 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.conf import settings
|
|
from django.core.apps import app_cache
|
|
from django.db import connection
|
|
from django.core.management.color import no_style
|
|
from django.test import TestCase, skipIfDBFeature, skipUnlessDBFeature
|
|
|
|
from .models import Article, ArticleRef, Authors, Reviewers, Scientist, ScientistRef
|
|
|
|
# We can't test the DEFAULT_TABLESPACE and DEFAULT_INDEX_TABLESPACE settings
|
|
# because they're evaluated when the model class is defined. As a consequence,
|
|
# @override_settings doesn't work, and the tests depend
|
|
|
|
|
|
def sql_for_table(model):
|
|
return '\n'.join(connection.creation.sql_create_model(model, no_style())[0])
|
|
|
|
|
|
def sql_for_index(model):
|
|
return '\n'.join(connection.creation.sql_indexes_for_model(model, no_style()))
|
|
|
|
|
|
class TablespacesTests(TestCase):
|
|
|
|
def setUp(self):
|
|
# The unmanaged models need to be removed after the test in order to
|
|
# prevent bad interactions with the flush operation in other tests.
|
|
self._old_models = app_cache.app_configs['tablespaces'].models.copy()
|
|
|
|
for model in Article, Authors, Reviewers, Scientist:
|
|
model._meta.managed = True
|
|
|
|
def tearDown(self):
|
|
for model in Article, Authors, Reviewers, Scientist:
|
|
model._meta.managed = False
|
|
|
|
app_cache.app_configs['tablespaces'].models = self._old_models
|
|
app_cache.all_models['tablespaces'] = self._old_models
|
|
app_cache._get_models_cache = {}
|
|
|
|
def assertNumContains(self, haystack, needle, count):
|
|
real_count = haystack.count(needle)
|
|
self.assertEqual(real_count, count, "Found %d instances of '%s', "
|
|
"expected %d" % (real_count, needle, count))
|
|
|
|
@skipUnlessDBFeature('supports_tablespaces')
|
|
def test_tablespace_for_model(self):
|
|
sql = sql_for_table(Scientist).lower()
|
|
if settings.DEFAULT_INDEX_TABLESPACE:
|
|
# 1 for the table
|
|
self.assertNumContains(sql, 'tbl_tbsp', 1)
|
|
# 1 for the index on the primary key
|
|
self.assertNumContains(sql, settings.DEFAULT_INDEX_TABLESPACE, 1)
|
|
else:
|
|
# 1 for the table + 1 for the index on the primary key
|
|
self.assertNumContains(sql, 'tbl_tbsp', 2)
|
|
|
|
@skipIfDBFeature('supports_tablespaces')
|
|
def test_tablespace_ignored_for_model(self):
|
|
# No tablespace-related SQL
|
|
self.assertEqual(sql_for_table(Scientist),
|
|
sql_for_table(ScientistRef))
|
|
|
|
@skipUnlessDBFeature('supports_tablespaces')
|
|
def test_tablespace_for_indexed_field(self):
|
|
sql = sql_for_table(Article).lower()
|
|
if settings.DEFAULT_INDEX_TABLESPACE:
|
|
# 1 for the table
|
|
self.assertNumContains(sql, 'tbl_tbsp', 1)
|
|
# 1 for the primary key + 1 for the index on code
|
|
self.assertNumContains(sql, settings.DEFAULT_INDEX_TABLESPACE, 2)
|
|
else:
|
|
# 1 for the table + 1 for the primary key + 1 for the index on code
|
|
self.assertNumContains(sql, 'tbl_tbsp', 3)
|
|
|
|
# 1 for the index on reference
|
|
self.assertNumContains(sql, 'idx_tbsp', 1)
|
|
|
|
@skipIfDBFeature('supports_tablespaces')
|
|
def test_tablespace_ignored_for_indexed_field(self):
|
|
# No tablespace-related SQL
|
|
self.assertEqual(sql_for_table(Article),
|
|
sql_for_table(ArticleRef))
|
|
|
|
@skipUnlessDBFeature('supports_tablespaces')
|
|
def test_tablespace_for_many_to_many_field(self):
|
|
sql = sql_for_table(Authors).lower()
|
|
# The join table of the ManyToManyField goes to the model's tablespace,
|
|
# and its indexes too, unless DEFAULT_INDEX_TABLESPACE is set.
|
|
if settings.DEFAULT_INDEX_TABLESPACE:
|
|
# 1 for the table
|
|
self.assertNumContains(sql, 'tbl_tbsp', 1)
|
|
# 1 for the primary key
|
|
self.assertNumContains(sql, settings.DEFAULT_INDEX_TABLESPACE, 1)
|
|
else:
|
|
# 1 for the table + 1 for the index on the primary key
|
|
self.assertNumContains(sql, 'tbl_tbsp', 2)
|
|
self.assertNumContains(sql, 'idx_tbsp', 0)
|
|
|
|
sql = sql_for_index(Authors).lower()
|
|
# The ManyToManyField declares no db_tablespace, its indexes go to
|
|
# the model's tablespace, unless DEFAULT_INDEX_TABLESPACE is set.
|
|
if settings.DEFAULT_INDEX_TABLESPACE:
|
|
self.assertNumContains(sql, settings.DEFAULT_INDEX_TABLESPACE, 2)
|
|
else:
|
|
self.assertNumContains(sql, 'tbl_tbsp', 2)
|
|
self.assertNumContains(sql, 'idx_tbsp', 0)
|
|
|
|
sql = sql_for_table(Reviewers).lower()
|
|
# The join table of the ManyToManyField goes to the model's tablespace,
|
|
# and its indexes too, unless DEFAULT_INDEX_TABLESPACE is set.
|
|
if settings.DEFAULT_INDEX_TABLESPACE:
|
|
# 1 for the table
|
|
self.assertNumContains(sql, 'tbl_tbsp', 1)
|
|
# 1 for the primary key
|
|
self.assertNumContains(sql, settings.DEFAULT_INDEX_TABLESPACE, 1)
|
|
else:
|
|
# 1 for the table + 1 for the index on the primary key
|
|
self.assertNumContains(sql, 'tbl_tbsp', 2)
|
|
self.assertNumContains(sql, 'idx_tbsp', 0)
|
|
|
|
sql = sql_for_index(Reviewers).lower()
|
|
# The ManyToManyField declares db_tablespace, its indexes go there.
|
|
self.assertNumContains(sql, 'tbl_tbsp', 0)
|
|
self.assertNumContains(sql, 'idx_tbsp', 2)
|