diff options
| author | Sergey Fedoseev <fedoseev.sergey@gmail.com> | 2017-08-22 00:34:20 +0500 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2017-08-21 15:34:20 -0400 |
| commit | 30eb92fc1199931d3743cd20b39f38eff7d30489 (patch) | |
| tree | de552dc3053e5385bfb13f2fd9be34672e785b26 /django/test | |
| parent | 8d095c6378666e6d5f6cabc9e485c9db2618ff88 (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.py | 8 |
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 |
