Refs #33308 -- Used psycopg's errors instead of errorcodes.

This commit is contained in:
Florian Apolloner 2022-12-01 08:35:14 +01:00 committed by Mariusz Felisiak
parent e20fd899be
commit 3cafb783f3
2 changed files with 8 additions and 7 deletions

View file

@ -13,7 +13,7 @@ try:
except ImportError:
pass
else:
from psycopg2 import errorcodes
from psycopg2 import errors
from django.db.backends.postgresql.creation import DatabaseCreation
@ -73,13 +73,13 @@ class DatabaseCreationTests(SimpleTestCase):
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.pgcode = errorcodes.DUPLICATE_DATABASE
error = errors.DuplicateDatabase(
"database %s already exists" % parameters["dbname"]
)
raise DatabaseError() from error
def _execute_raise_permission_denied(self, cursor, parameters, keepdb=False):
error = DatabaseError("permission denied to create database")
error.pgcode = errorcodes.INSUFFICIENT_PRIVILEGE
error = errors.InsufficientPrivilege("permission denied to create database")
raise DatabaseError() from error
def patch_test_db_creation(self, execute_create_test_db):