From 62a4868adb1a2cc536e6cd7352274a81ca25d38f Mon Sep 17 00:00:00 2001 From: Orgad Shaneh Date: Fri, 4 Jul 2025 11:48:42 +0300 Subject: [PATCH] Do not overwrite ~/.netrc when running tests on Windows os.expanduser uses USERPROFILE on Windows. --- tests/test_oauth2_session.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/test_oauth2_session.py b/tests/test_oauth2_session.py index 7e3e63c..317f86f 100644 --- a/tests/test_oauth2_session.py +++ b/tests/test_oauth2_session.py @@ -535,9 +535,10 @@ class OAuth2SessionNetrcTest(OAuth2SessionTest): def setUp(self): # Set up a temporary home directory + self.home_env = "HOME" if os.name != "nt" else "USERPROFILE" self.homedir = tempfile.mkdtemp() - self.prehome = os.environ.get("HOME", None) - os.environ["HOME"] = self.homedir + self.prehome = os.environ.get(self.home_env, None) + os.environ[self.home_env] = self.homedir # Write a .netrc file that will cause problems netrc_loc = os.path.expanduser("~/.netrc") @@ -550,5 +551,5 @@ def tearDown(self): super(OAuth2SessionNetrcTest, self).tearDown() if self.prehome is not None: - os.environ["HOME"] = self.prehome + os.environ[self.home_env] = self.prehome shutil.rmtree(self.homedir)