From: Victor Stinner <victor.stinner@haypocalc.com>
Date: Thu, 19 May 2011 23:56:42 +0000 (+0200)
Subject: packaging.tests.test_command_install_data: restore correctly sysconfig._SCHEMES
X-Git-Tag: v3.3.0a1~2225
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=4050c69f78a9cfb9b16fe7a1bfae8e9fbda89024;p=python

packaging.tests.test_command_install_data: restore correctly sysconfig._SCHEMES
---

diff --git a/Lib/packaging/tests/test_command_install_data.py b/Lib/packaging/tests/test_command_install_data.py
index 8b8bbac774..0486427a2a 100644
--- a/Lib/packaging/tests/test_command_install_data.py
+++ b/Lib/packaging/tests/test_command_install_data.py
@@ -11,15 +11,22 @@ class InstallDataTestCase(support.TempdirManager,
                           unittest.TestCase):
 
     def test_simple_run(self):
-        self.addCleanup(setattr, sysconfig, '_SCHEMES', sysconfig._SCHEMES)
+        scheme = _get_default_scheme()
+        old_items = sysconfig._SCHEMES.items(scheme)
+        def restore():
+            sysconfig._SCHEMES.remove_section(scheme)
+            sysconfig._SCHEMES.add_section(scheme)
+            for option, value in old_items:
+                sysconfig._SCHEMES.set(scheme, option, value)
+        self.addCleanup(restore)
 
         pkg_dir, dist = self.create_dist()
         cmd = install_data(dist)
         cmd.install_dir = inst = os.path.join(pkg_dir, 'inst')
 
-        sysconfig._SCHEMES.set(_get_default_scheme(), 'inst',
+        sysconfig._SCHEMES.set(scheme, 'inst',
                                os.path.join(pkg_dir, 'inst'))
-        sysconfig._SCHEMES.set(_get_default_scheme(), 'inst2',
+        sysconfig._SCHEMES.set(scheme, 'inst2',
                                os.path.join(pkg_dir, 'inst2'))
 
         one = os.path.join(pkg_dir, 'one')
@@ -59,7 +66,7 @@ class InstallDataTestCase(support.TempdirManager,
         three = os.path.join(cmd.install_dir, 'three')
         self.write_file(three, 'xx')
 
-        sysconfig._SCHEMES.set(_get_default_scheme(), 'inst3',
+        sysconfig._SCHEMES.set(scheme, 'inst3',
                                cmd.install_dir)
 
         cmd.data_files = {one: '{inst}/one', two: '{inst2}/two',