test_platform: Save/restore os.environ on Windows

vstinner · vstinner · commit 0fc55a234f03 · 2016-09-10T06:24:47.000-04:00
diff --git a/Lib/test/test_platform.py b/Lib/test/test_platform.py
@@ -18,6 +18,12 @@ def test_architecture_via_symlink(self): # issue3762
         # On Windows, the EXE needs to know where pythonXY.dll is at so we have
         # to add the directory to the path.
         if sys.platform == "win32":
+            def restore_environ(old_env):
+                os.environ.clear()
+                os.environ.update(old_env)
+
+            self.addCleanup(restore_environ, dict(os.environ))
+
             os.environ["Path"] = "{};{}".format(
                 os.path.dirname(sys.executable), os.environ["Path"])
 
@@ -26,6 +32,7 @@ def get(python):
                 'import platform; print(platform.architecture())']
             p = subprocess.Popen(cmd, stdout=subprocess.PIPE)
             return p.communicate()
+
         real = os.path.realpath(sys.executable)
         link = os.path.abspath(support.TESTFN)
         os.symlink(real, link)
