summaryrefslogtreecommitdiff
path: root/tests/utils_tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests/utils_tests')
-rw-r--r--tests/utils_tests/test_timezone.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/tests/utils_tests/test_timezone.py b/tests/utils_tests/test_timezone.py
index ae78d68103..9414a6097e 100644
--- a/tests/utils_tests/test_timezone.py
+++ b/tests/utils_tests/test_timezone.py
@@ -71,6 +71,36 @@ class TimezoneTests(unittest.TestCase):
finally:
timezone.deactivate()
+ def test_override_decorator(self):
+ default = timezone.get_default_timezone()
+
+ @timezone.override(EAT)
+ def func_tz_eat():
+ self.assertIs(EAT, timezone.get_current_timezone())
+
+ @timezone.override(None)
+ def func_tz_none():
+ self.assertIs(default, timezone.get_current_timezone())
+
+ try:
+ timezone.activate(ICT)
+
+ func_tz_eat()
+ self.assertIs(ICT, timezone.get_current_timezone())
+
+ func_tz_none()
+ self.assertIs(ICT, timezone.get_current_timezone())
+
+ timezone.deactivate()
+
+ func_tz_eat()
+ self.assertIs(default, timezone.get_current_timezone())
+
+ func_tz_none()
+ self.assertIs(default, timezone.get_current_timezone())
+ finally:
+ timezone.deactivate()
+
def test_copy(self):
self.assertIsInstance(copy.copy(timezone.UTC()), timezone.UTC)
self.assertIsInstance(copy.copy(timezone.LocalTimezone()), timezone.LocalTimezone)