mirror of
https://github.com/django/django.git
synced 2024-12-23 09:36:06 +00:00
93 lines
3.8 KiB
Python
93 lines
3.8 KiB
Python
"""Tests for django.db.utils."""
|
|
import unittest
|
|
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from django.db import DEFAULT_DB_ALIAS, ProgrammingError, connection
|
|
from django.db.utils import ConnectionHandler, load_backend
|
|
from django.test import SimpleTestCase, TestCase
|
|
from django.utils.connection import ConnectionDoesNotExist
|
|
|
|
|
|
class ConnectionHandlerTests(SimpleTestCase):
|
|
|
|
def test_connection_handler_no_databases(self):
|
|
"""
|
|
Empty DATABASES and empty 'default' settings default to the dummy
|
|
backend.
|
|
"""
|
|
for DATABASES in (
|
|
{}, # Empty DATABASES setting.
|
|
{'default': {}}, # Empty 'default' database.
|
|
):
|
|
with self.subTest(DATABASES=DATABASES):
|
|
self.assertImproperlyConfigured(DATABASES)
|
|
|
|
def assertImproperlyConfigured(self, DATABASES):
|
|
conns = ConnectionHandler(DATABASES)
|
|
self.assertEqual(conns[DEFAULT_DB_ALIAS].settings_dict['ENGINE'], 'django.db.backends.dummy')
|
|
msg = (
|
|
'settings.DATABASES is improperly configured. Please supply the '
|
|
'ENGINE value. Check settings documentation for more details.'
|
|
)
|
|
with self.assertRaisesMessage(ImproperlyConfigured, msg):
|
|
conns[DEFAULT_DB_ALIAS].ensure_connection()
|
|
|
|
def test_no_default_database(self):
|
|
DATABASES = {'other': {}}
|
|
conns = ConnectionHandler(DATABASES)
|
|
msg = "You must define a 'default' database."
|
|
with self.assertRaisesMessage(ImproperlyConfigured, msg):
|
|
conns['other'].ensure_connection()
|
|
|
|
def test_nonexistent_alias(self):
|
|
msg = "The connection 'nonexistent' doesn't exist."
|
|
conns = ConnectionHandler({
|
|
DEFAULT_DB_ALIAS: {'ENGINE': 'django.db.backends.dummy'},
|
|
})
|
|
with self.assertRaisesMessage(ConnectionDoesNotExist, msg):
|
|
conns['nonexistent']
|
|
|
|
def test_ensure_defaults_nonexistent_alias(self):
|
|
msg = "The connection 'nonexistent' doesn't exist."
|
|
conns = ConnectionHandler({
|
|
DEFAULT_DB_ALIAS: {'ENGINE': 'django.db.backends.dummy'},
|
|
})
|
|
with self.assertRaisesMessage(ConnectionDoesNotExist, msg):
|
|
conns.ensure_defaults('nonexistent')
|
|
|
|
def test_prepare_test_settings_nonexistent_alias(self):
|
|
msg = "The connection 'nonexistent' doesn't exist."
|
|
conns = ConnectionHandler({
|
|
DEFAULT_DB_ALIAS: {'ENGINE': 'django.db.backends.dummy'},
|
|
})
|
|
with self.assertRaisesMessage(ConnectionDoesNotExist, msg):
|
|
conns.prepare_test_settings('nonexistent')
|
|
|
|
|
|
class DatabaseErrorWrapperTests(TestCase):
|
|
|
|
@unittest.skipUnless(connection.vendor == 'postgresql', 'PostgreSQL test')
|
|
def test_reraising_backend_specific_database_exception(self):
|
|
with connection.cursor() as cursor:
|
|
msg = 'table "X" does not exist'
|
|
with self.assertRaisesMessage(ProgrammingError, msg) as cm:
|
|
cursor.execute('DROP TABLE "X"')
|
|
self.assertNotEqual(type(cm.exception), type(cm.exception.__cause__))
|
|
self.assertIsNotNone(cm.exception.__cause__)
|
|
self.assertIsNotNone(cm.exception.__cause__.pgcode)
|
|
self.assertIsNotNone(cm.exception.__cause__.pgerror)
|
|
|
|
|
|
class LoadBackendTests(SimpleTestCase):
|
|
|
|
def test_load_backend_invalid_name(self):
|
|
msg = (
|
|
"'foo' isn't an available database backend or couldn't be "
|
|
"imported. Check the above exception. To use one of the built-in "
|
|
"backends, use 'django.db.backends.XXX', where XXX is one of:\n"
|
|
" 'mysql', 'oracle', 'postgresql', 'sqlite3'"
|
|
)
|
|
with self.assertRaisesMessage(ImproperlyConfigured, msg) as cm:
|
|
load_backend('foo')
|
|
self.assertEqual(str(cm.exception.__cause__), "No module named 'foo'")
|