diff options
Diffstat (limited to 'tests/backends/postgresql/test_creation.py')
| -rw-r--r-- | tests/backends/postgresql/test_creation.py | 57 |
1 files changed, 35 insertions, 22 deletions
diff --git a/tests/backends/postgresql/test_creation.py b/tests/backends/postgresql/test_creation.py index eb68c6e471..319029334d 100644 --- a/tests/backends/postgresql/test_creation.py +++ b/tests/backends/postgresql/test_creation.py @@ -18,12 +18,11 @@ else: from django.db.backends.postgresql.creation import DatabaseCreation -@unittest.skipUnless(connection.vendor == 'postgresql', 'PostgreSQL tests') +@unittest.skipUnless(connection.vendor == "postgresql", "PostgreSQL tests") class DatabaseCreationTests(SimpleTestCase): - @contextmanager def changed_test_settings(self, **kwargs): - settings = connection.settings_dict['TEST'] + settings = connection.settings_dict["TEST"] saved_values = {} for name in kwargs: if name in settings: @@ -47,66 +46,80 @@ class DatabaseCreationTests(SimpleTestCase): self.assertEqual(suffix, expected) def test_sql_table_creation_suffix_with_none_settings(self): - settings = {'CHARSET': None, 'TEMPLATE': None} + settings = {"CHARSET": None, "TEMPLATE": None} self.check_sql_table_creation_suffix(settings, "") def test_sql_table_creation_suffix_with_encoding(self): - settings = {'CHARSET': 'UTF8'} + settings = {"CHARSET": "UTF8"} self.check_sql_table_creation_suffix(settings, "WITH ENCODING 'UTF8'") def test_sql_table_creation_suffix_with_template(self): - settings = {'TEMPLATE': 'template0'} + settings = {"TEMPLATE": "template0"} self.check_sql_table_creation_suffix(settings, 'WITH TEMPLATE "template0"') def test_sql_table_creation_suffix_with_encoding_and_template(self): - settings = {'CHARSET': 'UTF8', 'TEMPLATE': 'template0'} - self.check_sql_table_creation_suffix(settings, '''WITH ENCODING 'UTF8' TEMPLATE "template0"''') + settings = {"CHARSET": "UTF8", "TEMPLATE": "template0"} + self.check_sql_table_creation_suffix( + settings, '''WITH ENCODING 'UTF8' TEMPLATE "template0"''' + ) def test_sql_table_creation_raises_with_collation(self): - settings = {'COLLATION': 'test'} + settings = {"COLLATION": "test"} msg = ( - 'PostgreSQL does not support collation setting at database ' - 'creation time.' + "PostgreSQL does not support collation setting at database " + "creation time." ) with self.assertRaisesMessage(ImproperlyConfigured, msg): self.check_sql_table_creation_suffix(settings, None) def _execute_raise_database_already_exists(self, cursor, parameters, keepdb=False): - error = DatabaseError('database %s already exists' % parameters['dbname']) + error = DatabaseError("database %s already exists" % parameters["dbname"]) error.pgcode = errorcodes.DUPLICATE_DATABASE raise DatabaseError() from error def _execute_raise_permission_denied(self, cursor, parameters, keepdb=False): - error = DatabaseError('permission denied to create database') + error = DatabaseError("permission denied to create database") error.pgcode = errorcodes.INSUFFICIENT_PRIVILEGE raise DatabaseError() from error def patch_test_db_creation(self, execute_create_test_db): - return mock.patch.object(BaseDatabaseCreation, '_execute_create_test_db', execute_create_test_db) + return mock.patch.object( + BaseDatabaseCreation, "_execute_create_test_db", execute_create_test_db + ) - @mock.patch('sys.stdout', new_callable=StringIO) - @mock.patch('sys.stderr', new_callable=StringIO) + @mock.patch("sys.stdout", new_callable=StringIO) + @mock.patch("sys.stderr", new_callable=StringIO) def test_create_test_db(self, *mocked_objects): creation = DatabaseCreation(connection) # Simulate test database creation raising "database already exists" with self.patch_test_db_creation(self._execute_raise_database_already_exists): - with mock.patch('builtins.input', return_value='no'): + with mock.patch("builtins.input", return_value="no"): with self.assertRaises(SystemExit): # SystemExit is raised if the user answers "no" to the # prompt asking if it's okay to delete the test database. - creation._create_test_db(verbosity=0, autoclobber=False, keepdb=False) + creation._create_test_db( + verbosity=0, autoclobber=False, keepdb=False + ) # "Database already exists" error is ignored when keepdb is on creation._create_test_db(verbosity=0, autoclobber=False, keepdb=True) # Simulate test database creation raising unexpected error with self.patch_test_db_creation(self._execute_raise_permission_denied): - with mock.patch.object(DatabaseCreation, '_database_exists', return_value=False): + with mock.patch.object( + DatabaseCreation, "_database_exists", return_value=False + ): with self.assertRaises(SystemExit): - creation._create_test_db(verbosity=0, autoclobber=False, keepdb=False) + creation._create_test_db( + verbosity=0, autoclobber=False, keepdb=False + ) with self.assertRaises(SystemExit): - creation._create_test_db(verbosity=0, autoclobber=False, keepdb=True) + creation._create_test_db( + verbosity=0, autoclobber=False, keepdb=True + ) # Simulate test database creation raising "insufficient privileges". # An error shouldn't appear when keepdb is on and the database already # exists. with self.patch_test_db_creation(self._execute_raise_permission_denied): - with mock.patch.object(DatabaseCreation, '_database_exists', return_value=True): + with mock.patch.object( + DatabaseCreation, "_database_exists", return_value=True + ): creation._create_test_db(verbosity=0, autoclobber=False, keepdb=True) |