mirror of
https://github.com/jazzband/django-constance.git
synced 2026-03-16 22:40:24 +00:00
bd8041c55f
* provides: base override class; unittest and pytest overrides * raise invalid config error earlier * update AUTHORS * avoid AttributeError * fix comment * add tests * fix tests, update docstring * update docs, improve tests * fix docs * fix markdown * refactor pytest override, use hidden fixture, refactor base and unittest classes * improve docstring and error * refactor pytest override to use hooks * set minimum pytest version * revert empty lines removal * introduce pytest test runner for package, refactoring * WIP * Finalize tox config, refactor docs, add global fixture * skip py35 * pytest command: remove unnecessary ignore * address comments * Update constance/test/pytest.py * address comments * add test for checking nested markers Co-authored-by: Camilo Nova <camilo.nova@gmail.com> Co-authored-by: Paweł Zarębski <ppjzarebski@gmail.com>
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
from functools import wraps
|
|
|
|
from django.test import SimpleTestCase
|
|
from django.test.utils import override_settings
|
|
|
|
from .. import config
|
|
|
|
__all__ = ('override_config',)
|
|
|
|
|
|
class override_config(override_settings):
|
|
"""
|
|
Decorator to modify constance setting for TestCase.
|
|
|
|
Based on django.test.utils.override_settings.
|
|
"""
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.original_values = {}
|
|
|
|
def __call__(self, test_func):
|
|
"""
|
|
Modify the decorated function to override config values.
|
|
"""
|
|
if isinstance(test_func, type):
|
|
if not issubclass(test_func, SimpleTestCase):
|
|
raise Exception(
|
|
"Only subclasses of Django SimpleTestCase can be "
|
|
"decorated with override_config")
|
|
return self.modify_test_case(test_func)
|
|
else:
|
|
@wraps(test_func)
|
|
def inner(*args, **kwargs):
|
|
with self:
|
|
return test_func(*args, **kwargs)
|
|
return inner
|
|
|
|
def modify_test_case(self, test_case):
|
|
"""
|
|
Override the config by modifying TestCase methods.
|
|
|
|
This method follows the Django <= 1.6 method of overriding the
|
|
_pre_setup and _post_teardown hooks rather than modifying the TestCase
|
|
itself.
|
|
"""
|
|
original_pre_setup = test_case._pre_setup
|
|
original_post_teardown = test_case._post_teardown
|
|
|
|
def _pre_setup(inner_self):
|
|
self.enable()
|
|
original_pre_setup(inner_self)
|
|
|
|
def _post_teardown(inner_self):
|
|
original_post_teardown(inner_self)
|
|
self.disable()
|
|
|
|
test_case._pre_setup = _pre_setup
|
|
test_case._post_teardown = _post_teardown
|
|
|
|
return test_case
|
|
|
|
def enable(self):
|
|
"""
|
|
Store original config values and set overridden values.
|
|
"""
|
|
# Store the original values to an instance variable
|
|
for config_key in self.options:
|
|
self.original_values[config_key] = getattr(config, config_key)
|
|
|
|
# Update config with the overriden values
|
|
self.unpack_values(self.options)
|
|
|
|
def disable(self):
|
|
"""
|
|
Set original values to the config.
|
|
"""
|
|
self.unpack_values(self.original_values)
|
|
|
|
@staticmethod
|
|
def unpack_values(options):
|
|
"""
|
|
Unpack values from the given dict to config.
|
|
"""
|
|
for name, value in options.items():
|
|
setattr(config, name, value)
|