diff options
author | Alexander Belopolsky <alexander.belopolsky@gmail.com> | 2012-04-29 16:12:27 -0400 |
---|---|---|
committer | Alexander Belopolsky <alexander.belopolsky@gmail.com> | 2012-04-29 16:12:27 -0400 |
commit | 5a38f80f9cb9bc2dc8692aa6ca7b6d738342749b (patch) | |
tree | aed4da78596e4b5939e5c8e82287494981841a23 /Lib/test/support.py | |
parent | 1182351e69c81b95186a5318f28a6e1bd2fc0a2f (diff) | |
parent | 2420d831582a5403d679b6383933112948d476fe (diff) | |
download | cpython-git-5a38f80f9cb9bc2dc8692aa6ca7b6d738342749b.tar.gz |
Issue #10941: Fix imaplib.Internaldate2tuple to produce correct result near
the DST transition. Patch by Joe Peterson.
Diffstat (limited to 'Lib/test/support.py')
-rw-r--r-- | Lib/test/support.py | 31 |
1 files changed, 30 insertions, 1 deletions
diff --git a/Lib/test/support.py b/Lib/test/support.py index a1ab09c2be..a7d0833a60 100644 --- a/Lib/test/support.py +++ b/Lib/test/support.py @@ -57,7 +57,7 @@ __all__ = [ "get_attribute", "swap_item", "swap_attr", "requires_IEEE_754", "TestHandler", "Matcher", "can_symlink", "skip_unless_symlink", "import_fresh_module", "requires_zlib", "PIPE_MAX_SIZE", "failfast", - "anticipate_failure" + "anticipate_failure", "run_with_tz" ] class Error(Exception): @@ -1100,6 +1100,35 @@ def run_with_locale(catstr, *locales): return decorator #======================================================================= +# Decorator for running a function in a specific timezone, correctly +# resetting it afterwards. + +def run_with_tz(tz): + def decorator(func): + def inner(*args, **kwds): + if 'TZ' in os.environ: + orig_tz = os.environ['TZ'] + else: + orig_tz = None + os.environ['TZ'] = tz + time.tzset() + + # now run the function, resetting the tz on exceptions + try: + return func(*args, **kwds) + finally: + if orig_tz == None: + del os.environ['TZ'] + else: + os.environ['TZ'] = orig_tz + time.tzset() + + inner.__name__ = func.__name__ + inner.__doc__ = func.__doc__ + return inner + return decorator + +#======================================================================= # Big-memory-test support. Separate from 'resources' because memory use # should be configurable. |