summaryrefslogtreecommitdiff
path: root/django/test
diff options
context:
space:
mode:
authorSergey Fedoseev <fedoseev.sergey@gmail.com>2017-08-22 00:34:20 +0500
committerTim Graham <timograham@gmail.com>2017-08-21 15:34:20 -0400
commit30eb92fc1199931d3743cd20b39f38eff7d30489 (patch)
treede552dc3053e5385bfb13f2fd9be34672e785b26 /django/test
parent8d095c6378666e6d5f6cabc9e485c9db2618ff88 (diff)
Removed need for hasattr(check, 'tags') in CheckRegistry.
It was needed to pass tests because registered checks weren't properly mocked.
Diffstat (limited to 'django/test')
-rw-r--r--django/test/utils.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/django/test/utils.py b/django/test/utils.py
index bf849ae0d1..e1cca58b27 100644
--- a/django/test/utils.py
+++ b/django/test/utils.py
@@ -499,10 +499,14 @@ class override_system_checks(TestContextDecorator):
def enable(self):
self.old_checks = self.registry.registered_checks
- self.registry.registered_checks = self.new_checks
+ self.registry.registered_checks = []
+ for check in self.new_checks:
+ self.registry.register(check, *getattr(check, 'tags', ()))
self.old_deployment_checks = self.registry.deployment_checks
if self.deployment_checks is not None:
- self.registry.deployment_checks = self.deployment_checks
+ self.registry.deployment_checks = []
+ for check in self.deployment_checks:
+ self.registry.register(check, *getattr(check, 'tags', ()), deploy=True)
def disable(self):
self.registry.registered_checks = self.old_checks