mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-03-06 15:32:33 +00:00
Code Issues 32 Releases Wiki Activity

Added TransactionTestCase.multi_db instead of using getattr().

Browse Source
This commit is contained in:
Tim Graham 2017-02-13 08:46:00 -05:00 committed by GitHub
parent 9718fa2e8a
commit 3f5a35d6e2
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
django/test/testcases.py
View File

@ -781,6 +781,9 @@ class TransactionTestCase(SimpleTestCase):
# Subclasses can define fixtures which will be automatically installed. # Subclasses can define fixtures which will be automatically installed.
fixtures = None fixtures = None
# Do the tests in this class query non-default databases?
multi_db = False
# If transactions aren't available, Django will serialize the database # If transactions aren't available, Django will serialize the database
# contents into a fixture during setup and flush and reload them # contents into a fixture during setup and flush and reload them
# during teardown (as flush does not restore data from migrations). # during teardown (as flush does not restore data from migrations).
@ -827,7 +830,7 @@ class TransactionTestCase(SimpleTestCase):
def _databases_names(cls, include_mirrors=True): def _databases_names(cls, include_mirrors=True):
# If the test case has a multi_db=True flag, act on all databases, # If the test case has a multi_db=True flag, act on all databases,
# including mirrors or not. Otherwise, just on the default DB. # including mirrors or not. Otherwise, just on the default DB.
if getattr(cls, 'multi_db', False): if cls.multi_db:
return [ return [
alias for alias in connections alias for alias in connections
if include_mirrors or not connections[alias].settings_dict['TEST']['MIRROR'] if include_mirrors or not connections[alias].settings_dict['TEST']['MIRROR']