[Python-checkins] r73282 - python/trunk/Lib/test/test_winreg.py
benjamin.peterson
python-checkins at python.org
Mon Jun 8 01:12:44 CEST 2009
Author: benjamin.peterson
Date: Mon Jun 8 01:12:44 2009
New Revision: 73282
Log:
backport r73273
Modified:
python/trunk/Lib/test/test_winreg.py
Modified: python/trunk/Lib/test/test_winreg.py
==============================================================================
--- python/trunk/Lib/test/test_winreg.py (original)
+++ python/trunk/Lib/test/test_winreg.py Mon Jun 8 01:12:44 2009
@@ -35,6 +35,27 @@
class WinregTests(unittest.TestCase):
remote_name = None
+ def setUp(self):
+ # Make sure that the test key is absent when the test
+ # starts.
+ self.delete_tree(HKEY_CURRENT_USER, test_key_name)
+
+ def delete_tree(self, root, subkey):
+ try:
+ hkey = OpenKey(root, subkey, KEY_ALL_ACCESS)
+ except WindowsError:
+ # subkey does not exist
+ return
+ while True:
+ try:
+ subsubkey = EnumKey(hkey, 0)
+ except WindowsError:
+ # no more subkeys
+ break
+ self.delete_tree(hkey, subsubkey)
+ CloseKey(hkey)
+ DeleteKey(root, subkey)
+
def WriteTestData(self, root_key):
# Set the default value for this key.
SetValue(root_key, test_key_name, REG_SZ, "Default value")
More information about the Python-checkins
mailing list