136 lines
5.3 KiB
Python
136 lines
5.3 KiB
Python
from django.apps import apps
|
|
from django.conf import settings
|
|
from django.db import connection
|
|
from django.test import TransactionTestCase, skipIfDBFeature, skipUnlessDBFeature
|
|
|
|
from .models.tablespaces import (
|
|
Article,
|
|
ArticleRef,
|
|
Authors,
|
|
Reviewers,
|
|
Scientist,
|
|
ScientistRef,
|
|
)
|
|
|
|
|
|
def sql_for_table(model):
|
|
with connection.schema_editor(collect_sql=True) as editor:
|
|
editor.create_model(model)
|
|
return editor.collected_sql[0]
|
|
|
|
|
|
def sql_for_index(model):
|
|
return "\n".join(
|
|
str(sql) for sql in connection.schema_editor()._model_indexes_sql(model)
|
|
)
|
|
|
|
|
|
# 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
|
|
class TablespacesTests(TransactionTestCase):
|
|
available_apps = ["model_options"]
|
|
|
|
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 = apps.app_configs["model_options"].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
|
|
|
|
apps.app_configs["model_options"].models = self._old_models
|
|
apps.all_models["model_options"] = self._old_models
|
|
apps.clear_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)
|