diff options
Diffstat (limited to 'tests/model_options/test_tablespaces.py')
| -rw-r--r-- | tests/model_options/test_tablespaces.py | 75 |
1 files changed, 41 insertions, 34 deletions
diff --git a/tests/model_options/test_tablespaces.py b/tests/model_options/test_tablespaces.py index 8502d8790f..0aa2e0fccf 100644 --- a/tests/model_options/test_tablespaces.py +++ b/tests/model_options/test_tablespaces.py @@ -1,12 +1,15 @@ from django.apps import apps from django.conf import settings from django.db import connection -from django.test import ( - TransactionTestCase, skipIfDBFeature, skipUnlessDBFeature, -) +from django.test import TransactionTestCase, skipIfDBFeature, skipUnlessDBFeature from .models.tablespaces import ( - Article, ArticleRef, Authors, Reviewers, Scientist, ScientistRef, + Article, + ArticleRef, + Authors, + Reviewers, + Scientist, + ScientistRef, ) @@ -17,19 +20,21 @@ def sql_for_table(model): def sql_for_index(model): - return '\n'.join(str(sql) for sql in connection.schema_editor()._model_indexes_sql(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'] + 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() + self._old_models = apps.app_configs["model_options"].models.copy() for model in Article, Authors, Reviewers, Scientist: model._meta.managed = True @@ -38,67 +43,69 @@ class TablespacesTests(TransactionTestCase): 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.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)) + self.assertEqual( + real_count, + count, + "Found %d instances of '%s', expected %d" % (real_count, needle, count), + ) - @skipUnlessDBFeature('supports_tablespaces') + @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) + 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) + self.assertNumContains(sql, "tbl_tbsp", 2) - @skipIfDBFeature('supports_tablespaces') + @skipIfDBFeature("supports_tablespaces") def test_tablespace_ignored_for_model(self): # No tablespace-related SQL - self.assertEqual(sql_for_table(Scientist), - sql_for_table(ScientistRef)) + self.assertEqual(sql_for_table(Scientist), sql_for_table(ScientistRef)) - @skipUnlessDBFeature('supports_tablespaces') + @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) + 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) + self.assertNumContains(sql, "tbl_tbsp", 3) # 1 for the index on reference - self.assertNumContains(sql, 'idx_tbsp', 1) + self.assertNumContains(sql, "idx_tbsp", 1) - @skipIfDBFeature('supports_tablespaces') + @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)) + self.assertEqual(sql_for_table(Article), sql_for_table(ArticleRef)) - @skipUnlessDBFeature('supports_tablespaces') + @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) + 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) + 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 @@ -106,23 +113,23 @@ class TablespacesTests(TransactionTestCase): 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) + 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) + 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) + 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) + self.assertNumContains(sql, "tbl_tbsp", 0) + self.assertNumContains(sql, "idx_tbsp", 2) |