From 3f5a35d6e2dfc0c981dbd8945a8c7cd35c32e5c8 Mon Sep 17 00:00:00 2001 From: Tim Graham Date: Mon, 13 Feb 2017 08:46:00 -0500 Subject: [PATCH] Added TransactionTestCase.multi_db instead of using getattr(). --- django/test/testcases.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/django/test/testcases.py b/django/test/testcases.py index 7955488f48..b8d57378de 100644 --- a/django/test/testcases.py +++ b/django/test/testcases.py @@ -781,6 +781,9 @@ class TransactionTestCase(SimpleTestCase): # Subclasses can define fixtures which will be automatically installed. 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 # contents into a fixture during setup and flush and reload them # during teardown (as flush does not restore data from migrations). @@ -827,7 +830,7 @@ class TransactionTestCase(SimpleTestCase): def _databases_names(cls, include_mirrors=True): # If the test case has a multi_db=True flag, act on all databases, # including mirrors or not. Otherwise, just on the default DB. - if getattr(cls, 'multi_db', False): + if cls.multi_db: return [ alias for alias in connections if include_mirrors or not connections[alias].settings_dict['TEST']['MIRROR']