Skip to content

Commit

Permalink
Merge pull request #131 from pganssle/fix_test_oldenv
Browse files Browse the repository at this point in the history
Handle check for PYTHONTZPATH failing
  • Loading branch information
pganssle committed Mar 10, 2023
2 parents 8aa8e6b + 2df26ba commit 666d80c
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion tests/test_zoneinfo.py
Original file line number Diff line number Diff line change
Expand Up @@ -1532,13 +1532,20 @@ class TzPathTest(TzPathUserMixin, ZoneInfoTestBase):
@contextlib.contextmanager
def python_tzpath_context(value):
path_var = "PYTHONTZPATH"
unset_env_sentinel = object()
old_env = unset_env_sentinel
try:
with OS_ENV_LOCK:
old_env = os.environ.get(path_var, None)
os.environ[path_var] = value
yield
finally:
if old_env is None:
if old_env is unset_env_sentinel:
# In this case, `old_env` was never retrieved from the
# environment for whatever reason, so there's no need to
# reset the environment TZPATH.
pass
elif old_env is None:
del os.environ[path_var]
else:
os.environ[path_var] = old_env # pragma: nocover
Expand Down

0 comments on commit 666d80c

Please sign in to comment.